1. 程式人生 > >mybatis-plus 程式碼生成

mybatis-plus 程式碼生成

之前採用maven-generator 生成程式碼,在配置檔案上因為從單模組到多模組時出現問題,
改用mybatis-plus自動生成程式碼。

碼雲地址:https://gitee.com/baomidou/mybatis-plus

github地址:https://github.com/baomidou/mybatis-plus

依賴

<dependency>
    <groupId>com.baomidou</groupId>
    <artifactId>mybatis-plus</artifactId>
    <version>3.0.6</version>
</dependency>
<dependency>
    <groupId>org.apache.velocity</groupId>
    <artifactId>velocity</artifactId>
    <version>1.7</version>
</dependency>

單模組生成

public class Generator {

    public static void main(String[] args) {
        String [] tableNames = new String[]{"表名1","表名2"};

        String location = "程式碼生成路徑位置";//例如: com/cn/jzedy
        generator(location,tableNames);
    }

    private static void generator(String location,String [] tableNames){

        GlobalConfig globalConfig = new GlobalConfig();// 全域性配置
                globalConfig.setOpen(false)//是否開啟輸出目錄 預設true
                        .setOutputDir(location)//生成檔案的輸出目錄
                        .setFileOverride(true)//是否覆蓋已有檔案 預設false
                        .setBaseResultMap(true)//開啟 BaseResultMap 預設false
                        .setBaseColumnList(true)//開啟 baseColumnList 預設false
                        .setActiveRecord(false)//開啟 ActiveRecord 模式 預設false
                        .setAuthor("Jzedy")//開發人員
                        .setServiceName("%sService");//service 命名方式 例如:%sBusiness 生成 UserBusiness

        
        DataSourceConfig dataSourceConfig = new DataSourceConfig();// 資料來源配置
        dataSourceConfig.setDbType(DbType.MYSQL)
                .setDriverName(Driver.class.getName())
                .setUsername("資料庫連線名稱")
                .setPassword("資料庫連線密碼")
                .setUrl("url地址");

        PackageConfig packageConfig = new PackageConfig();// 包配置
        packageConfig.setParent(location)
                .setEntity("entity")//Entity包名
                .setMapper("mapper")//mapper包名
                .setService("service")
                .setController("controller");

        StrategyConfig strategyConfig = new StrategyConfig();// 策略配置
                strategyConfig
                        .setCapitalMode(true)//駝峰命名
                        .setEntityLombokModel(false)//【實體】是否為lombok模型(預設 false)
                        .setRestControllerStyle(false)//生成 @RestController 控制器
                        .setNaming(NamingStrategy.underline_to_camel)//資料庫表對映到實體的命名策略,該處下劃線轉駝峰命名
                        .setInclude(tableNames);//需要包含的表名,允許正則表示式(與exclude二選一配置)


        new AutoGenerator()//// 程式碼生成器
                .setGlobalConfig(globalConfig)
                .setDataSource(dataSourceConfig)
                .setPackageInfo(packageConfig)
                .setStrategy(strategyConfig)
                .execute();

    }


}

具體配置查閱https://mp.baomidou.com/

多模組配置

因為專案分模組時候 想程式碼自動生成在對應位置,例如下面程式碼 將entity mapper service程式碼生成在
service模組中,controller生成在web模組中。可以自行模組進一步細分,只是對程式碼生成路徑在但模組上
基礎上調整而已。同時下述程式碼的程式碼生成模板是採用mybatis-plus自己的模板,若需自定義模板不再此論述,

public class Generator {

    public static void main(String[] args) {
        String [] tableNames = new String[]{"users","roles"};

        String [] modules = new String[]{"service","web"};//專案模組名,需自定義
        for (String module : modules) {
            moduleGenerator(module,tableNames);
        }
    }

    private static void moduleGenerator(String module,String [] tableNames){

        GlobalConfig globalConfig = getGlobalConfig(module);// 全域性配置

        DataSourceConfig dataSourceConfig = getDataSourceConfig();// 資料來源配置

        PackageConfig packageConfig = getPackageConfig(module);// 包配置

        StrategyConfig strategyConfig = getStrategyConfig(tableNames);// 策略配置

        TemplateConfig templateConfig = getTemplateConfig(module);// 配置模板

        new AutoGenerator()
                .setGlobalConfig(globalConfig)
                .setDataSource(dataSourceConfig)
                .setPackageInfo(packageConfig)
                .setStrategy(strategyConfig)
                .setTemplate(templateConfig)
                .execute();

    }

    private static TemplateConfig getTemplateConfig(String module) {
        TemplateConfig templateConfig = new TemplateConfig();
        if ("service".equals(module)){
            templateConfig.setEntity(new TemplateConfig().getEntity(false))
                    .setMapper(new TemplateConfig().getMapper())//mapper模板採用mybatis-plus自己模板
                    .setXml(new TemplateConfig().getXml())
                    .setService(new TemplateConfig().getService())
                    .setServiceImpl(new TemplateConfig().getServiceImpl())
                    .setController(null);//service模組不生成controller程式碼
        }else if ("web".equals(module)){//web模組只生成controller程式碼
            templateConfig.setEntity(null)
                    .setMapper(null)
                    .setXml(null)
                    .setService(null)
                    .setServiceImpl(null)
                    .setController(new TemplateConfig().getController());
        }else throw new IllegalArgumentException("引數匹配錯誤,請檢查");
        return templateConfig;
    }

    private static StrategyConfig getStrategyConfig(String[] tableNames) {
        StrategyConfig strategyConfig = new StrategyConfig();
        strategyConfig
                .setCapitalMode(true)//駝峰命名
                .setEntityLombokModel(false)
                .setRestControllerStyle(false)
                .setNaming(NamingStrategy.underline_to_camel)
                .setInclude(tableNames);
        return strategyConfig;
    }

    private static PackageConfig getPackageConfig(String module) {
        PackageConfig packageConfig = new PackageConfig();
        String packageName = "com.cn.jzedy";//不同模組 程式碼生成具體路徑自定義指定
        if ("service".equals(module)){
            packageName+=".web";
        }else if ("web".equals(module)){

        }
        packageConfig.setParent(packageName)
                .setEntity("entity")
                .setMapper("mapper")
                .setService("service")
                .setController("controller");
        return packageConfig;
    }

    private static DataSourceConfig getDataSourceConfig() {
        String dbUrl = "jdbc:mysql://localhost:3306/z-blogs";
        DataSourceConfig dataSourceConfig = new DataSourceConfig();
        dataSourceConfig.setDbType(DbType.MYSQL)
                .setDriverName(Driver.class.getName())
                .setUsername("root")
                .setPassword("root")
                .setUrl(dbUrl);
        return dataSourceConfig;
    }

    private static GlobalConfig getGlobalConfig(String module) {
        GlobalConfig globalConfig = new GlobalConfig();
        globalConfig.setOpen(false)//new File(module).getAbsolutePath()得到模組根目錄路徑,因事Maven專案,程式碼指定路徑自定義調整
                .setOutputDir(new File(module).getAbsolutePath()+"/src/main/java")//生成檔案的輸出目錄
                .setFileOverride(true)//是否覆蓋已有檔案
                .setBaseResultMap(true)
                .setBaseColumnList(true)
                .setActiveRecord(false)
                .setAuthor("Jzedy")
                .setServiceName("%sService");
        return globalConfig;
    }

}