Add support for assigning query-names.

qubole · Oct 21, 2014 · baa2266 · baa2266
1 parent b515460
commit baa2266
Show file tree

Hide file tree

Showing 10 changed files with 40 additions and 0 deletions.
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()
     {