package com.utils;
import com.baomidou.mybatisplus.annotation.DbType;
import com.baomidou.mybatisplus.annotation.IdType;
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 java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;
public class MyBatisPlusGenerator {
public static void main(String[] args) throws SQLException {
String projectPath = System.getProperty("user.dir");
System.out.println(projectPath);
//1. 全局配置
GlobalConfig config = new GlobalConfig();
config.setActiveRecord(true) // 是否支持AR模式
.setAuthor("XX") // 作者
//.setOutputDir("D:\\workspace_mp\\mp03\\src\\main\\java") // 生成路径
.setOutputDir(projectPath+"/project/src/main/java") // 生成路径
.setFileOverride(true) // 文件覆盖
.setIdType(IdType.AUTO) // 主键策略
.setServiceName("%sService") // 设置生成的service接口的名字的首字母是否为I
.setBaseResultMap(true)//生成基本的resultMap
.setBaseColumnList(true)
// .setSwagger2(true)
;//生成基本的SQL片段
//2. 数据源配置
DataSourceConfig dsConfig = new DataSourceConfig();
dsConfig.setDbType(DbType.MYSQL) // 设置数据库类型
.setDriverName("com.mysql.jdbc.Driver")
.setUrl("数据库地址/库")
.setUsername("root")
.setPassword("123456");
//3. 策略配置globalConfiguration中
StrategyConfig stConfig = new StrategyConfig();
stConfig.setCapitalMode(true) //全局大写命名
.setNaming(NamingStrategy.underline_to_camel) // 数据库表映射到实体的命名策略
// .setTablePrefix("grading_")
.setInclude(
"table1",
"table2"
); // 生成的表
//4. 包名策略配置
PackageConfig pkConfig = new PackageConfig();
pkConfig.setParent("com.mybatis.unit")
.setMapper("mapper")//dao
.setService("service")
// .setController("controller")//controller
.setEntity("model")
.setXml("mapper");//mapper.xml
InjectionConfig injectionConfig = new InjectionConfig() {
@Override
public void initMap() {
// Map<String, Object> map = new HashMap<String, Object>();
// map.put("abc", this.getConfig().getGlobalConfig().getAuthor() + "-mp");
// this.setMap(map);
}
};
//自定义文件输出位置(非必须)
List<FileOutConfig> fileOutList = new ArrayList();
fileOutList.add(new FileOutConfig("/templates/mapper.xml.vm") {
@Override
public String outputFile(TableInfo tableInfo) {
return projectPath + "/project/src/main/resources/mybatis/" + tableInfo.getEntityName() + "Mapper.xml";
}
});
fileOutList.add(new FileOutConfig("/templates/entity.java.vm") {
@Override
public String outputFile(TableInfo tableInfo) {
return projectPath + "/project/src/main/java/com/mybatis/api/model/po/" + tableInfo.getEntityName() + ".java";
}
});
injectionConfig.setFileOutConfigList(fileOutList);
//5. 整合配置
AutoGenerator ag = new AutoGenerator();
ag.setGlobalConfig(config)
.setDataSource(dsConfig)
.setStrategy(stConfig)
.setPackageInfo(pkConfig)
.setCfg(injectionConfig)
.setTemplate(// 关闭默认 xml 生成,调整生成 至 根目录
new TemplateConfig().setXml(null).setController(null).setEntity(null)
);
//6. 执行
ag.execute();
}
}
根据表关系创建实体,mapper,service等
最新推荐文章于 2023-02-16 21:54:53 发布