Merge pull request #21 from mindprince/query-names

Add support for assigning query-names.
qubole · Oct 21, 2014 · 28e922c · 28e922c
2 parents d4251be + baa2266
commit 28e922c
Show file tree

Hide file tree

Showing 11 changed files with 41 additions and 1 deletion.
diff --git a/src/main/java/com/qubole/qds/sdk/java/api/CommandApi.java b/src/main/java/com/qubole/qds/sdk/java/api/CommandApi.java
@@ -52,7 +52,7 @@ public interface CommandApi
 
     /**
      * @param query the query to be run
-     * @param db_tap_id the db_tap id where the query will be run
+     * @param dbTapId the db_tap id where the query will be run
      * @return new builder
      */
     public InvokableBuilder<CommandResponse> dbTapQuery(String query, int dbTapId);

diff --git a/src/main/java/com/qubole/qds/sdk/java/api/HadoopCommandBuilder.java b/src/main/java/com/qubole/qds/sdk/java/api/HadoopCommandBuilder.java
@@ -15,4 +15,6 @@ public enum SubCommandType
     public HadoopCommandBuilder sub_command_args(String sub_command_args);
 
     public HadoopCommandBuilder clusterLabel(String clusterLabel);
+
+    public HadoopCommandBuilder name(String commandName);
 }
diff --git a/src/main/java/com/qubole/qds/sdk/java/api/HiveCommandBuilder.java b/src/main/java/com/qubole/qds/sdk/java/api/HiveCommandBuilder.java
@@ -23,4 +23,6 @@ public interface HiveCommandBuilder extends InvokableBuilder<CommandResponse>
     public HiveCommandBuilder macro(String name, String value);
 
     public HiveCommandBuilder clusterLabel(String clusterLabel);
+
+    public HiveCommandBuilder name(String queryName);
 }
diff --git a/src/main/java/com/qubole/qds/sdk/java/api/PigCommandBuilder.java b/src/main/java/com/qubole/qds/sdk/java/api/PigCommandBuilder.java
@@ -12,4 +12,6 @@ public interface PigCommandBuilder extends InvokableBuilder<CommandResponse>
     public PigCommandBuilder latin_statements(String latin_statements);
 
     public PigCommandBuilder clusterLabel(String clusterLabel);
+
+    public PigCommandBuilder name(String commmandName);
 }
diff --git a/src/main/java/com/qubole/qds/sdk/java/api/PrestoCommandBuilder.java b/src/main/java/com/qubole/qds/sdk/java/api/PrestoCommandBuilder.java
@@ -9,4 +9,6 @@ public interface PrestoCommandBuilder extends InvokableBuilder<CommandResponse>
     public PrestoCommandBuilder query(String query);
 
     public PrestoCommandBuilder clusterLabel(String clusterLabel);
+
+    public PrestoCommandBuilder name(String queryName);
 }
diff --git a/src/main/java/com/qubole/qds/sdk/java/api/ShellCommandBuilder.java b/src/main/java/com/qubole/qds/sdk/java/api/ShellCommandBuilder.java
@@ -15,4 +15,6 @@ public interface ShellCommandBuilder extends InvokableBuilder<CommandResponse>
     public ShellCommandBuilder archives(List<String> archiveList);
 
     public ShellCommandBuilder clusterLabel(String clusterLabel);
+
+    public ShellCommandBuilder name(String commandName);
 }
diff --git a/src/main/java/com/qubole/qds/sdk/java/details/HadoopCommandBuilderImpl.java b/src/main/java/com/qubole/qds/sdk/java/details/HadoopCommandBuilderImpl.java
@@ -29,6 +29,12 @@ public HadoopCommandBuilder clusterLabel(String clusterLabel)
         return this;
     }
 
+    @Override
+    public HadoopCommandBuilder name(String commandName) {
+        node.put("name", commandName);
+        return this;
+    }
+
     @Override
     protected ObjectNode getEntity()
     {

diff --git a/src/main/java/com/qubole/qds/sdk/java/details/HiveCommandBuilderImpl.java b/src/main/java/com/qubole/qds/sdk/java/details/HiveCommandBuilderImpl.java
@@ -82,6 +82,12 @@ public HiveCommandBuilder clusterLabel(String clusterLabel)
         return this;
     }
 
+    @Override
+    public HiveCommandBuilder name(String queryName) {
+        node.put("name", queryName);
+        return this;
+    }
+
     @Override
     protected ObjectNode getEntity()
     {

diff --git a/src/main/java/com/qubole/qds/sdk/java/details/PigCommandBuilderImpl.java b/src/main/java/com/qubole/qds/sdk/java/details/PigCommandBuilderImpl.java
@@ -37,6 +37,12 @@ public PigCommandBuilder clusterLabel(String clusterLabel)
         return this;
     }
 
+    @Override
+    public PigCommandBuilder name(String commmandName) {
+        node.put("name", commmandName);
+        return this;
+    }
+
     @Override
     protected ObjectNode getEntity()
     {

diff --git a/src/main/java/com/qubole/qds/sdk/java/details/PrestoCommandBuilderImpl.java b/src/main/java/com/qubole/qds/sdk/java/details/PrestoCommandBuilderImpl.java
@@ -29,6 +29,12 @@ public PrestoCommandBuilder clusterLabel(String clusterLabel)
         return this;
     }
 
+    @Override
+    public PrestoCommandBuilder name(String queryName) {
+        node.put("name", queryName);
+        return this;
+    }
+
     @Override
     protected ObjectNode getEntity()
     {

diff --git a/src/main/java/com/qubole/qds/sdk/java/details/ShellCommandBuilderImpl.java b/src/main/java/com/qubole/qds/sdk/java/details/ShellCommandBuilderImpl.java
@@ -48,6 +48,12 @@ public ShellCommandBuilder clusterLabel(String clusterLabel)
         return this;
     }
 
+    @Override
+    public ShellCommandBuilder name(String commandName) {
+        node.put("name", commandName);
+        return this;
+    }
+
     @Override
     protected ObjectNode getEntity()
     {