Ver código fonte

生成代码优化

wanggaokun 11 meses atrás
pai
commit
5fc880017d

+ 8 - 4
eco-modules/generator/src/main/resources/vm/java/listener.java.vm

@@ -57,10 +57,7 @@ public class ${ClassName}ImportListener extends AnalysisEventListener<${ClassNam
             #end
             if (ObjectUtil.isNull(${className}Vo1)) {
                 //不存在就新增
-                ${className}Bo.setVersion(0);
-                ${className}Bo.setCreateBy(loginUser.getUserId());
-                ${className}Bo.setUpdateBy(loginUser.getUserId());
-                ${className}Bo.setTenantId(loginUser.getTenantId());
+                setBo(${className}Bo)
                 ValidatorUtils.validate(${className}Bo);
                 #if($table.tree)
                     boolean inserted = ${className}Service.insertWithPk(${className}Bo);//树表需要前台传来主键值
@@ -100,6 +97,13 @@ public class ${ClassName}ImportListener extends AnalysisEventListener<${ClassNam
         }
     }
 
+    private void setBo(${ClassName}Bo ${className}Bo) {
+        ${className}Bo.setVersion(0);
+        ${className}Bo.setCreateBy(loginUser.getUserId());
+        ${className}Bo.setUpdateBy(loginUser.getUserId());
+        ${className}Bo.setTenantId(loginUser.getTenantId());
+    }
+
     @Override
     public void doAfterAllAnalysed(AnalysisContext context) {
         log.info("解析完成");