Merge pull request #669 from brightics/brtc-issue-668

Support Enterprise version of UDF
brightics · Oct 2, 2019 · 0a9f2af · 0a9f2af
2 parents b68cacf + 5aa9d86
commit 0a9f2af
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/...ent/src/main/java/com/samsung/sds/brightics/agent/context/python/PythonScriptBuilder.java b/...ent/src/main/java/com/samsung/sds/brightics/agent/context/python/PythonScriptBuilder.java
@@ -91,6 +91,18 @@ PythonScriptBuilder addInputsScript() {
 
         return this;
     }
+
+    PythonScriptBuilder addInputsUDFScript() {
+        DataAttribute inputs = message.getInputs();
+
+        for (Entry<String, Collection<String>> entry : inputs.entrySetAsIterableValue()) {
+            script.add(entry.getValue().stream().
+                    map(this::makeDataLoadScript).
+                    collect(Collectors.joining(", ", String.format("%s = ", entry.getKey()), " ")));
+        }
+
+        return this;
+    }
 
     PythonScriptBuilder addOutDataScript() {
         String[] outDatas = message.getOutData();

diff --git a/...-agent/src/main/java/com/samsung/sds/brightics/agent/context/python/PythonScriptType.java b/...-agent/src/main/java/com/samsung/sds/brightics/agent/context/python/PythonScriptType.java
@@ -59,7 +59,7 @@ public String getSource(TaskMessageWrapper message) {
         @Override
         public String getSource(TaskMessageWrapper message) {
             return scriptBuilder(message).
-                    addInputsScript().
+                    addInputsUDFScript().
                     addParameters().
                     addScript().
                     addPutOutputsScript().