前言
- 代码生成器顾名思义就是为我们生成一些代码,省去了我们一些时间。
MyBatis-Plus 的代码生成器可以生成 Entity、Mapper、Mapper XML、Service、Controller 模块代码。
须知
MyBatis-Plus 从
3.0.3
之后移除了代码生成器与模板引擎的默认依赖,需要手动添加相关依赖,才能实现代码生成器功能。玩熟 MyBatis-Plus 代码生成器
1.新建 MyBatis-Plus 代码生成器项目
使用 Spring 脚手架创建 SpringBoot 项目,如果不太熟悉 IDEA 快速生成 SpringBoot 项目,可以先看下面一篇博客,几分钟就搞定。
SpringBoot 快速入门2.添加代码生成器依赖
<dependency>
<groupId>com.baomidou</groupId>
<artifactId>mybatis-plus-generator</artifactId>
<version>3.2.0</version>
</dependency>
3.添加模板引擎依赖
- MyBatis-Plus 支持 Velocity(默认)、Freemarker、Beetl,可以选择自己熟悉的模板引擎,如果都不满足您的要求,可以采用自定义模板引擎。
pom.xml
<!-- https://mvnrepository.com/artifact/org.apache.velocity/velocity-engine-core -->
<dependency>
<groupId>org.apache.velocity</groupId>
<artifactId>velocity-engine-core</artifactId>
<version>2.2</version>
</dependency>
<!-- https://mvnrepository.com/artifact/org.freemarker/freemarker -->
<dependency>
<groupId>org.freemarker</groupId>
<artifactId>freemarker</artifactId>
<version>2.3.30</version>
</dependency>
<!-- https://mvnrepository.com/artifact/com.ibeetl/beetl -->
<dependency>
<groupId>com.ibeetl</groupId>
<artifactId>beetl</artifactId>
<version>3.1.3.RELEASE</version>
</dependency>
如果我们选择了非默认引擎,需要在 AutoGenerator 中 设置模板引擎。
AutoGenerator generator = new AutoGenerator();
// set freemarker engine
generator.setTemplateEngine(new FreemarkerTemplateEngine());
// set beetl engine
generator.setTemplateEngine(new BeetlTemplateEngine());
// set custom engine (reference class is your custom engine class)
generator.setTemplateEngine(new CustomTemplateEngine());
// other config
...
4.完整pom.xml
代码生成器需要依赖数据库表,所以也需要 MySQL 驱动包 ``` <?xml version=”1.0” encoding=”UTF-8”?> <project xmlns=”http://maven.apache.org/POM/4.0.0“ xmlns:xsi=”http://www.w3.org/2001/XMLSchema-instance“
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 https://maven.apache.org/xsd/maven-4.0.0.xsd">
4.0.0 <groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-parent</artifactId>
<version>2.5.4</version>
<relativePath/> <!-- lookup parent from repository -->
com.mybatis.generator generator 0.0.1-SNAPSHOT generator Demo project for Spring Boot 1.8 org.apache.velocity velocity-engine-core 2.0 org.springframework.boot spring-boot-test org.springframework.boot spring-boot-starter-web com.baomidou mybatis-plus-boot-starter 3.2.0 com.baomidou mybatis-plus-generator 3.2.0 org.freemarker freemarker mysql mysql-connector-java org.projectlombok lombok org.junit.jupiter junit-jupiter test
<a name="KH7NL"></a>
## 5.全局配置
1. `user.dir`获取到你当前工程的 src 目录路径<br />![image.png](https://cdn.nlark.com/yuque/0/2021/png/21469639/1631089843789-aa087d37-3818-467b-8103-c82639456c96.png#height=252&id=bXZUK&margin=%5Bobject%20Object%5D&name=image.png&originHeight=504&originWidth=1366&originalType=binary&ratio=1&size=540546&status=done&style=none&width=683)
<a name="cW5Vb"></a>
## 6.数据库信息配置
![image.png](https://cdn.nlark.com/yuque/0/2021/png/21469639/1631089821996-2e4ffcd7-5646-4284-9fc4-a6809aa1a1f6.png#height=168&id=nCGaK&margin=%5Bobject%20Object%5D&name=image.png&originHeight=335&originWidth=1281&originalType=binary&ratio=1&size=329568&status=done&style=none&width=640.5)
<a name="uKiIn"></a>
## 7.包配置
![image.png](https://cdn.nlark.com/yuque/0/2021/png/21469639/1631089809080-f765cbfb-2bfa-4aa6-8633-5320f5a666bc.png#height=182&id=pkUYt&margin=%5Bobject%20Object%5D&name=image.png&originHeight=363&originWidth=1315&originalType=binary&ratio=1&size=337103&status=done&style=none&width=657.5)
<a name="dXHhd"></a>
## 8.策略配置
1. 配置根据哪张表生成代码<br />![image.png](https://cdn.nlark.com/yuque/0/2021/png/21469639/1631089786337-280a91c5-4229-422f-8f43-ef8f3bbd1dae.png#height=217&id=ku5o3&margin=%5Bobject%20Object%5D&name=image.png&originHeight=433&originWidth=1370&originalType=binary&ratio=1&size=477477&status=done&style=none&width=685)
<a name="6JJIW"></a>
## 9.完整 MyBatis-Plus 代码生成器代码
1. MyBatis-Plus 代码生成器所有配置都可以使用 Java Config 完成,不需要单独在 XML 配置。
```java
package com.mybatis.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 org.junit.jupiter.api.Test;
import org.springframework.boot.test.context.SpringBootTest;
//@SpringBootTest
class GeneratorApplicationTests {
// @Test
public static void main(String[] args) {
// 1、创建代码生成器
AutoGenerator mpg = new AutoGenerator();
// 2、全局配置
GlobalConfig gc = new GlobalConfig();
String projectPath = System.getProperty("user.dir");
gc.setOutputDir(projectPath + "/src/main/java");
gc.setAuthor("zwq");
gc.setOpen(false); //生成后是否打开资源管理器
gc.setFileOverride(false); //重新生成时文件是否覆盖
gc.setServiceName("%sService"); //去掉Service接口的首字母I
gc.setIdType(IdType.ID_WORKER_STR); //主键策略
gc.setDateType(DateType.ONLY_DATE);//定义生成的实体类中日期类型
gc.setSwagger2(false);//开启Swagger2模式
mpg.setGlobalConfig(gc);
// 3、数据源配置
DataSourceConfig dsc = new DataSourceConfig();
dsc.setUrl("jdbc:mysql://81.71.13.91:3306/usercenter?serverTimezone=GMT%2B8");
dsc.setDriverName("com.mysql.cj.jdbc.Driver");
dsc.setUsername("root");
dsc.setPassword("123456");
dsc.setDbType(DbType.MYSQL);
mpg.setDataSource(dsc);
// 4、包配置
PackageConfig pc = new PackageConfig();
pc.setModuleName(null); //模块名
pc.setParent("com.mybatis.generator");
pc.setController("controller");
pc.setEntity("entity");
pc.setService("service");
pc.setMapper("mapper");
mpg.setPackageInfo(pc);
// 5、策略配置
StrategyConfig strategy = new StrategyConfig();
strategy.setInclude("sys_user");//对那一张表生成代码
strategy.setNaming(NamingStrategy.underline_to_camel);//数据库表映射到实体的命名策略
strategy.setTablePrefix(pc.getModuleName() + "_"); //生成实体时去掉表前缀
strategy.setColumnNaming(NamingStrategy.underline_to_camel);//数据库表字段映射到实体的命名策略
strategy.setEntityLombokModel(true); // lombok 模型 @Accessors(chain = true) setter链式操作
strategy.setRestControllerStyle(true); //restful api风格控制器
strategy.setControllerMappingHyphenStyle(true); //url中驼峰转连字符
mpg.setStrategy(strategy);
// 6、执行
mpg.execute();
}
}
总结
- 代码生成器的全部代码都在这篇博客当中了,没有那块代码是缺漏,相信大家看完这篇博客之后就能玩熟代码生成器了。