MyBatis-Plus代码生成器(AutoGenerator)用法及说明

<dependency>
    <groupId>com.baomidou</groupId>
    <artifactId>mybatis-plus-generator</artifactId>
    <version>3.5.3.1</version> <!-- 请使用最新稳定版本 -->
</dependency>

<dependency>
    <groupId>org.freemarker</groupId>
    <artifactId>freemarker</artifactId>
    <version>2.3.31</version> <!-- 请使用最新稳定版本 -->
</dependency>

FastAutoGenerator.create(DATA_SOURCE_URL, USERNAME, PASSWORD)
    .globalConfig(builder -> {
        // 1. 全局配置 (GlobalConfig)
    })
    .packageConfig(builder -> {
        // 2. 包配置 (PackageConfig)
    })
    .strategyConfig(builder -> {
        // 3. 策略配置 (StrategyConfig)
    })
    .templateEngine(new FreemarkerTemplateEngine()) // 4. 模板引擎
    .execute(); // 5. 执行生成

CREATE TABLE `sys_user` (
  `id` bigint NOT NULL AUTO_INCREMENT COMMENT '主键ID',
  `user_name` varchar(50) NOT NULL COMMENT '用户名',
  `age` int DEFAULT NULL COMMENT '年龄',
  `email` varchar(100) DEFAULT NULL COMMENT '邮箱',
  `create_time` datetime DEFAULT CURRENT_TIMESTAMP COMMENT '创建时间',
  PRIMARY KEY (`id`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COMMENT='系统用户表';

package com.example.generator;

import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.generator.FastAutoGenerator;
import com.baomidou.mybatisplus.generator.config.OutputFile;
import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;

import java.util.Collections;

public class CodeGenerator {

    // 数据库连接信息
    private static final String URL = "jdbc:mysql://localhost:3306/test_db?useUnicode=true&characterEncoding=utf8&serverTimezone=Asia/Shanghai";
    private static final String USERNAME = "root";
    private static final String PASSWORD = "your_password";
    private static final String PARENT_PACKAGE = "com.example.demo";
    private static final String MODULE_NAME = "system";
    private static final String AUTHOR = "Manus AI";

    public static void main(String[] args) {
        FastAutoGenerator.create(URL, USERNAME, PASSWORD)
            // 1. 全局配置
            .globalConfig(builder -> {
                builder.author(AUTHOR) // 设置作者
                    .enableSwagger() // 开启 Swagger 模式
                    .fileOverride() // 覆盖已生成文件
                    .outputDir(System.getProperty("user.dir") + "/src/main/java"); // 指定输出目录
            })
            // 2. 包配置
            .packageConfig(builder -> {
                builder.parent(PARENT_PACKAGE) // 设置父包名
                    .moduleName(MODULE_NAME) // 设置父包模块名 (com.example.demo.system)
                    .pathInfo(Collections.singletonMap(OutputFile.xml, 
                            System.getProperty("user.dir") + "/src/main/resources/mapper/" + MODULE_NAME)); // 设置 Mapper XML 输出路径
            })
            // 3. 策略配置
            .strategyConfig(builder -> {
                builder.addInclude("sys_user") // 设置需要生成的表名
                    .addTablePrefix("sys_") // 设置过滤表前缀
                    
                    // 3.1 Entity 策略
                    .entityBuilder()
                    .enableLombok() // 开启 Lombok
                    .enableChainModel() // 开启链式模型
                    .idType(IdType.AUTO) // 设置主键生成策略
                    .columnNaming(com.baomidou.mybatisplus.generator.config.rules.NamingStrategy.underline_to_camel) // 字段名下划线转驼峰
                    
                    // 3.2 Controller 策略
                    .controllerBuilder()
                    .enableRestStyle() // 开启 @RestController
                    
                    // 3.3 Service 策略
                    .serviceBuilder()
                    .formatServiceFileName("%sService") // 格式化 Service 接口文件名称
                    
                    // 3.4 Mapper 策略
                    .mapperBuilder()
                    .enableMapperAnnotation() // 开启 @Mapper 注解
                    .enableBaseResultMap() // 启用 BaseResultMap
                    .enableBaseColumnList(); // 启用 BaseColumnList
            })
            // 4. 模板引擎配置 (使用 Freemarker)
            .templateEngine(new FreemarkerTemplateEngine()) 
            // 5. 执行生成
            .execute();
    }
}

// 示例：配置自定义 Entity 模板
.templateConfig(builder -> {
    builder.entity("templates/my_entity.java"); // 使用自定义模板
})

// 在 GlobalConfig 中设置时间类型策略
.globalConfig(builder -> {
    // ... 其他配置
    builder.dateType(com.baomidou.mybatisplus.generator.config.rules.DateType.TIME_PACK); // 映射为 LocalDateTime
})

.strategyConfig(builder -> {
    // ... 其他配置
    builder.entityBuilder()
        .superClass("com.example.BaseEntity") // 设置父类
        .addSuperEntityColumns("id", "create_time", "update_time"); // 排除父类已有的字段
})