diff --git a/server/like-generator/src/main/java/com/hxkj/generator/util/VelocityUtil.java b/server/like-generator/src/main/java/com/hxkj/generator/util/VelocityUtil.java index 90fb7e34..b1165147 100644 --- a/server/like-generator/src/main/java/com/hxkj/generator/util/VelocityUtil.java +++ b/server/like-generator/src/main/java/com/hxkj/generator/util/VelocityUtil.java @@ -117,17 +117,17 @@ public class VelocityUtil { public static List getTemplateList(String genTpl, List columns) { List templates = new LinkedList<>(); templates.add("java/controller.java.vm"); -// templates.add("java/entity.java.vm"); -// templates.add("java/mapper.java.vm"); -// templates.add("java/service.java.vm"); -// templates.add("java/serviceImpl.java.vm"); -// templates.add("java/validate.java.vm"); -// if (VelocityUtil.getIsEqually(columns)) { -// templates.add("java/vo.java.vm"); -// } else { -// templates.add("java/voList.java.vm"); -// templates.add("java/voDetail.java.vm"); -// } + templates.add("java/entity.java.vm"); + templates.add("java/mapper.java.vm"); + templates.add("java/service.java.vm"); + templates.add("java/serviceImpl.java.vm"); + templates.add("java/validate.java.vm"); + if (VelocityUtil.getIsEqually(columns)) { + templates.add("java/vo.java.vm"); + } else { + templates.add("java/voList.java.vm"); + templates.add("java/voDetail.java.vm"); + } // if (GenConstants.TPL_CRUD.equals(genTpl)) { // templates.add("vue/index.vue.vm"); // } @@ -173,6 +173,14 @@ public class VelocityUtil { fileName = StringUtil.format("{}/vo/{}/{}Vo.java", GenConfig.adminPackage, moduleName, entityName); } + else if (template.contains("voList.java.vm")) { + fileName = StringUtil.format("{}/vo/{}/{}ListVo.java", GenConfig.adminPackage, moduleName, entityName); + } + + else if (template.contains("voDetail.java.vm")) { + fileName = StringUtil.format("{}/vo/{}/{}DetailVo.java", GenConfig.adminPackage, moduleName, entityName); + } + return fileName; }