简易版 | 代码生成器(包含插件)

一、代码生成器

先导入依赖

<!-- Mybatis-Plus -->
<dependency><groupId>com.baomidou</groupId><artifactId>mybatis-plus-boot-starter</artifactId><version>3.5.6</version>
</dependency><!-- 代码生成器 -->
<dependency><groupId>com.baomidou</groupId><artifactId>mybatis-plus-generator</artifactId><version>3.5.6</version>
</dependency><!-- 其他依赖 -->
<dependency><groupId>org.freemarker</groupId><artifactId>freemarker</artifactId><version>2.3.31</version>
</dependency>
<dependency><groupId>io.springfox</groupId><artifactId>springfox-swagger2</artifactId><version>2.9.2</version>
</dependency>
<dependency><groupId>io.springfox</groupId><artifactId>springfox-swagger-ui</artifactId><version>2.9.2</version>
</dependency>
<!-- 加载数据库驱动 -->
<dependency><groupId>mysql</groupId><artifactId>mysql-connector-java</artifactId><version>8.0.33</version>
</dependency>

写入application.yaml 文件的配置

# 加载驱动
datasource:driver-class-name: com.mysql.cj.jdbc.Driverurl: jdbc:mysql://localhost:3306/blog?allowPublicKeyRetrieval=true&serverTimezone=GMT%2B8username: rootpassword: Qwer1234

编写代码生成器类

import com.baomidou.mybatisplus.generator.FastAutoGenerator;
import com.baomidou.mybatisplus.generator.config.OutputFile;
import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;
import com.mysql.cj.protocol.Resultset;import java.sql.*;
import java.util.Collections;public class CodeGenerator {public static final String URL = "jdbc:mysql://localhost:3306/blog?allowPublicKeyRetrieval=true";public static final String[] arrStr = {// 这里是对应的数据库"ad","ad_type","article","article_tag","article_tag_list","comment","comment_reply","link","user"};public static void main(String[] args) throws SQLException {for (int i = 0; i < arrStr.length; i++) {int finalI = i;FastAutoGenerator.create(URL, "root", "Qwer1234").globalConfig(builder -> {builder.author("XIAOMIE") // 设置作者.enableSwagger() // 开启swagger模式.outputDir("src\\main\\java\\"); // 指定输出目录}).packageConfig(builder -> {builder.parent("com.blog") // 设置父包名aa.moduleName(null) // 设置父包模块名.pathInfo(Collections.singletonMap(OutputFile.xml, "src/main/resources/mapper")); // 设置mapperXml生成路径}).strategyConfig(builder -> {builder.entityBuilder().enableLombok(); // 使用lombokbuilder.controllerBuilder().enableRestStyle(); // 开启生成@RestController控制器builder.addInclude(arrStr[finalI]); // 设置需要生成的表名}).templateEngine(new FreemarkerTemplateEngine()) // 使用Freemarker引擎模板，默认的是Velocity引擎模板.execute();}}
}