Mybatis Plus是一款增强版的mybatis，功能强大，简化开发，大大提高了开发效率，虽然mbp支持小众的国产达梦数据库，但是在代码生成这块，还是存在一些bug。
目前发现的Bug
public class DMQuery extends AbstractDbQuery {
// 部分源码
...
public String tableName() {
return "TABLE_NAME";
}
...
}
调试源码，会看到这个tableName()方法，顾名思义，该方法返回一个表名叫"TABLE_NAME"，因为没有带上别名"T1.TABLE_NAME"，所以会导致生成的SQL语句中的"TABLE_NAME"有歧义。
解决方案
一、创建 MyDMQuery 类继承 DMQuery 类，重写 tableName() 方法，新增一个 flag 变量来记录调用次数：
- 源码会出现“有歧义的列名[TABLE_NAME] ”，所以当第一次调用时，返回一个带有别名的表名"T1.TABLE_NAME"
- 但是因为第一步修改了带有别名的表名，会出现“列无效”，所以之后的调用直接返回表名"TABLE_NAME"
public class MyDMQuery extends DMQuery {
private int flag = 1;
@Override
public String tableName() {
if (flag == 1) {
flag++;
// 有歧义的列名[TABLE_NAME] 解决方案
return "T1.TABLE_NAME";
} else {
// 列无效 解决方案
return "TABLE_NAME";
}
}
}
二、重写数据源配置类DataSourceConfig中的getDbQuery()方法，应用我们自定义的MyDMQuery类
public class MyDataSourceConfig extends DataSourceConfig {
@Override
public IDbQuery getDbQuery() {
switch(this.getDbType()) {
case ORACLE:
setDbQuery(new OracleQuery());
break;
case SQL_SERVER:
setDbQuery(new SqlServerQuery());
break;
case POSTGRE_SQL:
setDbQuery(new PostgreSqlQuery());
break;
case DB2:
setDbQuery(new DB2Query());
break;
case MARIADB:
setDbQuery(new MariadbQuery());
break;
case H2:
setDbQuery(new H2Query());
break;
case SQLITE:
setDbQuery(new SqliteQuery());
break;
case DM:
// 重新指定自定义的DMQuery
setDbQuery(new MyDMQuery());
break;
case KINGBASE_ES:
setDbQuery(new KingbaseESQuery());
break;
default:
setDbQuery(new MySqlQuery());
}
return super.getDbQuery();
}
}
到此，Bug已解决，只需要在代码生成工具中，应用自定义的数据源配置类MyDataSourceConfig即可。
三、自动生成代码工具
public class DmCodeGeneration {
//main函数
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
System.out.print("请输入作者名称：");
String author = sc.nextLine();
// 模块名
System.out.print("请输入模块名：");
String moduleName = sc.nextLine();
// 表名
System.out.println("请输入表名（多张表以空格隔开）:");
String tableName = sc.nextLine();
String[] tables = tableName.split(" ");
// 包名父路径
String parentPackage = "com.ruoyi";
// 输出目标项目的路径
String oPath = System.getProperty("user.dir") + "/ruoyi-system";
System.out.println("目标项目路径：" + oPath);
AutoGenerator autoGenerator = new AutoGenerator();
//全局配置
GlobalConfig gc = new GlobalConfig();
//生成文件输出根目录
gc.setOutputDir(oPath + "/src/main/java");
//生成完成后不弹出文件框
gc.setOpen(false);
//文件覆盖
gc.setFileOverride(false);
// 不需要ActiveRecord特性的请改为false
gc.setActiveRecord(false);
// XML 二级缓存
gc.setEnableCache(false);
// XML ResultMap
gc.setBaseResultMap(true);
// XML columList
gc.setBaseColumnList(false);
// 作者
gc.setAuthor(author);
gc.setSwagger2(false);
// 自定义文件命名，注意 %s 会自动填充表实体属性！
gc.setControllerName("%sController");
gc.setServiceName("%sService");
gc.setServiceImplName("%sServiceImpl");
gc.setMapperName("%sMapper");
gc.setXmlName("%sMapper");
// gc.setEntityName("%s");
gc.setIdType(IdType.AUTO);
autoGenerator.setGlobalConfig(gc);
// 数据源配置：此处应用自定义的数据源配置类，不用原始的DataSourceConfig类
DataSourceConfig dsc = new MyDataSourceConfig();
//设置数据库类型为达梦数据库
dsc.setDbType(DbType.DM);
dsc.setDriverName("dm.jdbc.driver.DmDriver");
//用户名
dsc.setUsername("填写你的数据库用户名");
//密码
dsc.setPassword("填写你的数据库密码");
//指定数据库
dsc.setUrl("jdbc:dm://127.0.0.1:5236/TEST_DB");
autoGenerator.setDataSource(dsc);
// 策略配置
StrategyConfig strategy = new StrategyConfig();
// 表名生成策略
strategy.setNaming(NamingStrategy.underline_to_camel);
// 需要生成的表
strategy.setInclude(tables);
// 自定义实体类父类
strategy.setSuperEntityClass("com.ruoyi.common.core.domain.CommonEntity");
strategy.setSuperServiceClass(null);
strategy.setSuperServiceImplClass(null);
strategy.setSuperMapperClass(null);
strategy.setEntityLombokModel(true);
// 自定义 mapper 父类
strategy.setSuperMapperClass("com.baomidou.mybatisplus.core.mapper.BaseMapper");
// 自定义 service 父类
// strategy.setSuperServiceImplClass("com.mayi.test.base.BaseService");
//逻辑删除属性名称
strategy.setLogicDeleteFieldName("del_flag");
//去除表前缀
// strategy.setTablePrefix("pnds_");
//去除字段前缀
// strategy.setFieldPrefix("");
List<TableFill> tableFillList= new ArrayList<>();
tableFillList.add(new TableFill("update_time", FieldFill.INSERT_UPDATE));
tableFillList.add(new TableFill("create_time", FieldFill.INSERT));
strategy.setTableFillList(tableFillList);
autoGenerator.setStrategy(strategy);
// 包配置
PackageConfig pc = new PackageConfig();
//父包路径
pc.setParent(parentPackage);
pc.setModuleName(moduleName);
// pc.setController("controller");
pc.setService("service");
pc.setServiceImpl("service.impl");
pc.setMapper("mapper");
pc.setEntity("domain");
// pc.setXml("mapper.test");
autoGenerator.setPackageInfo(pc);
// 自定义文件输出配置
VelocityTemplateEngine velocityTemplateEngine = new VelocityTemplateEngine();
InjectionConfig cfg = new InjectionConfig() {
@Override
public void initMap() {}
};
// 自定义输出配置
List<FileOutConfig> focList = new ArrayList<>();
// 自定义配置会被优先输出
focList.add(new FileOutConfig("/template/mapper.xml.vm") {
@Override
public String outputFile(TableInfo tableInfo) {
// 自定义输出文件名 ， 如果你 Entity 设置了前后缀、此处注意 xml 的名称会跟着发生变化！！
return oPath + "/src/main/resources/mapper/" + pc.getModuleName()
+ "/" + tableInfo.getEntityName() + "Mapper" + StringPool.DOT_XML;
}
});
cfg.setFileOutConfigList(focList);
autoGenerator.setCfg(cfg);
TemplateConfig templateConfig = new TemplateConfig();
//控制是否按默认的来生成文件 设置为null则不生成默认的
templateConfig.setXml(null);
templateConfig.setController(null);
autoGenerator.setTemplate(templateConfig);
// 执行生成
autoGenerator.execute();
System.out.println("代码生成完毕！");
}
}
网友评论