diff --git a/src/main/java/com/sjhy/plugin/provider/EasyCodeGlobalVariableProvider.java b/src/main/java/com/sjhy/plugin/provider/EasyCodeGlobalVariableProvider.java index d68ae5f..d22f1cf 100644 --- a/src/main/java/com/sjhy/plugin/provider/EasyCodeGlobalVariableProvider.java +++ b/src/main/java/com/sjhy/plugin/provider/EasyCodeGlobalVariableProvider.java @@ -38,6 +38,7 @@ public class EasyCodeGlobalVariableProvider extends VtlGlobalVariableProvider { result.add(new VtlLightVariable("time", file, "com.sjhy.plugin.tool.TimeUtils")); result.add(new VtlLightVariable("tableInfo", file, "com.sjhy.plugin.entity.TableInfo")); result.add(new VtlLightVariable("tableInfoList", file, "java.util.List")); + result.add(new VtlLightVariable("generateService", file, "com.sjhy.plugin.tool.ExtraCodeGenerateUtils")); return result; } diff --git a/src/main/java/com/sjhy/plugin/service/impl/CodeGenerateServiceImpl.java b/src/main/java/com/sjhy/plugin/service/impl/CodeGenerateServiceImpl.java index d64e9d7..ed4cee0 100644 --- a/src/main/java/com/sjhy/plugin/service/impl/CodeGenerateServiceImpl.java +++ b/src/main/java/com/sjhy/plugin/service/impl/CodeGenerateServiceImpl.java @@ -102,22 +102,41 @@ public class CodeGenerateServiceImpl implements CodeGenerateService { * @param title 是否显示提示 */ private void generate(Collection