MyBatisPlus代码生成器

import com.baomidou.mybatisplus.core.toolkit.StringPool;
import com.baomidou.mybatisplus.generator.AutoGenerator;
import com.baomidou.mybatisplus.generator.InjectionConfig;
import com.baomidou.mybatisplus.generator.config.*;
import com.baomidou.mybatisplus.generator.config.converts.OracleTypeConvert;
import com.baomidou.mybatisplus.generator.config.converts.PostgreSqlTypeConvert;
import com.baomidou.mybatisplus.generator.config.po.TableInfo;
import com.baomidou.mybatisplus.generator.config.rules.DbColumnType;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;

import java.io.File;
import java.util.ArrayList;
import java.util.List;

/**
 * 代码生成器
 * <p>当前使用 freemarker 引擎需要 pom 依赖</p>
 *
 * @author maxzhao
 * @date 2021-04-08
 */
public class CodeGenerator {
    /**
     * 文件存放位置
     * <p>当前为当前项目位置</p>
     */
    private static final String PROJECT_PATH = System.getProperty("user.dir");
    /**
     * 包的起始路径
     */
    private static final String PACKAGE_PARENT = "boot";
    /**
     * 模块名
     */
    private static final String MODULE_NAME = "chats";
    /**
     * 文件注释下的 auther 名称
     */
    public static final String AUTHOR = "maxzhao";
    /**
     * 表名
     */
    private static final String[] TABLE_NAMES = "chat_group,chat_group_user,chat_history,chat_history_status,chat_session,chat_user,chat_user_grouping,chat_user_grouping_user".split(",");

    /**
     * 连接地址
     * oracle = "jdbc:mysql://127.0.0.1:3306/boot?useUnicode=true&characterEncoding=utf8&serverTimezone=GMT%2B8&useSSL=false&allowPublicKeyRetrieval=true";
     * postgresql = "jdbc:postgresql://127.0.0.1:5432/postgres?charSet=utf8&currentSchema=pg_schema";
     * mysql8 = "jdbc:oracle:thin:@32.1.6.219:1521/orcl";
     */
    private static final String DS_URL = "jdbc:mysql://127.0.0.1:3306/boot?useUnicode=true&characterEncoding=utf8&serverTimezone=GMT%2B8&useSSL=false&allowPublicKeyRetrieval=true";
    /**
     * 驱动
     * oracle = "oracle.jdbc.driver.OracleDriver";
     * postgresql = "org.postgresql.Driver";
     * mysql8 = "com.mysql.cj.jdbc.Driver";
     */
    private static final String DS_DRIVER_NAME = "com.mysql.cj.jdbc.Driver";
    private static final String DS_USERNAME = "root";
    private static final String DS_PASSWORD = "maxzhao";

    /**
     * PostgreSql 的 schema（其它数据库不需要）
     */
    public static final String POSTGRESQL_SCHEMA = "pg_schema";

    /**
     * RUN
     */
    public static void main(String[] args) {
        codeGenerate();
    }

    /**
     * 代码生成器
     */
    private static void codeGenerate() {
        /*代码生成器*/
        AutoGenerator mpg = new AutoGenerator();
        /*全局配置*/
        mpg.setGlobalConfig(getGlobalConfig());
        /*数据源配置*/
        mpg.setDataSource(getDataSourceConfig());
        /*跟包相关的配置项*/
        mpg.setPackageInfo(getPackageConfig());
        /*自定义配置*/
        mpg.setCfg(getInjectionConfig());
        mpg.setTemplate(new TemplateConfig().setXml(null));
        /*策略配置*/
        mpg.setStrategy(getStrategyConfig());
        // 选择 freemarker 引擎需要指定如下加，注意 pom 依赖必须有！
        mpg.setTemplateEngine(new FreemarkerTemplateEngine());
        mpg.execute();
    }

    /**
     * 全局配置
     */
    private static GlobalConfig getGlobalConfig() {
        GlobalConfig gc = new GlobalConfig();
        /*生成文件的输出目录【默认 D 盘根目录】*/
        gc.setOutputDir(PROJECT_PATH + File.separator + "src" + File.separator + "main" + File.separator + "java");
        /*是否覆盖已有文件*/
        gc.setFileOverride(false);
        /*是否打开输出目录*/
        gc.setOpen(false);
        gc.setAuthor(AUTHOR);
        gc.setSwagger2(true);
        return gc;
    }

    /**
     * 数据库配置
     */
    private static DataSourceConfig getDataSourceConfig() {
        DataSourceConfig dsc = new DataSourceConfig();
        dsc.setUrl(DS_URL);
        dsc.setDriverName(DS_DRIVER_NAME);
        dsc.setUsername(DS_USERNAME);
        dsc.setPassword(DS_PASSWORD);
        if (DS_DRIVER_NAME.contains("postgresql")) {
            initPGSqlDataSourceConfig(dsc);
        } else if (DS_DRIVER_NAME.contains("mysql")) {
        } else if (DS_DRIVER_NAME.contains("oracle")) {
            initOracleDataSourceConfig(dsc);
        }
        return dsc;
    }

    /**
     * PostgreSQL数据库连接
     */
    private static void initPGSqlDataSourceConfig(DataSourceConfig dsc) {
        dsc.setSchemaName(POSTGRESQL_SCHEMA);
        dsc.setTypeConvert(new PostgreSqlTypeConvert());
    }

    /**
     * ORACLE 数据库连接
     */
    private static void initOracleDataSourceConfig(DataSourceConfig dsc) {
        dsc.setTypeConvert(new OracleTypeConvert() {
            @Override
            public DbColumnType processTypeConvert(GlobalConfig globalConfig, String fieldType) {
                /*可以参考 super.processTypeConvert*/
                return (DbColumnType) super.processTypeConvert(globalConfig, fieldType);
            }
        });
    }

    /**
     * 跟包相关的配置项
     */
    private static PackageConfig getPackageConfig() {
        PackageConfig pc = new PackageConfig();
        /*父包名。如果为空，将下面子包名必须写全部， 否则就只需写子包名*/
        pc.setParent(PACKAGE_PARENT);
        /*父包模块名*/
        pc.setModuleName(MODULE_NAME);
        /*Entity包名*/
        pc.setEntity("model.entity");
        /*Controller包名*/
        pc.setController("api");
        return pc;
    }

    /**
     * 自定义配置
     */
    private static InjectionConfig getInjectionConfig() {
        InjectionConfig cfg = new InjectionConfig() {
            @Override
            public void initMap() {
                // to do nothing
            }
        };
        List<FileOutConfig> focList = new ArrayList<>();
        /*配置 Mapper.xml的生成位置*/
        focList.add(new FileOutConfig("/templates/mapper.xml.ftl") {
            @Override
            public String outputFile(TableInfo tableInfo) {
                // 自定义输入文件名称
                return PROJECT_PATH + File.separator +
                        "src" + File.separator +
                        "main" + File.separator +
                        "resources" + File.separator +
                        PACKAGE_PARENT.replace(".", File.separator) +
                        File.separator +
                        MODULE_NAME.replace(".", File.separator) +
                        File.separator + "mapper" + File.separator +
                        tableInfo.getEntityName() + "Mapper" + StringPool.DOT_XML;
            }
        });
        cfg.setFileOutConfigList(focList);
        return cfg;
    }

    /**
     * 策略配置
     */
    private static StrategyConfig getStrategyConfig() {
        StrategyConfig strategy = new StrategyConfig();
        /*数据库表映射到实体的命名策略,下划线转驼峰*/
        strategy.setNaming(NamingStrategy.underline_to_camel);
        /*数据库表字段映射到实体的命名策略，未指定按照 naming 执行*/
        strategy.setColumnNaming(NamingStrategy.underline_to_camel);
        /*表前缀*/
        strategy.setTablePrefix("t_", "sys_", "p_");
        /*字段前缀*/
        strategy.setFieldPrefix("is", "has");
        /*需要包含的表名，允许正则表达式（与exclude二选一配置）*/
        strategy.setInclude(TABLE_NAMES);
        /*实体是否生成 serialVersionUID*/
        strategy.setEntitySerialVersionUID(true);
        /*【实体】是否为lombok模型（默认 false）*/
        strategy.setEntityLombokModel(true);
        /*Boolean类型字段是否移除is前缀（默认 false） 比如 : 数据库字段名称 : 'is_xxx',类型为 : tinyint.
        在映射实体的时候则会去掉is,在实体类中映射最终结果为 xxx*/
        strategy.setEntityBooleanColumnRemoveIsPrefix(true);
        /*是否生成实体时，生成字段注解*/
        strategy.setEntityTableFieldAnnotationEnable(true);
        /*生成 @RestController 控制器*/
        strategy.setRestControllerStyle(true);
        /*驼峰转连字符*/
        strategy.setControllerMappingHyphenStyle(false);
        /*是否生成实体时，生成字段注解*/
        strategy.setEntityTableFieldAnnotationEnable(true);
        return strategy;
    }
}