创建MybatisPlus代码生成器

叁歲伎倆 2021-09-25 06:48 511阅读 0赞

>  MybatisPlus中的代码生成器，有特定的源码，我们只需要更改相应的地方就行

### **1.首先要引入相应的依赖。** ###

<dependency>
                <groupId>org.apache.velocity</groupId>
                <artifactId>velocity-engine-core</artifactId>
            </dependency>
            
            <dependency>
                <groupId>com.baomidou</groupId>
                <artifactId>mybatis-plus-generator</artifactId>
                <version>3.0.5</version>
            </dependency>

### 2.需要改动的地方     ###

\* 1.gc.setOutputDir("F:\\\\IdeaFile\\\\IdeaFile1\\\\bishe\_parent\\\\service\\\\service\_edu" + "/src/main/java");  
\* 2.gc.setIdType(IdType.ID\_WORKER\_STR); //主键策略，根据实际是int还是string类型的  
\* 3.pc.setParent("com.zt");进行包配置  
\* 4.strategy.setInclude("edu\_teacher");改你的数据库中的表，多个表就(""," ","")

/**
     *
     *
     * 需要改动的地方
     * 1.gc.setOutputDir("F:\\IdeaFile\\IdeaFile1\\bishe_parent\\service\\service_edu" + "/src/main/java");
     * 2.gc.setIdType(IdType.ID_WORKER_STR); //主键策略，根据实际是int还是string类型的
     * 3.pc.setParent("com.zt");进行包配置
     * 4.strategy.setInclude("edu_teacher");改你的数据库中的表，多个表就(""," ","")
     */
    public class CodeGenerator {
    
        @Test
        public void run() {
    
            // 1、创建代码生成器
            AutoGenerator mpg = new AutoGenerator();
    
            // 2、全局配置
            GlobalConfig gc = new GlobalConfig();
            String projectPath = System.getProperty("user.dir");
            //更换为绝对路径
            gc.setOutputDir("F:\\IdeaFile\\IdeaFile1\\bishe_parent\\service\\service_edu" + "/src/main/java");
            gc.setAuthor("testjava");
            gc.setOpen(false); //生成后是否打开资源管理器
            gc.setFileOverride(false); //重新生成时文件是否覆盖
            gc.setServiceName("%sService");	//去掉Service接口的首字母I
            gc.setIdType(IdType.ID_WORKER_STR); //主键策略
            gc.setDateType(DateType.ONLY_DATE);//定义生成的实体类中日期类型
            gc.setSwagger2(true);//开启Swagger2模式
    
            mpg.setGlobalConfig(gc);
    
            // 3、数据源配置,mp需要单独配置数据库，与之前的application没有关系
            DataSourceConfig dsc = new DataSourceConfig();
            dsc.setUrl("jdbc:mysql://localhost:3306/bishe?useUnicode=true&characterEncoding=utf8&useSSL=false&serverTimezone=UTC");
            dsc.setDriverName("com.mysql.cj.jdbc.Driver");
            dsc.setUsername("root");
            dsc.setPassword("147258369");
            dsc.setDbType(DbType.MYSQL);
            mpg.setDataSource(dsc);
    
            // 4、包配置就是自动生成包及其路径 com.zt.eduservice
            PackageConfig pc = new PackageConfig();
            pc.setParent("com.zt");
            pc.setModuleName("edu"); //模块名
            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);//数据库表字段映射到实体的命名策略
            strategy.setEntityLombokModel(true); // lombok 模型 @Accessors(chain = true) setter链式操作
    
            strategy.setRestControllerStyle(true); //restful api风格控制器
            strategy.setControllerMappingHyphenStyle(true); //url中驼峰转连字符
    
            mpg.setStrategy(strategy);
    
    
            // 6、执行
            mpg.execute();
        }
    }

### 3.测试 ###

运行后直接根据相应的包路径自动生成代码

![20210408205629904.png][]

[20210408205629904.png]: /images/20210923/238765efa84b4b1889f3606333a32d18.png