mybatis-plus自动生成代码

1.引入pom依赖

<!-- mybatis-plus generator -->
<dependency>
    <groupId>com.baomidou</groupId>
    <artifactId>mybatis-plus-generator</artifactId>
    <version>3.5.2</version>
</dependency>
<!-- velocity -->
<dependency>
    <groupId>org.apache.velocity</groupId>
    <artifactId>velocity-engine-core</artifactId>
    <version>2.3</version>
</dependency>

2.构建自动生成代码类 test/xx/xx

package com.xxx;

import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.generator.FastAutoGenerator;
import com.baomidou.mybatisplus.generator.config.TemplateType;
import com.baomidou.mybatisplus.generator.config.rules.DateType;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;

import java.util.Arrays;
import java.util.List;


/**
 * 生成代码
 *
 * @author xxx
 * @since 2022/9/8 18:24
 */
public class MyBatisPlusGenerator {
    public static void main(String[] args) {
        // 要生成的数据库表名
        List<String> tableNames = Arrays.asList("t_org_info");
        // 创建代码生成工具类
        FastAutoGenerator generator = create(tableNames, "system.org","T");
        // 执行生成代码
        generator.execute();
    }

    private static FastAutoGenerator create(List<String> tableNames, String packageName,String tablePre) {
        // 数据库连接地址，
        String url = "jdbc:mysql://127.0.0.1:3306/dev?autoReconnect=true&useServerPreparedStmts=false&rewriteBatchedStatements=true&characterEncoding=UTF-8&useSSL=false&allowMultiQueries=true&serverTimezone=Asia/Shanghai";
        // 数据库用户名
        String name = "root";
        // 数据库密码
        String password = "root";
        FastAutoGenerator generator = FastAutoGenerator.create(url, name, password)
                // 全局配置
                .globalConfig(builder -> {
                    // 获取生成的代码路径,这里没有写死，是运行时获取的，这样可以防止不同的开发人员来回修改生成路径的问题。
                    String outputDir = System.getProperty("user.home") + "\\Desktop\\generatorCode";
                    builder.outputDir(outputDir)
                            .dateType(DateType.ONLY_DATE)
                            // 生成的类注释中的作者名称，为了统一表示，这里写死了
                            .author("xxx");
                })
                // 生成的代码包路径配置
                .packageConfig(builder -> {
                    // 生成的代包公共路径
                    builder.parent("com.xxx.xxx." + packageName);
                    // 生成的mapper xml的存放目录，是在parent路径下面的
                    builder.xml("mapper")
                            // 生成的实例类目录
                            .entity("domain.entity")
                            // 生成的service目录
                            .service("service")
                            // 生成的ampper目录
                            .mapper("dao");

                }).strategyConfig(builder -> {
                    // 添加要生成的的数据库表
                    builder.addInclude(tableNames);
                    // 启用大写模式
                    //.enableCapitalMode();
                    // 配置生成的实体类策略，不生成serialVersionID
                    builder.entityBuilder().disableSerialVersionUID()
                            // 如果数据库表名带下划线，按驼峰命名法
                            .columnNaming(NamingStrategy.underline_to_camel)
                            // 开启AR模式
                            .enableActiveRecord()
                            // 使用lombok
                            .enableLombok()
                            // 标记实例类的主键生成方式，如果插入时没有指定，刚自动分配一个，默认是雪花算法
                            .idType(IdType.ASSIGN_ID)
                            .enableTableFieldAnnotation()
                            // 指定生成的实体类名称
                            .convertFileName(entityName -> entityName.replace(tablePre,"") + "Entity")
                            .mapperBuilder().convertXmlFileName(entityName ->  entityName.replace(tablePre,"") + "Dao")
                            .convertMapperFileName(entityName ->  entityName.replace(tablePre,"") + "Dao")
                            // 指定生成的service接口名称
                            .serviceBuilder().convertServiceFileName(entityName ->  entityName.replace(tablePre,"") + "ServiceImpl")
                            // 指定生成的serviceImpl的名称
                            .convertServiceImplFileName(entityName -> entityName.replace(tablePre,"") + "Service");

                }).templateConfig(builder -> {
                    // 不生成Controller
                    builder.disable(TemplateType.CONTROLLER);
                });
        return generator;
    }

}