简介
Mybatis 相信大家都用过，是一款优秀的持久层框架，支持支持定制化Sql，存储过程和高级映射，大大提高了开发过程中与数据库交互的效率，而且非常的灵活，而Mybatis Plus 是Mybatis更高层级的封装，我们经常使用的增删改查她已经帮我们集成好了，我们直接使用即可，比Mybaits 更智能，开发效率更高。本文章Mybatis 版本为3.0.6，话不多说，一个字：干
1.Maven 配置文件
com.baomidou
mybatis-plus-boot-starter
3.0.6
org.apache.tomcat
tomcat-jdbc
mysql
mysql-connector-java
runtime
org.springframework.boot
spring-boot-starter-freemarker
io.springfox
springfox-swagger2
2.8.0
org.mapstruct
mapstruct
io.springfox
springfox-swagger-ui
2.8.0
2.Mybatis配置文件
resources目录下创建config文件夹，然后再config文件夹下创建json类型的配置文件
{
"author": "gaoleijie",
"url": "jdbc:mysql://127.0.0.1:3306/security_oauth?serverTimezone=GMT%2B8&useUnicode=true&characterEncoding=utf-8",
"userName": "root",
"password": "root",
"driverName": "com.mysql.cj.jdbc.Driver",
"rootPath": "com.glj.order",
"mapping": [
{
"moduleName": "base",
"table": [
"oauth_client_details",
"sys_permission",
"sys_role",
"sys_role_permission",
"sys_user"
]
}
]
}
3.生成代码类
package com.glj.order;
import com.baomidou.mybatisplus.annotation.DbType;
import com.baomidou.mybatisplus.core.toolkit.StringPool;
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 com.fasterxml.jackson.databind.ObjectMapper;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.util.ResourceUtils;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
public class CodeGenerator {
private static final Logger logger = LoggerFactory.getLogger(CodeGenerator.class);
private final static String AUTHOR = "author";
private final static String URL = "url";
private final static String TABLE = "table";
private final static String PASSWORD = "password";
private final static String USER_NAME = "userName";
private final static String MODULE_NAME = "moduleName";
private final static String DRIVER_NAME = "driverName";
private final static String MODULE_ROOT_PATH = "rootPath";
private final static String TABLE_MODULE_MAPPING = "mapping";
public static void main(String[] args) {
//代码自动生成
Map configData;
try {
ObjectMapper mapper = new ObjectMapper();
configData = mapper.readValue(ResourceUtils.
getFile("classpath:config/generateCfg.json"), Map.class);
} catch (Exception e) {
logger.debug("Parse the config failure , reason : {}", e);
throw new RuntimeException("Parse config generate file failure");
}
List> listMapping = (List>) configData.get(TABLE_MODULE_MAPPING);
for (Map obj : listMapping) {
String moduleName = obj.get(MODULE_NAME).toString();
List tables = (List) obj.get(TABLE);
String[] arr = new String[tables.size()];
tables.toArray(arr);
createCode(configData, moduleName, arr);
}
}
/**
* 生成代码
*
* @param configData
* @param moduleName
* @param tables
*/
private static void createCode(Map configData, String moduleName, String[] tables) {
// 代码生成器
AutoGenerator mpg = new AutoGenerator();
String projectPath = CodeGenerator.class.getResource("").getPath().split("target")[0];
String author = configData.get(AUTHOR).toString();
//全局配置
GlobalConfig gc = buildGlobalConfig(author, projectPath);
mpg.setGlobalConfig(gc);
//数据库配置
String url = configData.get(URL).toString();
String password = configData.get(PASSWORD).toString();
String userName = configData.get(USER_NAME).toString();
String driverName = configData.get(DRIVER_NAME).toString();
DataSourceConfig ds = buildDataSource(url, driverName, userName, password);
mpg.setDataSource(ds);
mpg.setTemplate(new TemplateConfig().setXml(null));
mpg.setTemplateEngine(new FreemarkerTemplateEngine());
String moduleRootPath = configData.get(MODULE_ROOT_PATH).toString();
InjectionConfig ic = buildInjectionConfig(projectPath, moduleName);
mpg.setCfg(ic);
//包配置
PackageConfig pc = buildPageConfig(moduleName, moduleRootPath);
mpg.setPackageInfo(pc);
//策略配置
StrategyConfig sc = buildStrategy(tables);
mpg.setStrategy(sc);
mpg.execute();
}
/**
* 全局配置
*
* @param author
* @param projectPath
* @return
*/
private static GlobalConfig buildGlobalConfig(String author, String projectPath) {
GlobalConfig gc = new GlobalConfig();
gc.setOutputDir(projectPath + "/src/main/java");
gc.setAuthor(author);
gc.setSwagger2(Boolean.TRUE);
gc.setActiveRecord(Boolean.TRUE);
gc.setEntityName("%sPo");
gc.setMapperName("%sMapper");
gc.setXmlName("%sMapper");
// gc.setServiceName("%sService");
// gc.setServiceImplName("%sServiceImpl");
// gc.setControllerName("%sController");
return gc;
}
/**
* @param url
* @param driverName
* @param userName
* @param password
* @return
*/
private static DataSourceConfig buildDataSource(String url, String driverName, String userName, String password) {
DataSourceConfig dsc = new DataSourceConfig();
dsc.setUrl(url);
dsc.setUsername(userName);
dsc.setPassword(password);
dsc.setDbType(DbType.MYSQL);
dsc.setDriverName(driverName);
return dsc;
}
/**
* 自定义配置
*
* @param projectPath
* @param moduleName
* @return
*/
private static InjectionConfig buildInjectionConfig(String projectPath, String moduleName) {
InjectionConfig cfg = new InjectionConfig() {
@Override
public void initMap() {
// to do nothing
}
};
List focList = new ArrayList<>();
focList.add(new FileOutConfig("/templates/mapper.xml.ftl") {
@Override
public String outputFile(TableInfo tableInfo) {
// 自定义输入文件名称
return projectPath + "/src/main/resources/mapper/" + moduleName
+ "/" + tableInfo.getEntityName() + "Mapper" + StringPool.DOT_XML;
}
});
cfg.setFileOutConfigList(focList);
return cfg;
}
/**
* 包配置
*
* @param moduleName
* @param moduleRootPath
* @return
*/
private static PackageConfig buildPageConfig(String moduleName, String moduleRootPath) {
PackageConfig pc = new PackageConfig();
pc.setModuleName(moduleName);
pc.setParent(moduleRootPath);
return pc;
}
/**
* 配置策略
*
* @param table
* @return
*/
private static StrategyConfig buildStrategy(String... table) {
// 策略配置
StrategyConfig strategy = new StrategyConfig();
strategy.setInclude(table);
strategy.setEntityLombokModel(Boolean.TRUE);
strategy.setEntityBuilderModel(Boolean.TRUE);
strategy.setRestControllerStyle(Boolean.TRUE);
strategy.setEntityColumnConstant(Boolean.TRUE);
strategy.setNaming(NamingStrategy.underline_to_camel);
strategy.setColumnNaming(NamingStrategy.underline_to_camel);
return strategy;
}
}