MP(MyBaitsPlus)代码生成器

一、导入依赖

        <!-- velocity 模板引擎, Mybatis Plus 代码生成器需要 -->
        <dependency>
            <groupId>org.apache.velocity</groupId>
            <artifactId>velocity-engine-core</artifactId>
            <version>2.0</version>
        </dependency>

        <!--mysql-->
        <dependency>
            <groupId>mysql</groupId>
            <artifactId>mysql-connector-java</artifactId>
        </dependency>

        <!--mybatis-plus-->
        <dependency>
            <groupId>com.baomidou</groupId>
            <artifactId>mybatis-plus-boot-starter</artifactId>
            <version>3.0.5</version>
        </dependency>

        <!--lombok用来简化实体类：需要安装lombok插件-->
        <dependency>
            <groupId>org.projectlombok</groupId>
            <artifactId>lombok</artifactId>
        </dependency>

        <!--swagger-->
        <dependency>
            <groupId>io.springfox</groupId>
            <artifactId>springfox-swagger2</artifactId>
            <version>2.7.0</version>
        </dependency>
        <dependency>
            <groupId>io.springfox</groupId>
            <artifactId>springfox-swagger-ui</artifactId>
            <version>2.7.0</version>
        </dependency>

二、数据库

demo_db 库 user 表

三、实现代码

@SpringBootTest
class DemoApplicationTests {

    @Test
    void contextLoads() {
        // 创建代码生成器
        AutoGenerator autoGenerator = new AutoGenerator();
        // 全局配置
        GlobalConfig globalConfig = new GlobalConfig();
        String projectPath = System.getProperty("user.dir");
        globalConfig.setOutputDir(projectPath + "/src/main/java");
        globalConfig.setAuthor("aword");
        // 生成后是否打开资源管理器
        globalConfig.setOpen(false);
        // 重新生成时文件是否覆盖
        globalConfig.setFileOverride(false);
        // mp生成service层代码，默认接口名称第一个字母有I
        // 去掉Service接口的首字母I
        globalConfig.setServiceName("%sService");
        // 主键策略
        globalConfig.setIdType(IdType.ID_WORKER);
        // 定义生成的实体类中日期类型
        globalConfig.setDateType(DateType.ONLY_DATE);
        // 开启Swagger2模式
        globalConfig.setSwagger2(true);
        autoGenerator.setGlobalConfig(globalConfig);
        // 数据源配置
        DataSourceConfig dataSourceConfig = new DataSourceConfig();
        dataSourceConfig.setUrl("jdbc:mysql://localhost:3306/demo_db?serverTimezone=GMT%2B8");
        dataSourceConfig.setDriverName("com.mysql.cj.jdbc.Driver");
        dataSourceConfig.setUsername("root");
        dataSourceConfig.setPassword("root");
        dataSourceConfig.setDbType(DbType.MYSQL);
        autoGenerator.setDataSource(dataSourceConfig);
        // 生成代码包配置
        PackageConfig packageConfig = new PackageConfig();
        // 模块包，controller,entity,mapper,service等包的上一级
        packageConfig.setModuleName("demo");
        // 模块包的上一级
        packageConfig.setParent("com.aword");
        packageConfig.setController("controller");
        packageConfig.setEntity("entity");
        packageConfig.setService("service");
        packageConfig.setMapper("mapper");
        autoGenerator.setPackageInfo(packageConfig);
        // 策略配置
        StrategyConfig strategyConfig = new StrategyConfig();
        // 数据库表名
        strategyConfig.setInclude("user");
        // 数据库表映射到实体的命名策略
        strategyConfig.setNaming(NamingStrategy.underline_to_camel);
        // 生成实体时去掉表前缀
        strategyConfig.setTablePrefix(packageConfig.getModuleName() + "_");
        // 数据库表字段映射到实体的命名策略
        strategyConfig.setColumnNaming(NamingStrategy.underline_to_camel);
        // lombok 模型 @Accessors(chain = true) setter链式操作
        strategyConfig.setEntityLombokModel(true);
        // restful api风格控制器
        strategyConfig.setRestControllerStyle(true);
        // url中驼峰转连字符
        strategyConfig.setControllerMappingHyphenStyle(true);
        // 设置策略配置
        autoGenerator.setStrategy(strategyConfig);
        // 执行
        autoGenerator.execute();
        System.out.println("生成代码成功......");
    }
}

生成目录