[ISSUE-205] Optimize type cast

TuGraph-family · Oct 20, 2023 · d65faec · d65faec
1 parent 4521a9b
commit d65faec
Show file tree

Hide file tree

Showing 5 changed files with 550 additions and 183 deletions.
diff --git a/...flow-dsl-common/src/main/java/com/antgroup/geaflow/dsl/common/util/FunctionCallUtils.java b/...flow-dsl-common/src/main/java/com/antgroup/geaflow/dsl/common/util/FunctionCallUtils.java
@@ -372,7 +372,11 @@ public static Object callMethod(Method method, Object target, Object[] params)
                 castParams[i] = TypeCastUtil.cast(params[i], getBoxType(defineTypes[i]));
             }
         }
-        return method.invoke(target, castParams);
+        Object result = method.invoke(target, castParams);
+        if (result instanceof String) { // convert string to binary string if the udf return string type.
+            result = BinaryString.fromString((String) result);
+        }
+        return result;
     }
 
     public static Class<?> typeClass(Class<?> type, boolean useBinary) {