MyBatis-Plusコードジェネレータ使用
7065 ワード
公式ドキュメント
https://mybatis.plus/guide/generator.html#%E4%BD%BF%E7%94%A8%E6%95%99%E7%A8%8B
導入依存
コードジェネレータクラス
プロジェクトの構成と起動
https://mybatis.plus/guide/generator.html#%E4%BD%BF%E7%94%A8%E6%95%99%E7%A8%8B
導入依存
mysql
mysql-connector-java
com.baomidou
mybatis-plus-boot-starter
3.2.0
com.baomidou
mybatis-plus-generator
3.2.0
org.freemarker
freemarker
2.3.28
コードジェネレータクラス
package com.example.vblog.common.generator;
import com.baomidou.mybatisplus.core.exceptions.MybatisPlusException;
import com.baomidou.mybatisplus.core.toolkit.StringPool;
import com.baomidou.mybatisplus.core.toolkit.StringUtils;
import com.baomidou.mybatisplus.generator.AutoGenerator;
import com.baomidou.mybatisplus.generator.InjectionConfig;
import com.baomidou.mybatisplus.generator.config.*;
import com.baomidou.mybatisplus.generator.config.po.TableInfo;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;
import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;
/**
*
* :user_info
* :user_id
* : run , , ,
*/
public class CodeGenerator {
/**
*
*
*
*/
public static String scanner(String tip) {
Scanner scanner = new Scanner(System.in);
StringBuilder help = new StringBuilder();
help.append(" " + tip + ":");
System.out.println(help.toString());
if (scanner.hasNext()) {
String ipt = scanner.next();
if (StringUtils.isNotEmpty(ipt)) {
return ipt;
}
}
throw new MybatisPlusException(" " + tip + "!");
}
public static void main(String[] args) {
//
AutoGenerator mpg = new AutoGenerator();
//
GlobalConfig gc = new GlobalConfig();
String projectPath = System.getProperty("user.dir");
gc.setOutputDir(projectPath + "/src/main/java");
gc.setAuthor("xp");
gc.setOpen(false);
// gc.setSwagger2(true); Swagger2
mpg.setGlobalConfig(gc);
//
DataSourceConfig dsc = new DataSourceConfig();
dsc.setUrl("jdbc:mysql://localhost:3306/vblog?useUnicode=true&useSSL=false&characterEncoding=utf8&serverTimezone=Asia/Shanghai");
// dsc.setSchemaName("public");
dsc.setDriverName("com.mysql.cj.jdbc.Driver");
dsc.setUsername("root");
dsc.setPassword("root");
mpg.setDataSource(dsc);
// : com.example.vblog
PackageConfig pc = new PackageConfig();
// xml :resources/mapper/vblog/**Mapper.xml
pc.setModuleName(scanner("vblog"));
// xml :resources/mapper/null/**Mapper.xml, null
// pc.setModuleName(null);
pc.setParent("com.example");
mpg.setPackageInfo(pc);
//
InjectionConfig cfg = new InjectionConfig() {
@Override
public void initMap() {
// to do nothing
}
};
// freemarker
String templatePath = "/templates/mapper.xml.ftl";
// velocity
// String templatePath = "/templates/mapper.xml.vm";
//
List focList = new ArrayList<>();
//
focList.add(new FileOutConfig(templatePath) {
@Override
public String outputFile(TableInfo tableInfo) {
// , Entity 、 xml !!
return projectPath + "/src/main/resources/mapper/" + pc.getModuleName()
+ "/" + tableInfo.getEntityName() + "Mapper" + StringPool.DOT_XML;
}
});
/*
cfg.setFileCreate(new IFileCreate() {
@Override
public boolean isCreate(ConfigBuilder configBuilder, FileType fileType, String filePath) {
//
checkDir(" , ");
if (fileType == FileType.MAPPER) {
// mapper , false
return !new File(filePath).exists();
}
//
return true;
}
});
*/
cfg.setFileOutConfigList(focList);
mpg.setCfg(cfg);
//
TemplateConfig templateConfig = new TemplateConfig();
//
// , .ftl/.vm,
// templateConfig.setEntity("templates/entity2.java");
// templateConfig.setService();
// templateConfig.setController();
templateConfig.setXml(null);
mpg.setTemplate(templateConfig);
//
StrategyConfig strategy = new StrategyConfig();
strategy.setNaming(NamingStrategy.underline_to_camel);
strategy.setColumnNaming(NamingStrategy.underline_to_camel);
// strategy.setSuperEntityClass(" , !");
strategy.setEntityLombokModel(true);
strategy.setRestControllerStyle(true);
//
// strategy.setSuperControllerClass(" , !");
// strategy.setSuperServiceClass(" , !");
// strategy.setSuperServiceImplClass(" , !");
//
// strategy.setSuperEntityColumns("id");
strategy.setInclude(scanner(" , ").split(","));
strategy.setControllerMappingHyphenStyle(true);
strategy.setTablePrefix(pc.getModuleName() + "_");
mpg.setStrategy(strategy);
mpg.setTemplateEngine(new FreemarkerTemplateEngine());
mpg.execute();
}
}
プロジェクトの構成と起動
server.port=8088
spring.datasource.driver-class-name=com.mysql.cj.jdbc.Driver
spring.datasource.url=jdbc:mysql://localhost:3306/vblog?useUnicode=true&useSSL=false&characterEncoding=utf8&serverTimezone=Asia/Shanghai
spring.datasource.username=root
spring.datasource.password=root
mybatis-plus.mapper-locations=classpath:/mapper/vblog/**.xml
// MapperScan
@SpringBootApplication
@MapperScan("com.example.vblog.mapper")
public class VblogApplication {
public static void main(String[] args) {
SpringApplication.run(VblogApplication.class, args);
System.out.println(" ");
}
}