Browse Source

生成代码优化

wanggaokun 11 months ago
parent
commit
71f3d783aa
1 changed files with 11 additions and 10 deletions
  1. 11 10
      eco-modules/generator/src/main/resources/vm/java/serviceImpl.java.vm

+ 11 - 10
eco-modules/generator/src/main/resources/vm/java/serviceImpl.java.vm

@@ -263,11 +263,7 @@ public class ${ClassName}ServiceImpl extends BaseServiceImpl<${ClassName}Mapper,
         } catch (IOException e) {
             throw new RuntimeException(e);
         }
-        ImportExportBo bo = new ImportExportBo();
-        BeanUtils.copyProperties(result, bo);
-        bo.setUpdateBy(loginUser.getUserId());
-        bo.setCreateBy(loginUser.getUserId());
-        bo.setType("0");
+        ImportExportBo bo = getImportExportBo(result, loginUser, "0");
         boolean flag = importExportService.insert(bo);
         if (flag) {
             log.info("异步导入日志写入成功");
@@ -277,17 +273,22 @@ public class ${ClassName}ServiceImpl extends BaseServiceImpl<${ClassName}Mapper,
     @Override
     public void asyncExport(List<${ClassName}Vo> listVo, String sheetName, LoginUser loginUser) {
         ExcelResultRes result = excelService.exportExcel(listVo, sheetName, ${ClassName}Vo.class);
-        ImportExportBo bo = new ImportExportBo();
-        BeanUtils.copyProperties(result, bo);
-        bo.setUpdateBy(loginUser.getUserId());
-        bo.setCreateBy(loginUser.getUserId());
-        bo.setType("1");
+        ImportExportBo bo = getImportExportBo(result, loginUser, "1");
         boolean flag = importExportService.insert(bo);
         if (flag) {
             log.info("异步导出日志写入成功");
         }
     }
 
+    private static ImportExportBo getImportExportBo(ExcelResultRes result, LoginUser loginUser, String type) {
+        ImportExportBo bo = new ImportExportBo();
+        BeanUtils.copyProperties(result, bo);
+        bo.setUpdateBy(loginUser.getUserId());
+        bo.setCreateBy(loginUser.getUserId());
+        bo.setType(type);
+        return bo;
+    }
+
     /**
      * 批量删除${functionName}
      *