springboot整合mybatis-plus新旧代码生成器

前置要求：已经引入springboot、springboot-test、mybatis-plus依赖(mybatis-plus的版本和生成器可以保持一致）

旧版(生成器版本3.5.1以下)

引入maven依赖

 <!-- 代码自动生成器依赖-->
        <dependency>
            <groupId>com.baomidou</groupId>
            <artifactId>mybatis-plus-generator</artifactId>
            <version>3.4.1</version>
        </dependency>
        <!-- 代码生成器模板引擎 -->
        <dependency>
            <groupId>org.apache.velocity</groupId>
            <artifactId>velocity-engine-core</artifactId>
            <version>2.0</version>
        </dependency>

在test目录下创建单元测试，并运行即可

@SpringBootTest
public class MybatisPlusGenerator {

    //3.5.1以下旧版生成
    @Test
    public void generate() {
        // 代码生成器
        AutoGenerator mpg = new AutoGenerator();

        // 全局配置
        GlobalConfig gc = new GlobalConfig();
        String projectPath = System.getProperty("user.dir");
        gc.setOutputDir(projectPath + "/src/main/java");//设置代码生成路径
        gc.setFileOverride(true);//是否覆盖以前文件
        gc.setOpen(false);//是否打开生成目录
        gc.setAuthor("pzp");//设置项目作者名称
        gc.setIdType(IdType.AUTO);//设置主键策略
        gc.setBaseResultMap(true);//生成基本ResultMap
        gc.setBaseColumnList(true);//生成基本ColumnList
        gc.setServiceName("%sService");//去掉服务默认前缀
        gc.setDateType(DateType.ONLY_DATE);//设置时间类型
        gc.setSwagger2(true);//生成swagger2注解
        mpg.setGlobalConfig(gc);

        // 数据源配置
        DataSourceConfig dsc = new DataSourceConfig();
        dsc.setUrl("jdbc:mysql://localhost:3306/pzp?useUnicode=true&characterEncoding=UTF-8&useSSL=false&serverTimezone=GMT");
        dsc.setDriverName("com.mysql.jdbc.Driver");
        dsc.setUsername("root");
        dsc.setPassword("root");
        mpg.setDataSource(dsc);

        // 包配置
        PackageConfig pc = new PackageConfig();
        pc.setParent("cn.szqiancheng.admin.project");
        pc.setMapper("mapper");
        pc.setXml("mapper.xml");
        pc.setEntity("pojo.entity");
        pc.setService("service");
        pc.setServiceImpl("service.impl");
        pc.setController("controller");
        mpg.setPackageInfo(pc);

        // 策略配置
        StrategyConfig sc = new StrategyConfig();
        sc.setNaming(NamingStrategy.underline_to_camel);// 表名 下划线映射小驼峰
        sc.setColumnNaming(NamingStrategy.underline_to_camel);// 表字段 下划线映射小驼峰
        sc.setEntityLombokModel(true);//自动lombok
        sc.setRestControllerStyle(true);// 自动生成@RestController注解
        sc.setControllerMappingHyphenStyle(true);// RequestMapping映射驼峰命名

        sc.setLogicDeleteFieldName("deleted");//设置逻辑删除

        //设置自动填充配置,需要搭配配置类MetaObjectHandler
        TableFill gmt_create = new TableFill("ts", FieldFill.INSERT); // 插入时填充
        TableFill gmt_modified = new TableFill("ts", FieldFill.INSERT_UPDATE);// 插入和更新时填充
        ArrayList<TableFill> tableFills = new ArrayList<>();
        tableFills.add(gmt_create);
        tableFills.add(gmt_modified);
        sc.setTableFillList(tableFills);

        //乐观锁
        sc.setVersionFieldName("version");
        sc.setRestControllerStyle(true);//驼峰命名


        sc.setTablePrefix("for_"); //设置表名前缀
        //传入表名字符串数组
        sc.setInclude("for_chip_code,for_country_code".split(","));
        mpg.setStrategy(sc);

        // 生成代码
        mpg.execute();
    }
}

新版(生成器版本3.5.1及以上)

引入maven依赖

 <!-- 代码自动生成器依赖-->
        <dependency>
            <groupId>com.baomidou</groupId>
            <artifactId>mybatis-plus-generator</artifactId>
            <version>3.5.1</version>
        </dependency>
        <!-- 代码生成器模板引擎 -->
        <dependency>
            <groupId>org.apache.velocity</groupId>
            <artifactId>velocity-engine-core</artifactId>
            <version>2.0</version>
        </dependency>

在test目录下创建单元测试，并运行即可

@SpringBootTest
public class MybatisPlusGeneratorNew {

    @Test
    public void generate() {
        // 配置url账号密码
        FastAutoGenerator.create("jdbc:mysql://localhost:3306/pzp?useUnicode=true&characterEncoding=UTF-8&useSSL=false&serverTimezone=GMT", "root", "root").
                globalConfig(builder -> {
                    builder
                            // 输出路径
                            .outputDir(System.getProperty("user.dir") + "/src/main/java")
                            // 是否覆盖之前文件
                            .fileOverride()
                            // 作者
                            .author("pzp")
                            // 生成swagger注解
                            .enableSwagger()
                            // 日期类型
                            .dateType(DateType.ONLY_DATE);
                })
                .packageConfig(builder -> {
                    builder
                            // 父文件夹
                            .parent("cn.szqiancheng.admin.project")
                            // 实体类输出路径
                            .entity("pojo.entity");
                            /*以下为默认值
                            .mapper("mapper")
                            .xml("mapper.xml")
                            .service("service")
                            .serviceImpl("service.impl")
                            .controller("controller");*/
                })

                .strategyConfig(builder -> {
                    builder.entityBuilder()
                            // 实体类Id类型
                            .idType(IdType.AUTO)
                            // 数据库表映射到实体的命名策略
                            .naming(NamingStrategy.underline_to_camel)
                            // 数据库表字段映射到实体的命名策略
                            .columnNaming(NamingStrategy.underline_to_camel)
                            // 生成lombok注解
                            .enableLombok()
                            // 配置自动填充的字段
                            .addTableFills(new Property("ts", FieldFill.INSERT_UPDATE));
                    // controller上加@restController注解
                    builder.controllerBuilder().enableRestStyle();
                    // 配置生成的Service的名字，不加这个配置的话，生成Service前面会有个I(例如IUserService)
                    builder.serviceBuilder().formatServiceFileName("%sService");
                    builder.mapperBuilder()
                            // 在xml文件生成BaseColumnList
                            .enableBaseColumnList()
                            // 在mapper上生成@Mapper注解
                            .enableMapperAnnotation();
                    builder //取出表前缀
                            .addTablePrefix("for_")
                            //需要自动生成代码的表
                            .addInclude("for_chip_code,for_country_code".split(","));
                })
                .execute();
    }
}

---

以上内容本人亲测有效，如果需要了解详细配置的，可以去mybatis官网进行学习，觉得有帮助的话，请帮我点下小赞赞~