1、目录结构
![](https://i-blog.csdnimg.cn/blog_migrate/2ce26fa13fc58099f741be6856ca2509.png)
2、Maven依赖引入
<dependency>
<groupId>com.baomidou</groupId>
<artifactId>mybatis-plus-generator</artifactId>
<version>3.3.2</version>
</dependency>
<dependency>
<groupId>org.freemarker</groupId>
<artifactId>freemarker</artifactId>
<version>2.3.30</version>
</dependency>
3、示例代码
package com.supermap.demo.generator;
import com.baomidou.mybatisplus.annotation.DbType;
import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.generator.AutoGenerator;
import com.baomidou.mybatisplus.generator.config.DataSourceConfig;
import com.baomidou.mybatisplus.generator.config.GlobalConfig;
import com.baomidou.mybatisplus.generator.config.PackageConfig;
import com.baomidou.mybatisplus.generator.config.StrategyConfig;
import com.baomidou.mybatisplus.generator.config.rules.DateType;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;
/**
* @author wenchenhui
* @ClassName: MyBatisPlusGenerator
* @Description:
* @date: 2023/2/5 16:21
*/
public class MyBatisPlusGenerator {
static String url = "jdbc:postgresql://127.0.0.1:5432/system?currentSchema=public";
static String username = "postgres";
static String password = "salis";
static String driverClassName = "org.postgresql.Driver";
public static void main(String[] args) {
//1. 全局配置
GlobalConfig config = new GlobalConfig();
// 作者
config.setAuthor("wenchenhui")
// 生成路径,最好使用绝对路径,window路径是不一样的
// 文件存放位置,自定义
.setOutputDir("/Users/src/main/java")
// 文件覆盖
.setFileOverride(true)
// 主键策略
.setIdType(IdType.AUTO)
.setDateType(DateType.ONLY_DATE)
// 设置生成的service接口的名字的首字母是否为I,默认Service是以I开头的
.setServiceName("%sService")
//实体类结尾名称
.setEntityName("%sDO")
//生成基本的resultMap
.setBaseResultMap(true)
//不使用AR模式
.setActiveRecord(false)
//生成基本的SQL片段
.setBaseColumnList(true);
//2. 数据源配置
DataSourceConfig dsConfig = new DataSourceConfig();
// 设置数据库类型
dsConfig.setDbType(DbType.POSTGRE_SQL)
.setDriverName(driverClassName)
.setUrl(url)
.setUsername(username)
.setPassword(password);
//3. 策略配置globalConfiguration中
StrategyConfig stConfig = new StrategyConfig();
//全局大写命名
stConfig.setCapitalMode(true)
// 数据库表映射到实体的命名策略
.setNaming(NamingStrategy.underline_to_camel)
// 使用lombok,前面我们已经添加了lombok插件
// lombok添加:https://www.jianshu.com/p/5c6f104bbcac
.setEntityLombokModel(true)
//使用restcontroller注解
.setRestControllerStyle(true)
// 生成的表, 支持多表一起生成,以数组形式填写
// 修改!
.setInclude("sys_user", "sys_role", "sys_permissions");
//4. 包名策略配置
PackageConfig pkConfig = new PackageConfig();
pkConfig.setParent("com.supermap.sys")
.setMapper("mapper")
.setService("service")
.setController("controller")
.setEntity("model")
.setXml("mapper");
//5. 整合配置
AutoGenerator ag = new AutoGenerator();
ag.setGlobalConfig(config)
.setDataSource(dsConfig)
.setStrategy(stConfig)
.setPackageInfo(pkConfig)
.setTemplateEngine(new FreemarkerTemplateEngine());
//6. 执行操作
ag.execute();
System.out.println("======= 代码自动生成完毕! ========");
}
}
参考文档:
代码生成器: 代码生成器(新) | MyBatis-Plus
FreeMarker在线手册:什么是 FreeMarker? - FreeMarker 中文官方参考手册