mybatis-plus-generator自定义模版

Maven坐标

<!-- 代码生成-->
        <dependency>
            <groupId>com.baomidou</groupId>
            <artifactId>mybatis-plus-boot-starter</artifactId>
            <version>3.5.3.1</version>
        </dependency>
        <dependency>
            <groupId>com.baomidou</groupId>
            <artifactId>mybatis-plus-generator</artifactId>
            <version>3.5.3.1</version>
            <scope>compile</scope>
        </dependency>

        <!-- Mysql驱动包 -->
        <dependency>
            <groupId>mysql</groupId>
            <artifactId>mysql-connector-java</artifactId>
            <version>8.0.30</version> <!-- 使用最新版本 -->
        </dependency>

Velocity的模版

<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
<mapper namespace="${package.Mapper}.${table.mapperName}">

#if(${enableCache})
    <!-- 开启二级缓存2 -->
    <cache type="${cacheClassName}"/>

#end
#if(${baseResultMap})
    <!-- 通用查询映射结果 -->
    <resultMap id="BaseResultMap" type="${package.Entity}.${entity}">
#foreach($field in ${table.fields})
#if(${field.keyFlag})##生成主键排在第一位
        <id column="${field.name}" property="${field.propertyName}" />
#end
#end
#foreach($field in ${table.commonFields})##生成公共字段
        <result column="${field.name}" property="${field.propertyName}" />
#end
#foreach($field in ${table.fields})
#if(!${field.keyFlag})##生成普通字段
        <result column="${field.name}" property="${field.propertyName}" />
#end
#end
    </resultMap>

#end
#if(${baseColumnList})
    <!-- 通用查询结果列 -->
    <sql id="Base_Column_List">
#foreach($field in ${table.commonFields})
        ${field.columnName},
#end
        ${table.fieldNames}
    </sql>

    <!-- 插入或更新用户 -->

    <insert id="insertOrUpdate${table.entityName}"  parameterType="${package.Entity}.${table.entityName}">
        INSERT INTO ${table.name} (
                #foreach($field in ${table.fields})
                    #if($foreach.hasNext)
                        `${field.name}` ,
                    #else
                        `${field.name}`
                    #end
                #end
            ) values
        <foreach item="item" index="index" collection="list" separator="," open="(" close=")"  >
            #foreach($field in ${table.fields})
                #if($foreach.hasNext)
                    #{item.${field.propertyName}},
                #else
                    #{item.${field.propertyName}}
                #end
            #end
        </foreach>
        ON DUPLICATE KEY UPDATE
        #foreach($field in ${table.fields})
            #if($foreach.hasNext)
                `${field.name}` = values(`${field.name}`),
            #else
                `${field.name}` = values(`${field.name}`)
            #end
        #end
    </insert>

#end
</mapper>

代码生成java类

public class MyBatisPlusGenerator {

    public static void main(String[] args) {
        // 要生成的表，多个逗号分开。过滤表前缀（没有就不用写）。包名。模块名
        String[] tableArray = {"sys_role" };
        String[] tablePrefixArray = {"t_"};
        String projectPath = "F:\\work\\test\\";
        String packageName = "com.renpho.cpc";
        // 数据源配置
        String url = "jdbc:mysql://localhost:3306/ry-vue?useUnicode=true&characterEncoding=utf8&zeroDateTimeBehavior=convertToNull&useSSL=true&serverTimezone=GMT%2B8";
        String username = "root";
        String password = "123456"; 



        FastAutoGenerator.create(url, username, password) //数据源

                //全局配置
                .globalConfig(builder -> {
                    builder.author("McCaw")        // 设置作者
//                            .enableSwagger()        // 开启 swagger 模式 默认值:false
                            .commentDate("yyyy-MM-dd") // 注释日期
                            .dateType(DateType.ONLY_DATE)   //定义生成的实体类中日期类型 DateType.ONLY_DATE 默认值: DateType.TIME_PACK
                            .outputDir(projectPath + "\\src\\main\\java"); // 指定输出目录
                })

                //包配置
                .packageConfig(builder -> {
                    builder.parent(packageName) // 父包模块名
                            .controller("controller")   //Controller 包名 默认值:controller
                            .entity("domain")           //Entity 包名 默认值:entity
                            .service("service")         //Service 包名 默认值:service
                            .serviceImpl("service.impl")
                            .mapper("mapper")           //Mapper 包名 默认值:mapper
//                            .xml("mapper.xml")         //默认存放在mapper的xml下
                            .pathInfo(Collections.singletonMap(OutputFile.xml, projectPath + "\\src\\main\\resources\\mapper")); // 设置mapperXml生成路径

                })

                //策略配置
                .strategyConfig(builder -> {
                    builder.addInclude(tableArray) // 添加表匹配，指定要生成的数据表名，不写默认选定数据库所有表 可变长参数“user”, “user1”
                            .addTablePrefix(tablePrefixArray) // 设置过滤表前缀
                            //Entity 策略配置
                            .entityBuilder()
                            .naming(NamingStrategy.underline_to_camel) //下划线转驼峰命
                            .columnNaming(NamingStrategy.underline_to_camel)
                            .addTableFills(new Column("create_time", FieldFill.INSERT)) // 自动填充配置
                            .addTableFills(new Property("update_time", FieldFill.INSERT_UPDATE))
                            .enableLombok() //开启lombok
                            .enableTableFieldAnnotation() //开启生成实体时生成字段注解
                            .versionPropertyName("version")
                            .versionColumnName("version")
//                            .logicDeleteColumnName("deleted")// 说明逻辑删除是哪个字段
                            .idType(IdType.AUTO)
                            .enableFileOverride() //覆盖已生成文件

                            //service策略配置
                            .serviceBuilder()
                            .formatServiceFileName("%sService")
                            .formatServiceImplFileName("%sServiceImpl")
                            .enableFileOverride() //覆盖已生成文件

                            //controller 策略配置
                            .controllerBuilder()
                            .enableHyphenStyle() //开启驼峰转连字符
                            .formatFileName("%sController")
                            .enableRestStyle() // 开启RestController注解
                            .enableFileOverride() //覆盖已生成文件

                            // mapper策略配置
                            .mapperBuilder()
                            .enableFileOverride() //覆盖已生成文件
                            .enableBaseResultMap()
                            .enableBaseColumnList()

                            .formatMapperFileName("%sMapper")
                            .formatXmlFileName("%sMapper");
                })
//                .templateEngine(new VelocityTemplateEngine()) // 使用 Velocity 模板引擎
                .templateConfig(templateConfig -> {
                    templateConfig.mapper("templates/mapper.xml.vm");
                })
                .execute();

        System.out.println("生成完成");
    }
}