Mybatis plus 自动生成代码与自定义模板

2年前 (2022) 程序员胖胖胖虎阿
480 0 0

Generate 基础使用

文章目录

  • Generate 基础使用
    • 一. 自动生成代码落地与展示
    • 二. 分析生成代码
      • 2.1 数据库相关配置
      • 2.2 全局配置(GlobalConfig)
      • 2.3 包配置(PackageConfig)
      • 2.4 策略配置(StrategyConfig)
        • 2.4.5 Entity 策略配置
        • 2.4.6 Controller 策略配置
        • 2.4.7 Service 策略配置
        • 2.4.8 Mapper 策略配置
    • 三. 额外扩展
      • 3.1 自定义模板
        • 3.1.1 流程
        • 3.1.2 前置知识
        • 3.1.3 注入配置(InjectionConfig)
        • 3.1.4 源码增强map和输出路径
        • 3.1.5 Velocity 部分展示
      • 3.2 交互式生成代码


MyBatis-Plus-Generate

部分笔记,参考官网资料

笔记只讲解基础的使用,更详细的推荐大家阅读官网

一. 自动生成代码落地与展示


⭐️源码:https://gitee.com/zhp1221/java/tree/master/lab_02_mybatis_plus/generate


⭐️相关概念说明:

  1. 全局配置:配置数据库相关信息及个人信息

  2. 包配置:配置controller,entity,mapper,mapper.xml,service,serviceImpl,Other生成的包路径

    Other:自定义模板,详细可看自定义模板讲解,在第二节里

  3. 策略配置:表相关信息配置信息

    ps: 配置字段,文件名称等等

    策略配置细分为:

    1. Entity 策略配置
    2. Controller 策略配置
    3. Service 策略配置
    4. Mapper 策略配置

ps:配置完这些就可轻松使用代码生成了


代码目录

Mybatis plus 自动生成代码与自定义模板

代码生成总代码

/**
* 代码生成
*/
public static void generate() {

    FastAutoGenerator.create("jdbc:mysql://localhost:3306/community_db?useSSL=false&useUnicode=true&characterEncoding=utf-8&serverTimezone=UTC",
                             "root",
                             "root")

        // 全局配置
        .globalConfig(builder -> {
            builder
                .enableSwagger() // 是否启用swagger注解
                .author("zhanghp") // 作者名称
                .dateType(DateType.ONLY_DATE) // 时间策略
                .commentDate("yyyy-MM-dd") // 注释日期
                .outputDir("C:\\Dpan\\workspace\\study\\Ibatis_generate\\src\\main\\java") // 输出目录
                .fileOverride() // 覆盖已生成文件
                .disableOpenDir(); // 生成后禁止打开所生成的系统目录
        })

        // 包配置
        .packageConfig(builder -> {
            builder
                .parent("com.zhanghp") // 父包名
                .moduleName("test") // 模块包名
                .entity("dao.entity") // 实体类包名
                .service("service") // service包名
                .serviceImpl("service.impl") // serviceImpl包名
                .mapper("dao.mapper") // mapper包名
                .pathInfo(Collections.singletonMap(OutputFile.mapperXml, "C:\\Dpan\\workspace\\study\\Ibatis_generate\\src\\main\\resources\\mappers")) // xml位置(还可自定义配置entity,service等位置)
                .controller("controller") // controller包名
                .other("model"); // 自定义包名
        })

        // 策略配置
        .strategyConfig(builder -> {
            builder
                .addTablePrefix("t_") // 增加过滤表前缀
                .addTableSuffix("_db") // 增加过滤表后缀
                .addFieldPrefix("t_") // 增加过滤字段前缀
                .addFieldSuffix("_field") // 增加过滤字段后缀
                .addInclude("test") // 表匹配

                // Entity 策略配置
                .entityBuilder()
                .enableLombok() // 开启lombok
                .enableChainModel() // 链式
                .enableRemoveIsPrefix() // 开启boolean类型字段移除is前缀
                .enableTableFieldAnnotation() //开启生成实体时生成的字段注解
                .versionColumnName("version") // 乐观锁数据库字段
                .versionPropertyName("version") // 乐观锁实体类名称
                .logicDeleteColumnName("is_deleted") // 逻辑删除数据库中字段名
                .logicDeletePropertyName("deleted") // 逻辑删除实体类中的字段名
                .naming(NamingStrategy.underline_to_camel) // 表名 下划线 -》 驼峰命名
                .columnNaming(NamingStrategy.underline_to_camel) // 字段名 下划线 -》 驼峰命名
                .idType(IdType.ASSIGN_ID) // 主键生成策略 雪花算法生成id
                .formatFileName("%s") // Entity 文件名称
                .addTableFills(new Column("create_time", FieldFill.INSERT)) // 表字段填充
                .addTableFills(new Column("update_time", FieldFill.INSERT_UPDATE)) // 表字段填充
                .enableColumnConstant()
                .enableActiveRecord()

                // Controller 策略配置
                .controllerBuilder()
                .enableRestStyle() // 开启@RestController
                .formatFileName("%sController") // Controller 文件名称

                // Service 策略配置
                .serviceBuilder()
                .formatServiceFileName("%sService") // Service 文件名称
                .formatServiceImplFileName("%sServiceImpl") // ServiceImpl 文件名称

                // Mapper 策略配置
                .mapperBuilder()
                .enableMapperAnnotation() // 开启@Mapper
                .enableBaseColumnList() // 启用 columnList (通用查询结果列)
                .enableBaseResultMap() // 启动resultMap
                .formatMapperFileName("%sMapper") // Mapper 文件名称
                .formatXmlFileName("%sMapper"); // Xml 文件名称
        })
        .execute(); // 执行
}

生成结果

Mybatis plus 自动生成代码与自定义模板


二. 分析生成代码

2.1 数据库相关配置

属性 说明 示例
url jdbc 路径 jdbc:mysql://localhost:3306/community_db
username 数据库账号 root
password 数据库密码 root
FastAutoGenerator.create("jdbc:mysql://localhost:3306/community_db?useSSL=false&useUnicode=true&characterEncoding=utf-8&serverTimezone=UTC",
                        "root",
                        "root")

2.2 全局配置(GlobalConfig)

方法 说明 示例
enableSwagger 开启 swagger注解 默认值:false
author 作者 baomidou 默认值:作者
dateType(DateType) 时间策略 DateType.ONLY_DATE 默认值: DateType.TIME_PACK
commentDate(String) 注释日期 默认值: yyyy-MM-dd
outputDir(String) 指定输出目录 /opt/baomidou/ 默认值: windows:D:// linux or mac : /tmp
fileOverride 覆盖已生成文件 默认值:false
disableOpenDir 代码生成后,是否打开输出盘符文件夹 默认值:true

代码

// 全局配置
.globalConfig(builder -> {
    builder
        .enableSwagger() // 是否启用swagger注解
        .author("zhanghp") // 作者名称
        .dateType(DateType.ONLY_DATE) // 时间策略
        .commentDate("yyyy-MM-dd") // 注释日期
        .outputDir("C:\\Dpan\\workspace\\study\\Ibatis_generate\\src\\main\\java") // 输出目录
        .fileOverride() // 覆盖已生成文件
        .disableOpenDir(); // 生成后禁止打开所生成的系统目录
})

2.3 包配置(PackageConfig)

方法 说明 示例
parent(String) 父包名 默认值:com.baomidou
moduleName(String) 父包模块名 默认值:无
entity(String) Entity 包名 默认值:entity
service(String) Service 包名 默认值:service
serviceImpl(String) Service Impl 包名 默认值:service.impl
mapper(String) Mapper 包名 默认值:mapper
controller(String) Controller 包名 默认值:controller
other(String) 自定义文件包名 输出自定义文件时所用到的包名
pathInfo(Map<OutputFile, String>) 路径输出指定文件夹配置信息 Collections.singletonMap(OutputFile.mapperXml, “D://”)

代码演示

// 包配置
.packageConfig(builder -> {
    builder
        .parent("com.zhanghp") // 父包名
        .moduleName("test") // 模块包名
        .entity("dao.entity") // 实体类包名
        .service("service") // service包名
        .serviceImpl("service.impl") // serviceImpl包名
        .mapper("dao.mapper") // mapper包名
        .controller("controller") // controller包名
        .other("model") // 自定义包名
        .pathInfo(Collections.singletonMap(OutputFile.mapperXml, "C:\\Dpan\\workspace\\study\\Ibatis_generate\\src\\main\\resources\\mappers")); // xml位置
})

2.4 策略配置(StrategyConfig)

方法 说明
addTablePrefix(String…) 增加过滤表前缀
addTableSuffix(String…) 增加过滤表后缀
addFieldPrefix(String…) 增加过滤字段前缀
addFieldSuffix(String…) 增加过滤字段后缀
addInclude(String…) 增加表匹配(内存过滤)

代码演示

.strategyConfig(builder -> {
    builder
        .addTablePrefix("t_") // 增加过滤表前缀
        .addTableSuffix("_db") // 增加过滤表后缀
        .addFieldPrefix("t_") // 增加过滤字段前缀
        .addFieldSuffix("_field") // 增加过滤字段后缀
        .addInclude("test") // 表匹配

2.4.5 Entity 策略配置

方法 说明 示例
enableLombok 开启 lombok 模型 默认值:false
enableChainModel 开启链式模型 默认值:false
enableRemoveIsPrefix 开启 Boolean 类型字段移除 is 前缀 默认值:false
enableTableFieldAnnotation 开启生成实体时生成字段注解 默认值:false
versionColumnName(String) 乐观锁字段名(数据库)
versionPropertyName(String) 乐观锁属性名(实体)
logicDeleteColumnName(String) 逻辑删除字段名(数据库)
logicDeletePropertyName(String) 逻辑删除属性名(实体)
naming 数据库表映射到实体的命名策略 默认下划线转驼峰命名:NamingStrategy.underline_to_camel
columnNaming 数据库表字段映射到实体的命名策略 默认为 null,未指定按照 naming 执行
convertFileName(ConverterFileName) 转换文件名称
idType(IdType) 全局主键类型
formatFileName(String) 格式化文件名称
addTableFills(IFill…) 添加表字段填充
addTableFills(List) 添加表字段填充
enableColumnConstant 开启生成字段常量 默认值:false
enableActiveRecord 开启 ActiveRecord 模型 默认值:false

代码演示

// Entity 策略配置
.entityBuilder()
    .enableLombok() // 开启lombok
    .enableChainModel() // 链式
    .enableRemoveIsPrefix() // 开启boolean类型字段移除is前缀
    .enableTableFieldAnnotation() //开启生成实体时生成的字段注解
    .versionColumnName("version") // 乐观锁数据库字段
    .versionPropertyName("version") // 乐观锁实体类名称
    .logicDeleteColumnName("is_deleted") // 逻辑删除数据库中字段名
    .logicDeletePropertyName("deleted") // 逻辑删除实体类中的字段名
    .naming(NamingStrategy.underline_to_camel) // 表名 下划线 -》 驼峰命名
    .columnNaming(NamingStrategy.underline_to_camel) // 字段名 下划线 -》 驼峰命名
    .idType(IdType.ASSIGN_ID) // 主键生成策略 雪花算法生成id
    .formatFileName("%s") // Entity 文件名称
    .addTableFills(new Column("create_time", FieldFill.INSERT)) // 表字段填充
    .addTableFills(new Column("update_time", FieldFill.INSERT_UPDATE)) // 表字段填充
    .enableColumnConstant()
    .enableActiveRecord()

2.4.6 Controller 策略配置

方法 说明 示例
enableRestStyle 开启生成@RestController 控制器 默认值:false
formatFileName(String) 格式化文件名称

代码演示

// Controller 策略配置
.controllerBuilder()
    .enableRestStyle() // 开启@RestController
    .formatFileName("%sController") // Controller 文件名称

2.4.7 Service 策略配置

方法 说明
formatServiceFileName(String) 格式化 service 接口文件名称
formatServiceImplFileName(String) 格式化 service 实现类文件名称

代码演示

// Service 策略配置
.serviceBuilder()
    .formatServiceFileName("%sService") // Service 文件名称
    .formatServiceImplFileName("%sServiceImpl") // ServiceImpl 文件名称

2.4.8 Mapper 策略配置

方法 说明 示例
enableMapperAnnotation 开启 @Mapper 注解 默认值:false
enableBaseResultMap 启用 BaseResultMap 生成 默认值:false
enableBaseColumnList 启用 BaseColumnList 默认值:false
formatMapperFileName(String) 格式化 mapper 文件名称
formatXmlFileName(String) 格式化 xml 实现类文件名称

代码演示

// Mapper 策略配置
.mapperBuilder()
    .enableMapperAnnotation() // 开启@Mapper
    .enableBaseColumnList() // 启用 columnList (通用查询结果列)
    .enableBaseResultMap() // 启动resultMap
    .formatMapperFileName("%sMapper") // Mapper 文件名称
    .formatXmlFileName("%sMapper"); // Xml 文件名称

三. 额外扩展

3.1 自定义模板

3.1.1 流程

Velocity 语法学习

注入配置-自定义模板配置

源码增强自定义配置

velocity 部分展示

3.1.2 前置知识

Velocity语法教学

http://www.51gjie.com/javaweb/896.html

# 自定义模板使用的是Velocity模板引擎(也可使用Freemarker)
- 需提前学习Velocity相关语法(语法很简单)
- ps:velocity的注释在一行中一定要顶头(否则生成的java,会在格式中多空格)

3.1.3 注入配置(InjectionConfig)

方法 说明 示例
customMap(Map<String, Object>) 自定义配置 Map 对象 Collections.singletonMap(“test”, “baomidou”)
customFile(Map<String, String>) 自定义配置模板文件 Collections.singletonMap(“test.txt”, “/templates/test.vm”)

代码演示

customMap我是基于源码增强的,所以没有在InjecttionConfig中配置

ps:自定义模板和map,是根据本人在公司项目中的使用所需进行的封装

// 注入配置
.injectionConfig(builder -> {

    // tag::custom file
    // 自定义vo,ro,qo等数据模型
    Map<String, String> customFile = new HashMap<>();
    customFile.put(CustomEnum.VO.getModelSuffix(), CustomEnum.VO.getPath());
    customFile.put(CustomEnum.RO.getModelSuffix(), CustomEnum.RO.getPath());
    customFile.put(CustomEnum.QO.getModelSuffix(), CustomEnum.QO.getPath());
    // 自定义MapStruct
    customFile.put(CustomEnum.CONVERTER.getModelSuffix(), CustomEnum.CONVERTER.getPath());
    // end::custom file

    builder
        .customFile(customFile); // 自定义模板
})
.templateEngine(new EnhanceVelocityTemplaEngine())

CustomEnum

注入配置中所引用的类

封装模块后缀,模板路径的枚举类

public enum CustomEnum {
    VO("VO", "templates/model/vo.java.vm"),
    RO("RO", "templates/model/ro.java.vm"),
    QO("QO", "templates/model/qo.java.vm"),
    CONVERTER("Converter", "templates/converter/converter.java.vm");

    CustomEnum(String modelSuffix, String path) {
        this.modelSuffix = modelSuffix;
        this.path = path;
    }

    @Getter
    private String modelSuffix;

    @Getter
    private String path;

    public static List<String> getModel(){
        List<String> result = new ArrayList<>();
        CustomEnum[] values = CustomEnum.values();
        for (CustomEnum it : values) {
            result.add(it.getModelSuffix());
        }
        return result;
    }
}

3.1.4 源码增强map和输出路径

内部类 - EnhanceVelocityTemplateEngine

getObjectMap:自定义包路径,返回结果,工具类等

outputCustomFile:输出路径封装

/**
* 代码生成器支持自定义[DTO\VO等]模版
 */
public final static class EnhanceVelocityTemplateEngine extends VelocityTemplateEngine {
        @Override
        public Map<String, Object> getObjectMap(@NotNull ConfigBuilder config, @NotNull TableInfo tableInfo) {
            // 获取实体类名字
            String entityName = tableInfo.getEntityName();
            // 获取object map
            Map<String, Object> objectMap = super.getObjectMap(config, tableInfo);
            // 获取Other的盘符路径
            String otherPath = (String)((Map<String, Object>) objectMap.get("package")).get("Other");
            // 自定义枚举
            List<String> modelList = CustomEnum.getModel();
            // 循环
            modelList.forEach(it -> {
                // 转小写
                String var = it.toLowerCase();
                // 存入object map
                objectMap.put(var, otherPath + "." + var + "." + entityName + it);
            });
            // 自定义converter
            objectMap.put("commonConverter", "com.zhanghp.common.converter.CommonConverter");
            // converter utils
            objectMap.put("converterUtil", "com.zhanghp.common.converter.utils.ConverterUtil");
            // 分页
            objectMap.put("pageParent", "com.zhanghp.common.model.PageParent");
            // utils
            objectMap.put("objectUtils", "com.zhanghp.common.utils.ObjectUtils");
            // 返回结果封装
            objectMap.put("r", "com.zhanghp.common.response.R");
            return objectMap;
        }

        @Override
        protected void outputCustomFile(@NotNull Map<String, String> customFile, @NotNull TableInfo tableInfo, @NotNull Map<String, Object> objectMap) {
            // 获取实体类名字
            String entityName = tableInfo.getEntityName();
            // 获取other包盘符路径
            String otherPath = this.getPathInfo(OutputFile.other);
            // 输出自定义java模板
            customFile.forEach((key, value) -> {
                // 输出路径
                String fileName = otherPath + File.separator + key.toLowerCase() + File.separator + entityName + key +".java";
                // 输出velocity的java模板
                this.outputFile(new File(fileName), objectMap, value);
            });
        }
    }

3.1.5 Velocity 部分展示

controller层引擎模板

更多的代码,可clone我的代码,详细看别的velocity模板用法

package ${package.Controller};


#if(${swagger})
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiModel;
import io.swagger.annotations.ApiOperation;
#end
import lombok.RequiredArgsConstructor;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;
import ${package.Service}.${table.serviceName};
import ${ro};
import ${qo};
import ${r};

## 判断是否包含父类
#if(${superControllerClassPackage})
import ${superControllerClassPackage};
#end

/**
 * <p>
 * $!{table.comment} controller
 * </p>
 *
 * @author ${author}
 * @since ${date}
 */
@RequiredArgsConstructor(onConstructor_ = @Autowired)
## swagger相关
#if(${swagger})
@Api(tags = "REST - $!{table.comment}")
#end
## controller注解
#if(${restControllerStyle})
@RestController
#else
@Controller
#end
## 请求地址
@RequestMapping("#if(${package.ModuleName})/${package.ModuleName}#end/#if(${controllerMappingHyphenStyle})${controllerMappingHyphen}#else${table.entityPath}#end")
## 类名及继承判断
#if(${superControllerClass})
public class ${table.controllerName} extends ${superControllerClass} {
#else
public class ${table.controllerName} {
#end
## 注入服务类
    private final ${table.serviceName} service;

## 主键类型判断
#foreach($field in ${table.fields})
    #if(${field.keyFlag})
## 主键类型
        #set($idType=${field.propertyType})
## 主键名称
        #set($idName=${field.propertyName})
## 跳出
        #break
    #end
#end
    @PostMapping("/create")
#if(${swagger})
    @ApiOperation(value = "新增")
#end
    public R createOne(@RequestBody ${entity}RO ro){
        boolean flag = service.create(ro);
        return R.judgeR(flag, flag ? "创建成功" : "创建失败");
    }

    @DeleteMapping("/delete/{${idName}}")
#if(${swagger})
    @ApiOperation(value = "删除")
#end
    public R delete(@PathVariable("${idName}")${idType} ${idName}){
        boolean flag = service.delete(${idName});
        return R.judgeR(flag, flag ? "删除成功" : "删除失败");
    }

    @PutMapping("/update")
#if(${swagger})
    @ApiOperation(value = "修改")
#end
    public R update(@RequestBody ${entity}RO ro){
        boolean flag = service.update(ro);
        return R.judgeR(flag, flag ? "修改成功" : "修改失败");
    }

    @GetMapping("/{${idName}}")
#if(${swagger})
    @ApiOperation(value = "单条查询")
#end
    public R getOne(@PathVariable("${idName}")${idType} ${idName}){

        return R.success(service.getById(${idName}));
    }

    @PostMapping("/findList")
#if(${swagger})
    @ApiOperation(value = "列表查询")
#end
    public R findList(@RequestBody ${entity}QO qo){

        return R.success(service.findList(qo));
    }

    @PostMapping("/findPage")
#if(${swagger})
    @ApiOperation(value = "分页查询")
#end
    public R findPage(@RequestBody ${entity}QO qo){

        return R.success(service.findPage(qo));
    }

}

3.2 交互式生成代码

/**
 * <p>
 *     代码生成:输入生成
 * </p>
 * <p>
 * <a href = "https://baomidou.com/pages/981406/#%E6%95%B0%E6%8D%AE%E5%BA%93%E9%85%8D%E7%BD%AE-datasourceconfig">
 * mybatis-plus generator
 * </a>
 * </p>
 *
 * @author zhanghp
 * @date 2022-07-16 14:53
 */
public class IbatisScannerGenerate {

    public static void main(String[] args) {
        generate();
    }
    /**
     * 代码生成
     */
    private static void generate() {

        FastAutoGenerator.create("jdbc:mysql://localhost:3306/community_db?useSSL=false&useUnicode=true&characterEncoding=utf-8&serverTimezone=UTC",
                        "root",
                        "root")

                // 全局配置
                .globalConfig((scanner, builder) -> {
                    builder
                            .enableSwagger() // 是否启用swagger注解
                            .author(scanner.apply("请输入作者名称")) // 作者名称
                            .dateType(DateType.ONLY_DATE) // 时间策略
                            .commentDate("yyyy-MM-dd") // 注释日期
                            .outputDir(scanner.apply("请输入-》指定输出目录")) // 输出目录
                            .fileOverride() // 覆盖已生成文件
                            .disableOpenDir(); // 生成后禁止打开所生成的系统目录
                })

                // 包配置
                .packageConfig((scanner, builder) -> {
                    builder
                            .parent(scanner.apply("清楚如父包路径,如com.***.***")) // 父包名
                            .moduleName(scanner.apply("请输入模块名称")) // 模块包名
                            .entity(scanner.apply("请输入实体类包名")) // 实体类包名
                            .service(scanner.apply("请输入Service层包名")) // service包名
                            .serviceImpl(scanner.apply("请输入ServiceImpl包名")) // serviceImpl包名
                            .mapper(scanner.apply("请输入Mapper层包名")) // mapper包名
                            .controller(scanner.apply("请输入controller层包名")) // controller包名
                            .other(scanner.apply("请输入自定义包名")); // 自定义包名
                })

                // 策略配置
                .strategyConfig((scanner, builder) -> {
                    builder
                            .addTablePrefix(scanner.apply("请输入所要过滤的表前缀")) // 增加过滤表前缀
                            .addTableSuffix(scanner.apply("请输入所要过滤的表后缀")) // 增加过滤表后缀
                            .addFieldPrefix(scanner.apply("请输入所要过滤字段前缀")) // 增加过滤字段前缀
                            .addFieldSuffix(scanner.apply("请输入所要过滤字段后缀")) // 增加过滤字段后缀
                            .addInclude(getTables(scanner.apply("请输入表名,多个英文逗号分隔?所有输入 all"))) // 表匹配

                            // Entity 策略配置
                            .entityBuilder()
                            .enableLombok() // 开启lombok
                            .enableChainModel() // 链式
                            .enableRemoveIsPrefix() // 开启boolean类型字段移除is前缀
                            .enableTableFieldAnnotation() //开启生成实体时生成的字段注解
                            .versionColumnName(scanner.apply("请输入数据库中的乐观锁字段")) // 乐观锁数据库字段
                            .versionPropertyName(scanner.apply("请输入字段中的乐观锁名称")) // 乐观锁实体类名称
                            .logicDeleteColumnName(scanner.apply("请输入数据库中的逻辑删除字段")) // 逻辑删除数据库中字段名
                            .logicDeletePropertyName(scanner.apply("请输入数据库中的实体名称")) // 逻辑删除实体类中的字段名
                            .naming(NamingStrategy.underline_to_camel) // 表名 下划线 -》 驼峰命名
                            .columnNaming(NamingStrategy.underline_to_camel) // 字段名 下划线 -》 驼峰命名
                            .idType(IdType.ASSIGN_ID) // 主键生成策略 雪花算法生成id
                            .formatFileName("%s") // Entity 文件名称
                            .addTableFills(new Column("create_time", FieldFill.INSERT)) // 表字段填充
                            .addTableFills(new Column("update_time", FieldFill.INSERT_UPDATE)) // 表字段填充

                            // Controller 策略配置
                            .controllerBuilder()
                            .enableRestStyle() // 开启@RestController
                            .formatFileName("%sController") // Controller 文件名称

                            // Service 策略配置
                            .serviceBuilder()
                            .formatServiceFileName("%sService") // Service 文件名称
                            .formatServiceImplFileName("%sServiceImpl") // ServiceImpl 文件名称

                            // Mapper 策略配置
                            .mapperBuilder()
                            .enableMapperAnnotation() // 开启@Mapper
                            .enableBaseColumnList() // 启用 columnList (通用查询结果列)
                            .enableBaseResultMap() // 启动resultMap
                            .formatMapperFileName("%sMapper") // Mapper 文件名称
                            .formatXmlFileName("%sMapper"); // Xml 文件名称
                })

                // 注入配置
                .injectionConfig((scanner, builder) -> {
                    // 自定义vo,ro,qo等数据模型
                    Map<String, String> customFile = new HashMap<>();
                    customFile.put("VO.java", "templates/model/vo.java.vm");
                    customFile.put("RO.java", "templates/model/ro.java.vm");
                    customFile.put("QO.java", "templates/model/qo.java.vm");
                    customFile.put("URO.java", "templates/model/uro.java.vm");
                    // 自定义MapStruct
                    customFile.put("Converter.java", "templates/converter/converter.java.vm");

                    // 自定义配置对象
                    Map<String, Object> customMap = new HashMap<>();
                    customMap.put("vo", "VO");
                    customMap.put("ro", "RO");
                    customMap.put("qo", "QO");
                    customMap.put("uro", "URO");
                    builder
                            .customFile(customFile) // 自定义模板
                            .customMap(customMap); // 自定义map
                });


    }

    // 处理 all 情况
    protected static List<String> getTables(String tables) {
        return "all".equals(tables) ? Collections.emptyList() : Arrays.asList(tables.split(","));
    }
}

⭐️源码:https://gitee.com/zhp1221/java/tree/master/lab_02_mybatis_plus/generate


版权声明:程序员胖胖胖虎阿 发表于 2022年9月20日 上午12:32。
转载请注明:Mybatis plus 自动生成代码与自定义模板 | 胖虎的工具箱-编程导航

相关文章

暂无评论

暂无评论...