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) {