MybatisPlus中的代码生成器

た入场券 2022-12-30 10:38 168阅读 0赞

### 代码自动生成器： ###

**1、依赖：**

<dependencies>
    	
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-web</artifactId>
        </dependency>
        
        <dependency>
            <groupId>mysql</groupId>
            <artifactId>mysql-connector-java</artifactId>
        </dependency>
        
        <dependency>
            <groupId>org.projectlombok</groupId>
            <artifactId>lombok</artifactId>
        </dependency>
        
        <dependency>
            <groupId>com.baomidou</groupId>
            <artifactId>mybatis-plus-boot-starter</artifactId>
            <version>3.4.1</version>
        </dependency>
        
        <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.1</version>
        </dependency>
        
        <dependency>
            <groupId>io.swagger</groupId>
            <artifactId>swagger-annotations</artifactId>
            <version>1.5.21</version>
        </dependency>
    	
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-test</artifactId>
            <scope>test</scope>
            <exclusions>
                <exclusion>
                    <groupId>org.junit.vintage</groupId>
                    <artifactId>junit-vintage-engine</artifactId>
                </exclusion>
            </exclusions>
        </dependency>
    </dependencies>

--------------------

**2、生成器代码：**

package com.pipi.mybatisplus02;
    
    import com.baomidou.mybatisplus.annotation.DbType;
    import com.baomidou.mybatisplus.annotation.FieldFill;
    import com.baomidou.mybatisplus.annotation.IdType;
    import com.baomidou.mybatisplus.generator.AutoGenerator;
    import com.baomidou.mybatisplus.generator.config.DataSourceConfig;
    import com.baomidou.mybatisplus.generator.config.GlobalConfig;
    import com.baomidou.mybatisplus.generator.config.PackageConfig;
    import com.baomidou.mybatisplus.generator.config.StrategyConfig;
    import com.baomidou.mybatisplus.generator.config.po.TableFill;
    import com.baomidou.mybatisplus.generator.config.rules.DateType;
    import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
    
    import java.util.ArrayList;
    import java.util.List;
    
    // 代码自动生成器
    public class InitCode { 
        public static void main(String[] args) { 
    
            // 实例一个代码自动生成器对象
            AutoGenerator mpg = new AutoGenerator();
            // 开始配置各种策略配置，并丢到自动生成器中
    
            // 1、全局配置
            GlobalConfig gc = new GlobalConfig();
            String projectPath = System.getProperty("user.dir");  //当前工程路径
            gc.setOutputDir(projectPath + "/src/main/java");  // 代码都放在该路径下
            gc.setAuthor("guanzhizhen");  // 设置生成代码的作者
            gc.setOpen(false);  // 生成后不打开文件管理器
            gc.setFileOverride(false);  // 是否覆盖已经存在的文件
            gc.setServiceName("%sService");  // 去除Service层接口的I前缀
            gc.setIdType(IdType.AUTO);  // 设置id自动生成策略
            gc.setDateType(DateType.ONLY_DATE);  // 日期类型设置为仅Date
            gc.setSwagger2(true);  // 实体属性，用Swagger2注解
            // 将全局配置丢到生成器对象中
            mpg.setGlobalConfig(gc);
    
            // 2、设置数据源
            DataSourceConfig dsc = new DataSourceConfig();
            dsc.setUrl("jdbc:mysql://localhost:3306/mybatis_plus?useSSL=false&useUnicode=true&characterEncoding=utf-8&serverTimezone=Asia/Shanghaijdbc:mysql://localhost:3306/mybatis_plus?useSSL=false&useUnicode=true&characterEncoding=utf-8&serverTimezone=Asia/Shanghai");
            dsc.setUsername("root");
            dsc.setPassword("666666");
            dsc.setDriverName("com.mysql.cj.jdbc.Driver");
            dsc.setDbType(DbType.MYSQL);  // 数据库的类型
            // 将数据源丢到自动生成器中
            mpg.setDataSource(dsc);
    
            // 3、配置包的信息，即要生成哪些包结构
            PackageConfig pc = new PackageConfig();
            pc.setParent("com.pipi");  // 放在哪个父包下，一般为公司域名倒过来写
            pc.setModuleName("mybatisplus02");  // 生成的模块包名字，它的父包是上面写的parent
            pc.setEntity("entity");  // 实体类包名
            pc.setMapper("mapper");  // Dao层接口包名
            pc.setService("service");  // Service层接口包名
            pc.setServiceImpl("service.impl");  // Service层接口实现类包名，默认不写也是这个
            pc.setController("controller");  // Web层控制器类包名
            // 将配置的包信息丢到自动生成器中
            mpg.setPackageInfo(pc);
    
            // 4、策略配置
            StrategyConfig strategy = new StrategyConfig();
            strategy.setInclude("user");  // 设置要映射的表名，可以填多个，参数是不定长参数，这个表名将作为实体类名
    strategy.setTablePrefix("tbl_");  // 设置表的前缀，这样生成实体类，和其他的类时，就不会带上tbl_
            strategy.setNaming(NamingStrategy.underline_to_camel);  // 支持属性下划线转驼峰命名
            strategy.setColumnNaming(NamingStrategy.underline_to_camel);  //支持列字段下划线转驼峰命名
            //strategy.setSuperEntityClass("你自己的父类实体,没有就不用设置!");
            strategy.setEntityLombokModel(true);  // 实体类使用Lombok注解
            strategy.setRestControllerStyle(true);  // 开启Restful驼峰命名
            strategy.setLogicDeleteFieldName("deleted");  // 逻辑删除字段的名字
            strategy.setControllerMappingHyphenStyle(true);  // 支持url映射风格，localhost:8080/hello_id_2
            // 自动填充策略
            TableFill version = new TableFill("version", FieldFill.INSERT);
            TableFill deleted = new TableFill("deleted", FieldFill.INSERT);
            TableFill createTime = new TableFill("create_time", FieldFill.INSERT);
            TableFill updateTime = new TableFill("update_time", FieldFill.INSERT_UPDATE);
            List<TableFill> tableFillList = new ArrayList<>();
            tableFillList.add(version);
            tableFillList.add(deleted);
            tableFillList.add(createTime);
            tableFillList.add(updateTime);
            // 将自动填充策略丢到策略配置中
            strategy.setTableFillList(tableFillList);
            // 配置乐观锁
            strategy.setVersionFieldName("version");  // 乐观锁属性名
            // 将策略配置丢到自动生成器中
            mpg.setStrategy(strategy);
    
            // 最后，执行代码构造器
            mpg.execute();
        }
    
    }

--------------------

只需要有数据库表，然后在空项目中运行此代码。

测试后查看各类的代码。该删除的删除，该添加的添加。  
不一定生成的特别完美，需要你按照规范来。

--------------------