Add support for array type

alibaba · Feb 5, 2024 · 16ec867 · 16ec867
1 parent c7f50f7
commit 16ec867
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/...in/java/com/alibaba/compileflow/engine/process/preruntime/generator/code/ClassTarget.java b/...in/java/com/alibaba/compileflow/engine/process/preruntime/generator/code/ClassTarget.java
@@ -217,7 +217,9 @@ public String generateCode() {
     }
 
     private String getImportName(ClassWrapper importType) {
-        return importType.getPackageName() + "." + importType.getShortRawName();
+        int index = importType.getShortRawName().indexOf("[]");
+        return index == -1 ? importType.getPackageName() + "." + importType.getShortRawName()
+                : importType.getPackageName() + "." + importType.getShortRawName().substring(0, index);
     }
 
     private boolean notContainImportedType(ClassWrapper classWrapper) {