MyBatisPlus 3.1.X代码生成器

本文介绍了一种基于数据库的代码生成器实现,通过AutoGenerator类及其配置,如DataSourceConfig、StrategyConfig、PackageConfig等,自动生成PO、Mapper、Service、Controller层代码,提高开发效率。

通过代码生成器可以根据数据库中的实体生成对应的po、mapper、service、controller层代码,可以大大提高开发效率。代码生成器的核心是AutoGenerator类中的execute方法,文章将从AutoGenerator类讲起

核心类AutoGenerator.java

public class AutoGenerator {
    private static final Logger logger = LoggerFactory.getLogger(AutoGenerator.class);
    protected ConfigBuilder config;
    protected InjectionConfig injectionConfig;
    private DataSourceConfig dataSource;
    private StrategyConfig strategy;
    private PackageConfig packageInfo;
    private TemplateConfig template;
    private GlobalConfig globalConfig;
    private AbstractTemplateEngine templateEngine;

    public void execute() {
        logger.debug("==========================准备生成文件...==========================");
        if (null == this.config) {
            this.config = new ConfigBuilder(this.packageInfo, this.dataSource, this.strategy, this.template, this.globalConfig);
            if (null != this.injectionConfig) {
                this.injectionConfig.setConfig(this.config);
            }
        }

        if (null == this.templateEngine) {
            this.templateEngine = new VelocityTemplateEngine();
        }

        this.templateEngine.init(this.pretreatmentConfigBuilder(this.config)).mkdirs().batchOutput().open();
        logger.debug("==========================文件生成完成!!!==========================");
    }
}

查看AutoGenerator类源码发现,自动生成代码的核心就是创建DataSourceConfigInjectionConfigStrategyConfigPackageConfigTemplateConfigGlobalConfig配置类,在execute方法中将根据这些配置为我们生成代码

创建DataSourceConfig类

private static DataSourceConfig buildDataSourceConfig() {
        DataSourceConfig dsc = new DataSourceConfig();
        dsc.setUrl(URL);
        // dsc.setSchemaName("public");
        dsc.setDriverName(DRIVER_NAME);
        dsc.setUsername(USERNAME);
        dsc.setPassword(PASSWORD);
        dsc.setTypeConvert(new MySqlTypeConvert() {
            @Override
            public DbColumnType processTypeConvert(GlobalConfig globalConfig, String fieldType) {
                //将数据库中timestamp转换成date
                if ( fieldType.toLowerCase().contains( "timestamp" ) ) {
                    return DbColumnType.DATE;
                }
                return (DbColumnType) super.processTypeConvert(globalConfig, fieldType);
            }
        });
        return dsc;
    }

创建StrategyConfig类

private static StrategyConfig buildStrategyConfig() {
        StrategyConfig strategy = new StrategyConfig();
        // 命名规则
        strategy.setNaming(NamingStrategy.underline_to_camel);
        strategy.setColumnNaming(NamingStrategy.underline_to_camel);

        // 默认生成的po类不继承,手动修改继承
        //strategy.setSuperEntityClass((String) null);
        // 实体是否使用Lombok插件
        strategy.setEntityLombokModel(false);
        // 自定义 service 父类
        strategy.setSuperServiceClass("com.h2t.study.BaseService");
        // 自定义 service 实现类父类
        strategy.setSuperServiceImplClass("com.h2t.study.BaseServiceImpl");
        // 控制层是否使用Rest风格
        strategy.setRestControllerStyle(true);
        return strategy;
    }

这里自定义Service以及serviceImpl的父类,BaseServiceImpl对原有的父类进行了封装,封装目的:

  • 不需要在每个业务实现类中都创建Wrapper,代码更通用
  • 可以对业务层进行一个统一的校验拦截,可以通过拦截BaseServiceImpl做一个通用的拦截校验

创建PackageConfig类

private static PackageConfig buildPackageConfig() {
        PackageConfig pc = new PackageConfig();
        //pc.setModuleName(scanner("模块名"));
        pc.setParent(PACKAGE_NAME);
        pc.setEntity("po");
        pc.setXml("mapper");
        pc.setController("controller");
        pc.setService("service");
        return pc;
    }

创建GlobalConfig类

 private static GlobalConfig buildGlobalConfig() {
        // 全局配置
        GlobalConfig gc = new GlobalConfig();
        //项目所在地址
        gc.setOutputDir(OUTPUT_DIR);
        //注释作者
        gc.setAuthor(AUTHOR);
        //生成文件不打开
        gc.setOpen(false);
        gc.setFileOverride(true);
        gc.setActiveRecord(true);
        // XML 二级缓存
        gc.setEnableCache(false);
        //生成result map
        // XML ResultMap
        gc.setBaseResultMap(true);
        //生成java mysql字段映射
        // XML columList
        gc.setBaseColumnList(true);
        // gc.setSwagger2(true); 实体属性 Swagger2 注解
        // 自定义文件命名,注意 %s 会自动填充表实体属性!
        gc.setMapperName("%sMapper");
        gc.setXmlName("%sMapper");
        gc.setServiceName("%sService");
        gc.setServiceImplName("%sServiceImpl");
        gc.setControllerName("%sController");
        return gc;
    }

创建InjectionConfig类

 InjectionConfig cfg = new InjectionConfig() {
            @Override
            public void initMap() {
                // to do nothing
            }
        };

TemplateConfig

TemplateConfig templateConfig = new TemplateConfig();

完整代码生成器代码

public class MpGenerator {
    /**
     * 包名
     */
    private static final String PACKAGE_NAME = "com.h2t.study";

    /**
     * 代码生成路径
     */
    private static final String OUTPUT_DIR = "C:\\Users\\hetiantian\\Desktop\\test";

    /**
     * 代码注释作者
     */
    private static final String AUTHOR = "hetiantian";

    private static final String DRIVER_NAME = "com.mysql.cj.jdbc.Driver";
    private static final String HOST = "localhost";
    private static final String PORT = "3306";
    /**
     * 数据库信息
     */
    private static final String DATABASE = "test";
    private static final String USERNAME = "root";
    private static final String PASSWORD = "123456";
    private static final String URL = "jdbc:mysql://" + HOST + ":" + PORT + "/" + DATABASE
            + "?characterEncoding=UTF8&serverTimezone=UTC";

    public static void main(String[] args) {
        // 代码生成器
        AutoGenerator mpg = new AutoGenerator();
        mpg.setGlobalConfig(buildGlobalConfig());
        // 数据源配置
        mpg.setDataSource(buildDataSourceConfig());
        // 包配置
        mpg.setPackageInfo(buildPackageConfig());
        // 自定义配置
        InjectionConfig cfg = new InjectionConfig() {
            @Override
            public void initMap() {
                // to do nothing
            }
        };
        mpg.setCfg(cfg);
        // 配置模板
        TemplateConfig templateConfig = new TemplateConfig();
        //不生成mapper xml文件
        //templateConfig.setXml(null);
        mpg.setTemplate(templateConfig);
        // 策略配置
        mpg.setStrategy(buildStrategyConfig());
        mpg.setTemplateEngine(new FreemarkerTemplateEngine());
        mpg.execute();
    }

    /**
     * 全局构造配置类
     *
     * @return
     * */
    private static GlobalConfig buildGlobalConfig() {
        // 全局配置
        GlobalConfig gc = new GlobalConfig();
        //项目所在地址
        gc.setOutputDir(OUTPUT_DIR);
        //注释作者
        gc.setAuthor(AUTHOR);
        //生成文件不打开
        gc.setOpen(false);
        gc.setFileOverride(true);
        gc.setActiveRecord(true);
        // XML 二级缓存
        gc.setEnableCache(false);
        //生成result map
        // XML ResultMap
        gc.setBaseResultMap(true);
        //生成java mysql字段映射
        // XML columList
        gc.setBaseColumnList(true);
        // gc.setSwagger2(true); 实体属性 Swagger2 注解
        // 自定义文件命名,注意 %s 会自动填充表实体属性!
        gc.setMapperName("%sMapper");
        gc.setXmlName("%sMapper");
        gc.setServiceName("%sService");
        gc.setServiceImplName("%sServiceImpl");
        gc.setControllerName("%sController");
        return gc;
    }

    /**
     * 数据库配置信息
     *
     * @return
     * */
    private static DataSourceConfig buildDataSourceConfig() {
        DataSourceConfig dsc = new DataSourceConfig();
        dsc.setUrl(URL);
        // dsc.setSchemaName("public");
        dsc.setDriverName(DRIVER_NAME);
        dsc.setUsername(USERNAME);
        dsc.setPassword(PASSWORD);
        dsc.setTypeConvert(new MySqlTypeConvert() {
            @Override
            public DbColumnType processTypeConvert(GlobalConfig globalConfig, String fieldType) {
                //将数据库中timestamp转换成date
                if ( fieldType.toLowerCase().contains( "timestamp" ) ) {
                    return DbColumnType.DATE;
                }
                return (DbColumnType) super.processTypeConvert(globalConfig, fieldType);
            }
        });
        return dsc;
    }

    private static PackageConfig buildPackageConfig() {
        PackageConfig pc = new PackageConfig();
        //pc.setModuleName(scanner("模块名"));
        pc.setParent(PACKAGE_NAME);
        pc.setEntity("po");
        pc.setXml("mapper");
        pc.setController("controller");
        pc.setService("service");
        return pc;
    }

    private static StrategyConfig buildStrategyConfig() {
        StrategyConfig strategy = new StrategyConfig();
        // 命名规则
        strategy.setNaming(NamingStrategy.underline_to_camel);
        strategy.setColumnNaming(NamingStrategy.underline_to_camel);

        // 默认生成的po类不继承,手动修改继承
        //strategy.setSuperEntityClass((String) null);
        // 实体是否使用Lombok插件
        strategy.setEntityLombokModel(false);
        // 自定义 service 父类
        strategy.setSuperServiceClass("com.h2t.study.BaseService");
        // 自定义 service 实现类父类
        strategy.setSuperServiceImplClass("com.h2t.study.BaseServiceImpl");
        // 控制层是否使用Rest风格
        strategy.setRestControllerStyle(true);
        return strategy;
    }
}

所需依赖

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

    <dependency>
      <groupId>com.baomidou</groupId>
      <artifactId>mybatis-plus-generator</artifactId>
      <version>3.1.2</version>
    </dependency>

    <dependency>
      <groupId>mysql</groupId>
      <artifactId>mysql-connector-java</artifactId>
      <version>6.0.6</version>
    </dependency>

    <!-- 模板引擎mybaties-plus  代码生成 -->
 
    <dependency>
      <groupId>org.freemarker</groupId>
      <artifactId>freemarker</artifactId>
      <version>2.3.23</version>
    </dependency>

目前代码生成器的不足

many errors, many bugs

  • 生成的po自动会继承Model类
  • 生成的po类缺少@TableName@TableField等注解与数据库字段进行映射
    这些不足需要手动进行修改。聪明的网友如果知道如果修改,请一定要告诉我!!!
最后附完整项目地址:https://github.com/TiantianUpup/mp-generator
<think>好的,用户想了解如何配置和使用MyBatis Plus代码生成器。首先,我需要回忆一下相关的步骤和注意事项。根据提供的引用内容,特别是引用[2],里面提到了创建Spring Boot项目、添加依赖、配置数据库连接、创建配置类等步骤。 首先,用户可能需要从零开始配置,所以第一步应该是创建Spring Boot项目。这时候需要确认用户是否已经熟悉如何创建,可能需要简要提及使用Spring Initializr或者IDE工具。然后添加必要的依赖,包括MyBatis Plus和代码生成器的依赖,这里要确保版本兼容性,但用户可能不需要太详细,除非有特殊版本需求。 接下来是配置数据库连接,这部分需要修改application.yml或application.properties文件,填写正确的数据库URL、用户名和密码。用户可能会遇到驱动类的问题,特别是如果用的是MySQL 8.x,驱动类应该是com.mysql.cj.jdbc.Driver,而旧版可能不同,需要提醒注意。 然后是代码生成器的配置类,这里需要详细说明如何编写配置类,包括包路径、全局配置、数据源配置、策略配置等。这部分是核心,用户可能需要具体的代码示例,比如如何设置作者、输出目录、开启Swagger或Lombok支持。此外,策略配置中的表名和字段名的策略需要解释清楚,避免用户生成代码时出现命名问题。 模板引擎部分,用户可能不知道需要添加Freemarker或Velocity依赖,所以需要明确说明,并给出依赖的示例。同时,运行代码生成器后,用户需要知道生成的代码位置,以及如何将这些代码整合到项目中。 可能还需要考虑用户在实际操作中遇到的常见问题,比如依赖冲突、配置文件路径错误、表名大小写敏感等。但根据用户的问题,可能不需要深入这些,但需要确保步骤清晰,避免遗漏关键点。 最后,用户可能需要生成后的代码结构和使用方法,比如如何注入Mapper进行数据库操作。此外,引用中的示例代码生成器配置类需要正确展示,确保语法正确,特别是使用正确的注释和配置方法。 总结一下,回答应该分步骤,先项目创建和依赖添加,然后是配置文件,接着是配置类的编写,最后是运行生成器。每个步骤需要简明扼要,必要时给出代码示例,并提醒可能的注意事项,比如版本和驱动类的问题。同时,根据用户提供的引用内容,确保引用标识正确添加,如引用[^2]在适当的位置。</think>以下是MyBatis Plus代码生成器的配置和使用方法详解: ### 一、环境准备 1. 创建Spring Boot项目 使用IDEA或Spring Initializr创建项目,选择Web、MyBatis Plus、数据库驱动(MySQL/Oracle等)依赖 2. 添加依赖(pom.xml) ```xml <dependency> <groupId>com.baomidou</groupId> <artifactId>mybatis-plus-boot-starter</artifactId> <version>3.5.3.1</version> </dependency> <dependency> <groupId>com.baomidou</groupId> <artifactId>mybatis-plus-generator</artifactId> <version>3.5.3.1</version> </dependency> <!-- 模板引擎 --> <dependency> <groupId>org.freemarker</groupId> <artifactId>freemarker</artifactId> </dependency> ``` ### 二、配置文件 在`application.yml`中配置数据库连接: ```yaml spring: datasource: driver-class-name: com.mysql.cj.jdbc.Driver url: jdbc:mysql://localhost:3306/your_db?useSSL=false username: root password: 123456 ``` ### 三、生成器配置类 创建`CodeGenerator.java`类: ```java public class CodeGenerator { public static void main(String[] args) { AutoGenerator generator = new AutoGenerator(); // 全局配置 GlobalConfig gc = new GlobalConfig.Builder() .author("YourName") .outputDir(System.getProperty("user.dir") + "/src/main/java") .enableSwagger() .build(); // 数据源配置 DataSourceConfig dsc = new DataSourceConfig.Builder( "jdbc:mysql://localhost:3306/your_db", "root", "123456") .build(); // 包配置 PackageConfig pc = new PackageConfig.Builder() .parent("com.example") .moduleName("system") .entity("entity") .mapper("mapper") .build(); // 策略配置 StrategyConfig strategy = new StrategyConfig.Builder() .addInclude("user", "order") // 要生成的表名 .entityBuilder() .enableLombok() .naming(NamingStrategy.underline_to_camel) .build(); generator.global(gc) .dataSource(dsc) .packageInfo(pc) .strategy(strategy); generator.execute(); } } ``` ### 四、运行生成器 直接运行`CodeGenerator`类的main方法,生成的代码结构为: ``` src/main/java └── com.example └── system ├── entity # 实体类 ├── mapper # Mapper接口 └── service # Service层 ``` ### 五、使用说明 1. 生成的Mapper接口已包含基本CRUD方法 2. Service层自动继承`IService`接口 3. 实体类自动添加`@TableName`注解映射数据库表 [^1][^2]
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值