From 8d650ae131ba8999fac2c8f347995eedad0fda7e Mon Sep 17 00:00:00 2001
From: Gian Merlino <gianmerlino@gmail.com>
Date: Tue, 22 Apr 2014 14:33:51 -0700
Subject: [PATCH] PullDependencies changes.

- Pull default Hadoop version by default.
- Allow additional coordinates on the command line.
---
 .../indexing/common/task/HadoopIndexTask.java |  5 +-
 .../java/io/druid/cli/PullDependencies.java   | 46 ++++++++++++++++++-
 2 files changed, 47 insertions(+), 4 deletions(-)

diff --git a/indexing-service/src/main/java/io/druid/indexing/common/task/HadoopIndexTask.java b/indexing-service/src/main/java/io/druid/indexing/common/task/HadoopIndexTask.java
index 7e9d4032e21..22529c62163 100644
--- a/indexing-service/src/main/java/io/druid/indexing/common/task/HadoopIndexTask.java
+++ b/indexing-service/src/main/java/io/druid/indexing/common/task/HadoopIndexTask.java
@@ -66,7 +66,8 @@ public class HadoopIndexTask extends AbstractTask
     extensionsConfig = Initialization.makeStartupInjector().getInstance(ExtensionsConfig.class);
   }
 
-  private static String defaultHadoopCoordinates = "org.apache.hadoop:hadoop-client:2.3.0";
+  public static String DEFAULT_HADOOP_COORDINATES = "org.apache.hadoop:hadoop-client:2.3.0";
+
   @JsonIgnore
   private final HadoopDruidIndexerSchema schema;
   @JsonIgnore
@@ -102,7 +103,7 @@ public class HadoopIndexTask extends AbstractTask
 
     this.schema = schema;
     this.hadoopDependencyCoordinates = hadoopDependencyCoordinates == null ? Arrays.<String>asList(
-        hadoopCoordinates == null ? defaultHadoopCoordinates : hadoopCoordinates
+        hadoopCoordinates == null ? DEFAULT_HADOOP_COORDINATES : hadoopCoordinates
     ) : hadoopDependencyCoordinates;
   }
 
diff --git a/services/src/main/java/io/druid/cli/PullDependencies.java b/services/src/main/java/io/druid/cli/PullDependencies.java
index 1131a42a0b0..c6d7a84885d 100644
--- a/services/src/main/java/io/druid/cli/PullDependencies.java
+++ b/services/src/main/java/io/druid/cli/PullDependencies.java
@@ -19,7 +19,17 @@
 
 package io.druid.cli;
 
+import com.google.api.client.util.Lists;
+import com.google.common.base.Throwables;
+import com.google.inject.Inject;
 import io.airlift.command.Command;
+import io.airlift.command.Option;
+import io.druid.indexing.common.task.HadoopIndexTask;
+import io.druid.initialization.Initialization;
+import io.druid.server.initialization.ExtensionsConfig;
+import io.tesla.aether.internal.DefaultTeslaAether;
+
+import java.util.List;
 
 
 @Command(
@@ -28,8 +38,40 @@ import io.airlift.command.Command;
 )
 public class PullDependencies implements Runnable
 {
+  @Option(name = "-c",
+          title = "coordinate",
+          description = "extra dependencies to pull down (e.g. hadoop coordinates)",
+          required = false)
+  public List<String> coordinates;
+
+  @Option(name = "--no-default-hadoop",
+          description = "don't pull down the default HadoopIndexTask dependencies",
+          required = false)
+  public boolean noDefaultHadoop;
+
+  @Inject
+  public ExtensionsConfig extensionsConfig = null;
+
   @Override
-  public void run() {
-    // dependencies are pulled down as a side-effect of Guice injection
+  public void run()
+  {
+    // Druid dependencies are pulled down as a side-effect of Guice injection. Extra dependencies are pulled down as
+    // a side-effect of getting class loaders.
+    final List<String> allCoordinates = Lists.newArrayList();
+    if (coordinates != null) {
+      allCoordinates.addAll(coordinates);
+    }
+    if (!noDefaultHadoop) {
+      allCoordinates.add(HadoopIndexTask.DEFAULT_HADOOP_COORDINATES);
+    }
+    try {
+      final DefaultTeslaAether aetherClient = Initialization.getAetherClient(extensionsConfig);
+      for (final String coordinate : allCoordinates) {
+        Initialization.getClassLoaderForCoordinates(aetherClient, coordinate);
+      }
+    }
+    catch (Exception e) {
+      throw Throwables.propagate(e);
+    }
   }
 }