mybatisplus-generator

 

import com.baomidou.mybatisplus.annotation.DbType;
import com.baomidou.mybatisplus.annotation.IdType;
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.po.TableInfo;
import com.baomidou.mybatisplus.generator.config.rules.DateType;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;

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

/**
 * @Author: 
 * @CreateTime: 2020/10/26 15:24
 * @Description: mybatis-plus-generator 代码生成器
 */
public class GeneratorCode {

    public static void main(String[] args) {

        // 1、全局配置         
        String projectPath = System.getProperty("user.dir");
        GlobalConfig globalConfig = new GlobalConfig();
        globalConfig.setActiveRecord(true)// 是否支持AR模式
                .setOutputDir(projectPath + "/src/main/java")
                .setAuthor("test")
                .setOpen(false)
                .setFileOverride(false)// 文件是否覆盖
                .setServiceName("%sService")// 去Service接口名称的I前缀
                .setIdType(IdType.AUTO)// 主键生成策略
                .setDateType(DateType.ONLY_DATE)
                .setSwagger2(false)
                .setBaseResultMap(true)//基本的resultMap对象
                .setBaseColumnList(true);//基本的 sql对象


        //2、设置数据源
        DataSourceConfig dataSource = new DataSourceConfig();
        dataSource.setDbType(DbType.MYSQL) // 设置数据源类型
                .setDriverName("com.mysql.cj.jdbc.Driver")
                .setUrl("jdbc:mysql://10.0.24.51:3306/cbb_tdf?useSSL=false&useUnicode=true&characterEncoding=utf-8&serverTimezone=GMT%2B8")
                .setUsername("root")
                .setPassword("root");

        //3、包的配置
        String moduleName = "generator";
        PackageConfig packageInfo = new PackageConfig();
        packageInfo.setModuleName(moduleName)
                .setParent("cn.com.mhpt.modules")
                .setModuleName(moduleName)
                .setEntity("domain")
                .setMapper("repository")
                .setService("service")
                .setController("rest");


        //4、策略配置
        StrategyConfig strategy = new StrategyConfig();
        strategy.setCapitalMode(true)// 全局大小写 命名
                .setNaming(NamingStrategy.underline_to_camel)// 数据表转驼峰的命名策略
                .setColumnNaming(NamingStrategy.underline_to_camel)
                .setEntityLombokModel(true) // 自动lombok；
                .setLogicDeleteFieldName("deleted")
                .setTablePrefix("t_mhpt")// 表前缀
                .setInclude("user_info") // 逆向工程生成的表 多张表： "t_ygpt_firewall", "user_info"
                .setVersionFieldName("version") // 乐观锁  
                .setRestControllerStyle(true)
                .setControllerMappingHyphenStyle(true);

        // 5.需要构建一个 代码自动生成器 对象        
        AutoGenerator autoGenerator = new AutoGenerator();
        autoGenerator.setGlobalConfig(globalConfig)
                .setDataSource(dataSource)
                .setPackageInfo(packageInfo)
                .setCfg(initInjectionConfig(projectPath, moduleName))
                .setTemplate(initTemplateConfig())
                .setStrategy(strategy)
                .execute(); //执行  
    }


    /**
     * 初始化自定义配置
     */
    private static InjectionConfig initInjectionConfig(String projectPath, String moduleName) {
        // 自定义配置
        InjectionConfig injectionConfig = new InjectionConfig() {
            @Override
            public void initMap() {
                // 可用于自定义属性
            }
        };
        // 模板引擎是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/mysql/mhpt/" + moduleName
                        + "/" + tableInfo.getEntityName() + "Mapper" + StringPool.DOT_XML;
            }
        });
        injectionConfig.setFileOutConfigList(focList);
        return injectionConfig;
    }

    /**
     * 初始化模板配置
     */
    private static TemplateConfig initTemplateConfig() {
        TemplateConfig templateConfig = new TemplateConfig();
        //可以对controller、service、entity模板进行配置
        //mapper.xml模板需单独配置
        templateConfig.setXml(null);
        return templateConfig;
    }
}