Mybatis-plus代码生成器模板与使用注意事项

Mybatis-plus代码生成器模板与使用注意事项

Mybatis-plus为我们提升了开发效率，还可以简化我们的操作，例如：Mybatis-plus的代码生成器

Mybatis-plus的代码生成器比官方Mybatis的生成器还要强大

接下来我将提供对官方文档的代码生成器进行一定修改的模板提供使用与说明注意事项

需要的jar包

必要jar包

mysql-connector-java

mybatis-plus

mybatis-plus-generator

freemarker

slf4j-log4j12

非必要jar包

junit

lombok

---

使用代码生成器快速搭建项目

默认已经创建了Maven新项目

1. 导入jar依赖包，以我的为例在dependencies标签导入

   
           <dependency>
               <groupId>mysqlgroupId>
               <artifactId>mysql-connector-javaartifactId>
               <version>8.0.18version>
           dependency>
           
           <dependency>
               <groupId>com.baomidougroupId>
               <artifactId>mybatis-plusartifactId>
               <version>3.3.1version>
           dependency>
           
           <dependency>
               <groupId>com.baomidougroupId>
               <artifactId>mybatis-plus-generatorartifactId>
               <version>3.3.1version>
           dependency>
   
           <dependency>
               <groupId>org.freemarkergroupId>
               <artifactId>freemarkerartifactId>
               <version>2.3.29version>
           dependency>
           
           <dependency>
               <groupId>junitgroupId>
               <artifactId>junitartifactId>
               <version>4.12version>
           dependency>
           
           <dependency>
               <groupId>org.projectlombokgroupId>
               <artifactId>lombokartifactId>
               <version>1.18.4version>
           dependency>
           
           <dependency>
               <groupId>org.slf4jgroupId>
               <artifactId>slf4j-log4j12artifactId>
               <version>1.6.4version>
           dependency>
   

2. 在项目下新建类运行如下代码，我的类运行在含junit依赖的Maven项目的test/java下，如果你也使用如下代码则需要修改一些属性

package com.gugu;

import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;

import com.baomidou.mybatisplus.core.exceptions.MybatisPlusException;
import com.baomidou.mybatisplus.core.toolkit.StringPool;
import com.baomidou.mybatisplus.core.toolkit.StringUtils;
import com.baomidou.mybatisplus.generator.AutoGenerator;
import com.baomidou.mybatisplus.generator.InjectionConfig;
import com.baomidou.mybatisplus.generator.config.DataSourceConfig;
import com.baomidou.mybatisplus.generator.config.FileOutConfig;
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.TemplateConfig;
import com.baomidou.mybatisplus.generator.config.po.TableInfo;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;

public class AutoGenerate {
    /**
     * 数据库名字
     */
    private final static String databaseName = "mybatisplus";
    /**
     * 连接MySQL的url,MySQL版本8及以上未在MysQL设置时区的最好在后面拼上serverTimezone参数
     */
    private final static String url = "jdbc:mysql://localhost:3306/" + databaseName + "?serverTimezone=GMT&characterEncoding=utf8";
    /**
     * MySQL版本在8以下的使用com.mysql.jdbc.Driver
     */
    private final static String driver = "com.mysql.cj.jdbc.Driver";
    /**
     * MySQL用户名
     */
    private final static String username = "root";
    /**
     * MySQL密码
     */
    private final static String password = "root";
    /**
     * 设置父包名
     */
    private final static String parentPackageName = "com.gugu";
    /**
     * 自己的父类实体,没有就不用设置
     */
    private final static String superEntityClass = "";
    /**
     * 是否开启Lombok
     */
    private final static Boolean lombokFlag = true;
    /**
     * 自己的父类控制器,没有就不用设置
     */
    private final static String superControllerClass = "";

    public static void main(String[] args) {
        String developerName = scanner("开发人员名字");
        String modelName = scanner("模块名");
        String tableName = scanner("数据库表名，多个表使用英文逗号分割");
        // 代码生成器
        AutoGenerator mpg = new AutoGenerator();
        // 全局配置
        GlobalConfig gc = new GlobalConfig();
        String projectPath = System.getProperty("user.dir");
        gc.setOutputDir(projectPath + "/src/main/java");
        gc.setAuthor(developerName);
        gc.setOpen(false);
        // gc.setSwagger2(true); 实体属性 Swagger2 注解
        mpg.setGlobalConfig(gc);

        // 数据源配置
        DataSourceConfig dsc = new DataSourceConfig();
        dsc.setUrl(url);
        // dsc.setSchemaName("public");
        dsc.setDriverName(driver);
        dsc.setUsername(username);
        dsc.setPassword(password);
        mpg.setDataSource(dsc);

        // 包配置
        PackageConfig pc = new PackageConfig();
        pc.setModuleName(modelName);
        pc.setParent(parentPackageName);
        mpg.setPackageInfo(pc);

        // 自定义配置
        InjectionConfig cfg = new InjectionConfig() {
            @Override
            public void initMap() {
                // to do nothing
            }
        };

        // 如果模板引擎是 freemarker
        String templatePath = "/templates/mapper.xml.ftl";
        // 如果模板引擎是 velocity
        // String templatePath = "/templates/mapper.xml.vm";

        // 自定义输出配置
        List<FileOutConfig> focList = new ArrayList<>();
        // 自定义配置会被优先输出
        focList.add(new FileOutConfig(templatePath) {
            @Override
            public String outputFile(TableInfo tableInfo) {
                // 自定义输出文件名 ， 如果你 Entity 设置了前后缀、此处注意 xml 的名称会跟着发生变化！！
                return projectPath + "/src/main/resources/mapper/" + pc.getModuleName()
                        + "/" + tableInfo.getEntityName() + "Mapper" + StringPool.DOT_XML;
            }
        });
        /*
        cfg.setFileCreate(new IFileCreate() {
            @Override
            public boolean isCreate(ConfigBuilder configBuilder, FileType fileType, String filePath) {
                // 判断自定义文件夹是否需要创建
                checkDir("调用默认方法创建的目录");
                return false;
            }
        });
        */
        cfg.setFileOutConfigList(focList);
        mpg.setCfg(cfg);

        // 配置模板
        TemplateConfig templateConfig = new TemplateConfig();

        // 配置自定义输出模板
        //指定自定义模板路径，注意不要带上.ftl/.vm, 会根据使用的模板引擎自动识别
        // templateConfig.setEntity("templates/entity2.java");
        // templateConfig.setService();
        // templateConfig.setController();

        templateConfig.setXml(null);
        mpg.setTemplate(templateConfig);

        // 策略配置
        StrategyConfig strategy = new StrategyConfig();
        strategy.setNaming(NamingStrategy.underline_to_camel);
        strategy.setColumnNaming(NamingStrategy.underline_to_camel);
        strategy.setSuperEntityClass(superEntityClass);
        strategy.setEntityLombokModel(lombokFlag);
        strategy.setRestControllerStyle(true);
        // 公共父类
        strategy.setSuperControllerClass(superControllerClass);
        // 写于父类中的公共字段
        strategy.setSuperEntityColumns("id");
        strategy.setInclude(tableName.split(","));
        strategy.setControllerMappingHyphenStyle(true);
        strategy.setTablePrefix(pc.getModuleName() + "_");
        mpg.setStrategy(strategy);
        mpg.setTemplateEngine(new FreemarkerTemplateEngine());
        mpg.execute();
    }

    /**
     * 根据输入的信息，提示信息并要求输入，返回输入的字符串
     *
     * @param tip 要提示输入的信息
     * @return 输入的字符串
     */
    public static String scanner(String tip) {
        Scanner scanner = new Scanner(System.in);
        System.out.println("请输入" + tip + "：");
        if (scanner.hasNext()) {
            String ipt = scanner.next();
            if (StringUtils.isNotBlank(ipt)) {
                return ipt;
            }
        }
        throw new MybatisPlusException("请输入正确的" + tip + "！");
    }
}

3. 运行过程与结果

(警告是因为没有设置log4j)

---

注意事项

1. 输入表名是必须要对应数据库中的表名，否则会报错
2. 不要忘记引入log4j

---

可能出现的问题

1. projectPath，new ArrayList<>()冒红报错

   • 使用快捷键Ctrl+Alt+Shift+s打开工程设置Project Settings->Moudues设置Language level语言等级为8

2. Error:java: 错误: 不支持发行版本 5

   • 在pom.xml中增加build，设置Maven语言编译级别为jdk1.8

     <build>
             <plugins>
                 <plugin>
                     <artifactId>maven-compiler-pluginartifactId>
                     <version>3.1version>
                     <configuration>
                         <source>1.8source>
                         <target>1.8target>
                     configuration>
                 plugin>
             plugins>
         build>
     

---

建议

如果不喜欢生成的文件名则自己手动修改