Mybatis plus代码生成器详解

Mybatis plus代码生成器详解

今天在b站上和一个老师学习了Mybatis plus中的代码生成器，这位老师做了详细的解释，我在此记录一下。

大家也可以看视频去深入理解一下代码生成器详解

---

public class CodeGenerator {

    @Test
    public void run() {

        // 1、创建代码生成器
        AutoGenerator mpg = new AutoGenerator();

        // 2、全局配置
        GlobalConfig gc = new GlobalConfig();
        //得到当前文件夹的路径
        String projectPath = System.getProperty("user.dir");
        // 代码输出目录，这里将代码输出到src/main/java目录下
        gc.setOutputDir(projectPath + "/src/main/java");

        //代码生成的作者
        gc.setAuthor("shallow");
        //生成后是否打开资源管理器（生成之后文件是否展开成树结构）
        gc.setOpen(false);
        //重新生成时文件是否覆盖
        gc.setFileOverride(false);
        //去掉Service接口的首字母I，如果不加，生成的文件名前加一个大写字母I，如IUserService
        gc.setServiceName("%sService");
        //主键策略(主键如果是long类型，则是IdType.ID_WORKER。如果是char类型，则是ID_WORKER_STR)
        gc.setIdType(IdType.ID_WORKER_STR);
        //定义生成的实体类中日期类型
        gc.setDateType(DateType.ONLY_DATE);
        //开启Swagger2模式
        gc.setSwagger2(true);

        mpg.setGlobalConfig(gc);

        // 3、数据源配置（和项目（application.properties）用的不是一个配置）
        DataSourceConfig dsc = new DataSourceConfig();
        dsc.setUrl("jdbc:mysql://localhost:3306/teacher?serverTimezone=GMT%2B8");
        dsc.setDriverName("com.mysql.cj.jdbc.Driver");
        dsc.setUsername("root");
        dsc.setPassword("root");
        dsc.setDbType(DbType.MYSQL);
        mpg.setDataSource(dsc);

        // 4、包配置
        PackageConfig pc = new PackageConfig();

        // 包名 会在java目录下生成一个名为com.shallow.eduService的包
        pc.setParent("com.shallow");
        pc.setModuleName("eduService"); //模块名

        // 子包
        pc.setController("controller");
        pc.setEntity("entity");
        pc.setService("service");
        pc.setMapper("mapper");
        mpg.setPackageInfo(pc);

        // 5、策略配置（逆向工程，根据表生成代码）
        StrategyConfig strategy = new StrategyConfig();

        // 根据表的名称生成相应代码
        strategy.setInclude("edu_teacher");
        //数据库表映射到实体的命名策略
        strategy.setNaming(NamingStrategy.underline_to_camel);
        //生成实体时去掉表前缀
        strategy.setTablePrefix(pc.getModuleName() + "_");
        //生成实体时去掉表前缀
        strategy.setColumnNaming(NamingStrategy.underline_to_camel);
        // lombok 模型 @Accessors(chain = true) setter链式操作
        strategy.setEntityLombokModel(true);
        //restful api风格控制器
        strategy.setRestControllerStyle(true);
        //url中驼峰转连字符
        strategy.setControllerMappingHyphenStyle(true);

        mpg.setStrategy(strategy);


        // 6、执行
        mpg.execute();
    }
}

---

注意一点，就是那个主键生成策略gc.setIdType(IdType.ID_WORKER_STR);，大家在对表生成代码的时候要看清楚主键是什么类型的。

• 如果是long类型的，就使用这个gc.setIdType(IdType.ID_WORKER);
• 如果是char字符串类型的话，就使用这个gc.setIdType(IdType.ID_WORKER_STR);