优化代码模板

als-modules/agile-assurance/src/main/java/org/eco/als/listener/AircraftImportListener.java

@@ -53,10 +53,7 @@ public class AircraftImportListener extends AnalysisEventListener<AircraftImport
             AircraftVo pAircraftVo = aircraftService.selectByName(aircraftBo.getParentName());
             if (ObjectUtil.isNull(aircraftVo1)) {
                 //不存在就新增
-                aircraftBo.setVersion(0);
-                aircraftBo.setCreateBy(loginUser.getUserId());
-                aircraftBo.setUpdateBy(loginUser.getUserId());
-                aircraftBo.setTenantId(loginUser.getTenantId());
+                setBo(aircraftBo);
                 aircraftBo.setParentId(ObjectUtil.isNull(pAircraftVo) ? 0 : pAircraftVo.getId());
                 aircraftBo.setParentName(ObjectUtil.isNull(pAircraftVo) ? "" : pAircraftVo.getName());
                 ValidatorUtils.validate(aircraftBo);
@@ -89,6 +86,13 @@ public class AircraftImportListener extends AnalysisEventListener<AircraftImport
         }
     }
 
+    private void setBo(AircraftBo aircraftBo) {
+        aircraftBo.setVersion(0);
+        aircraftBo.setCreateBy(loginUser.getUserId());
+        aircraftBo.setUpdateBy(loginUser.getUserId());
+        aircraftBo.setTenantId(loginUser.getTenantId());
+    }
+
     @Override
     public void doAfterAllAnalysed(AnalysisContext context) {
         log.info("解析完成" );

als-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);//树表需要前台传来主键值
@@ -99,6 +96,12 @@ public class ${ClassName}ImportListener extends AnalysisEventListener<${ClassNam
             log.error(msg, e);
         }
     }
+    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) {