diff --git a/src/main/java/cn/dalgen/mybatis/gen/dataloaders/DalgenLoader.java b/src/main/java/cn/dalgen/mybatis/gen/dataloaders/DalgenLoader.java index 26d048808ec804b4eccc554457f40e9fe86b3970..384bd3c2e35823db18486e18eb9b0bacf3d35067 100755 --- a/src/main/java/cn/dalgen/mybatis/gen/dataloaders/DalgenLoader.java +++ b/src/main/java/cn/dalgen/mybatis/gen/dataloaders/DalgenLoader.java @@ -336,7 +336,9 @@ public class DalgenLoader extends AbstractDalgenLoader { private DAO preDAO(Gen gen, CfTable cfTable, Table table, DO doClass, Map resultMaps) { DAO dao = new DAO(); - dao.setClassName(table.getJavaName() + "DAO"); + String daoNaming = ConfigUtil.getConfig().getExtParam("DaoNaming"); + String daoClazzName = daoNaming.replace("[EntityName]", table.getJavaName()); + dao.setClassName(daoClazzName); dao.setPackageName(ConfigUtil.getCurrentDb().getGenPackage("dao")); @@ -464,11 +466,14 @@ public class DalgenLoader extends AbstractDalgenLoader { private DOMapper preDOMapper(Gen gen, CfTable cfTable, Table table, DO doClass, Map resultMaps) { DOMapper doMapper = new DOMapper(); - if (StringUtils.equals("DO", ConfigUtil.getConfig().getExtParam("DoMapperName"))) { - doMapper.setClassName(doClass.getClassName() + "Mapper"); - }else{ - doMapper.setClassName(table.getJavaName()+ "Mapper"); - } + String mapperInterfaceNaming = ConfigUtil.getConfig().getExtParam("MapperInterfaceNaming"); + String doMapperClazzName = mapperInterfaceNaming.replace("[EntityName]", table.getJavaName()); + doMapper.setClassName(doMapperClazzName); + + String xmlMapperNaming = ConfigUtil.getConfig().getExtParam("XMLMapperNaming"); + String xmlFileName = xmlMapperNaming.replace("[EntityName]", table.getJavaName()); + doMapper.setXmlFileName(xmlFileName); + doMapper.setPackageName(ConfigUtil.getCurrentDb().getGenPackage("mapper")); doMapper.setXmlPackageName(ConfigUtil.getCurrentDb().getGenPackage("mapper.xml")); doMapper.setDesc(cfTable.getRemark()); @@ -792,7 +797,10 @@ public class DalgenLoader extends AbstractDalgenLoader { */ private DO preDo(Table table, List cfColumns) { DO doClass = new DO(); - doClass.setClassName(table.getJavaName() + "DO"); + String entityNaming = ConfigUtil.getConfig().getExtParam("EntityNaming"); + String doClazzName = entityNaming.replace("[EntityName]", table.getJavaName()); + doClass.setClassName(doClazzName); + doClass.setPackageName(ConfigUtil.getCurrentDb().getGenPackage("dataobject")); doClass.setDesc(table.getRemark()); doClass.setTableName(table.getSqlName()); diff --git a/src/main/java/cn/dalgen/mybatis/gen/model/Config.java b/src/main/java/cn/dalgen/mybatis/gen/model/Config.java index 54a46c151bfa8338245918129f290f4d2e5ef053..97631e70cce28e6de9eb82f9a1483e555b59537f 100755 --- a/src/main/java/cn/dalgen/mybatis/gen/model/Config.java +++ b/src/main/java/cn/dalgen/mybatis/gen/model/Config.java @@ -45,7 +45,12 @@ public class Config { private Map extParamMap = Maps.newHashMap(); { extParamMap.put("Repository","true"); - extParamMap.put("DoMapperName","DO"); + extParamMap.put("DoMapperName","DO"); // 废弃 by Vr80s 2019-05-23 + // 生成文件命名规则参数 by Vr80s 2019-05-23 + extParamMap.put("EntityNaming","[EntityName]DO"); + extParamMap.put("XMLMapperNaming","[EntityName]DOMapper"); + extParamMap.put("MapperInterfaceNaming","[EntityName]DOMapper"); + extParamMap.put("DaoNaming","[EntityName]DAO"); } /** * 软删除字段 diff --git a/src/main/java/cn/dalgen/mybatis/gen/model/java/DOMapper.java b/src/main/java/cn/dalgen/mybatis/gen/model/java/DOMapper.java index 30ca5b570342fa24c64f7c99e2f31fe3b6034cc3..1a6ab19168179f2606c71cf59345822eccd696cc 100755 --- a/src/main/java/cn/dalgen/mybatis/gen/model/java/DOMapper.java +++ b/src/main/java/cn/dalgen/mybatis/gen/model/java/DOMapper.java @@ -11,8 +11,17 @@ import java.util.List; */ public class DOMapper extends Base { + private String xmlFileName; private String xmlPackageName; + public String getXmlFileName() { + return StringUtils.isBlank(xmlFileName) ? getClassName() : xmlFileName; + } + + public void setXmlFileName(String xmlFileName) { + this.xmlFileName = xmlFileName; + } + public String getXmlPackageName() { return StringUtils.isBlank(xmlPackageName)?getPackageName():xmlPackageName; } diff --git a/src/main/resources/dalgen/config/config.xml b/src/main/resources/dalgen/config/config.xml index ee85be4250edc77f4a5276a506d1bca67590ed0c..7e9c359f00bdcfe99100eecbaa141030cc5100e6 100755 --- a/src/main/resources/dalgen/config/config.xml +++ b/src/main/resources/dalgen/config/config.xml @@ -87,8 +87,20 @@ - + + + + + + + + + + + diff --git a/src/main/resources/dalgen/templates/dalgen/XMLMapper.xml.ftl b/src/main/resources/dalgen/templates/dalgen/XMLMapper.xml.ftl index 4acdc18e54a3d7dd1abd24866ba5bd18ad62ebde..f2206b5776828af71ebb933fbeae5d9331a39908 100755 --- a/src/main/resources/dalgen/templates/dalgen/XMLMapper.xml.ftl +++ b/src/main/resources/dalgen/templates/dalgen/XMLMapper.xml.ftl @@ -3,7 +3,7 @@ <#import "../lib/function.ftl" as fun/> <#import "../lib/macro.ftl" as m> <#list dalgen.xmlMappers as xmlMapper> -<@pp.changeOutputFile name = "/main/resources/${xmlMapper.doMapper.xmlPath}/${xmlMapper.doMapper.className}.xml" /> +<@pp.changeOutputFile name = "/main/resources/${xmlMapper.doMapper.xmlPath}/${xmlMapper.doMapper.xmlFileName}.xml" />