mybatis-plus-generator 代碼自動(dòng)生成器

添加依賴

<properties>
    <mybatis-plus-generator.version>3.2.0</mybatis-plus-generator.version>
    <velocity.version>2.1</velocity.version>
    <freemarker.version>2.3.29</freemarker.version>
    <ibeetl.version>3.0.13.RELEASE</ibeetl.version>
    <junit.version>4.12</junit.version>
</properties>

<dependency>
    <groupId>com.baomidou</groupId>
    <artifactId>mybatis-plus-generator</artifactId>
    <version>${mybatis-plus-generator.version}</version>
</dependency>
<dependency>
    <groupId>org.apache.velocity</groupId>
    <artifactId>velocity-engine-core</artifactId>
    <version>${velocity.version}</version>
</dependency>
<dependency>
    <groupId>org.freemarker</groupId>
    <artifactId>freemarker</artifactId>
    <version>${freemarker.version}</version>
</dependency>
<dependency>
    <groupId>com.ibeetl</groupId>
    <artifactId>beetl</artifactId>
    <version>${ibeetl.version}</version>
</dependency>
    <dependency>
      <groupId>junit</groupId>
      <artifactId>junit</artifactId>
      <version>${junit.version}</version>
      <scope>compile</scope>
    </dependency>

工具類

package com.zzf.springboot.util;


import com.baomidou.mybatisplus.annotation.DbType;
import com.baomidou.mybatisplus.generator.AutoGenerator;
import com.baomidou.mybatisplus.generator.config.*;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;
import org.junit.Test;

/**
 *  運(yùn)行此方法生成mybatis代碼
 *  生成代碼自動(dòng)放入對(duì)應(yīng)目錄
 * @author wangshu
 * @create 2019/10/18
 */
public class MyBatisGeneratorRun {
    //數(shù)據(jù)庫(kù)類型
    private final DbType dbType = DbType.MYSQL;
    //數(shù)據(jù)庫(kù)連結(jié)信息
    private final String dbUrl = "jdbc:mysql://localhost:3306/zzf?useUnicode=true&characterEncoding=UTF-8";
    private final String driver = "com.mysql.jdbc.Driver";
    private final String userName = "root";
    private final String password = "root";

    //項(xiàng)目名
    private final String projectName = "spring-boot";
    //指定包名
    private final String packageName = "com.zzf.springboot";
    //controller基礎(chǔ)類
    private final String superControllerClass = packageName + ".common.BaseController";
    //entity基礎(chǔ)類
    private final String superEntityClass = packageName + ".common.BaseEntity";
    //模塊名 如果有模塊名梳毙,則需在模塊名前加. 例:.log
    private final String moduleName = "";
    //作者名
    private final String author = "zhuzhifeng";
    //指定生成的表名
    private final String[] tableNames = new String[]{"sys_role"};


    @Test
    public void generateCode() {
        //serviceNameStartWithI:user -> UserService, 設(shè)置成true: user -> IUserService
        generateByTables(false, packageName, tableNames);
    }
    /**
     * 根據(jù)表自動(dòng)生成
     * @param serviceNameStartWithI 默認(rèn)為false
     * @param packageName      包名
     * @param tableNames      表名
     * @author Terry
     */
    private void generateByTables(boolean serviceNameStartWithI, String packageName, String... tableNames) {
        //配置數(shù)據(jù)源
        DataSourceConfig dataSourceConfig = getDataSourceConfig();
        // 策略配置
        StrategyConfig strategyConfig = getStrategyConfig(tableNames);
        //全局變量配置
        GlobalConfig globalConfig = getGlobalConfig(serviceNameStartWithI);
        //包名配置
        PackageConfig packageConfig = getPackageConfig(packageName);
        //自動(dòng)生成
        atuoGenerator(dataSourceConfig, strategyConfig, globalConfig, packageConfig);
    }
    /**
     * 集成
     * @param dataSourceConfig 配置數(shù)據(jù)源
     * @param strategyConfig  策略配置
     * @param config      全局變量配置
     * @param packageConfig  包名配置
     * @author Terry
     */
    private void atuoGenerator(DataSourceConfig dataSourceConfig, StrategyConfig strategyConfig, GlobalConfig config, PackageConfig packageConfig) {
        new AutoGenerator()
                .setGlobalConfig(config)
                .setDataSource(dataSourceConfig)
                .setStrategy(strategyConfig)
                .setPackageInfo(packageConfig)
                .setTemplateEngine(new FreemarkerTemplateEngine())
                .execute();
    }
    /**
     * 設(shè)置包名
     * @param packageName 父路徑包名
     * @param packageName 模塊名
     * @return PackageConfig 包名配置
     * @author Terry
     */
    private PackageConfig getPackageConfig(String packageName) {
        return new PackageConfig()
                .setParent(packageName)
                .setXml("mapper.xml")
                .setMapper("mapper")
                .setController("controller")
                .setEntity("entity");
    }
    /**
     * 全局配置
     * @param serviceNameStartWithI false
     * @return GlobalConfig
     * @author Terry
     */
    private GlobalConfig getGlobalConfig(boolean serviceNameStartWithI) {
        GlobalConfig globalConfig = new GlobalConfig();
        globalConfig
                .setBaseColumnList(true)
                .setBaseResultMap(true)
                .setActiveRecord(false)
                //作者
                .setAuthor(author)
                //設(shè)置輸出路徑
                .setOutputDir(getOutputDir(projectName))
                .setFileOverride(true);
        if (!serviceNameStartWithI) {
            //設(shè)置service名
            globalConfig.setServiceName("%sService");
        }
        return globalConfig;
    }
    /**
     * 返回項(xiàng)目路徑
     * @param projectName 項(xiàng)目名
     * @return 項(xiàng)目路徑
     * @author Terry
     */
    private String getOutputDir(String projectName) {
//        String path = this.getClass().getClassLoader().getResource("").getPath();
//        int index = path.indexOf(projectName);
        return "E:/repository/git-repository/project/" + projectName + "/src/main/java/";
    }
    /**
     * 策略配置
     * @param tableNames 表名
     * @return StrategyConfig
     * @author Terry
     */
    private StrategyConfig getStrategyConfig(String... tableNames) {
        return new StrategyConfig()
                // 全局大寫命名 ORACLE 注意
                .setCapitalMode(true)
                //從數(shù)據(jù)庫(kù)表到文件的命名策略
                .setNaming(NamingStrategy.underline_to_camel)
                //需要生成的的表名或辖,多個(gè)表名傳數(shù)組
                .setInclude(tableNames)
                //公共父類
                //.setSuperControllerClass(superControllerClass)
                //.setSuperEntityClass(superEntityClass)
                // 寫于父類中的公共字段
                //.setSuperEntityColumns("id")
                //使用lombok
                .setEntityLombokModel(true)
                //rest風(fēng)格
                .setRestControllerStyle(true);
    }
    /**
     * 配置數(shù)據(jù)源
     * @return 數(shù)據(jù)源配置 DataSourceConfig
     * @author Terry
     */
    private DataSourceConfig getDataSourceConfig() {
        return new DataSourceConfig().setDbType(dbType)
                .setUrl(dbUrl)
                .setUsername(userName)
                .setPassword(password)
                .setDriverName(driver);
    }
    /**
     * 根據(jù)表自動(dòng)生成
     * @param packageName 包名
     * @param tableNames 表名
     * @author Terry
     */
    @SuppressWarnings("unused")
    private void generateByTables(String packageName, String... tableNames) {
        generateByTables(true, packageName, tableNames);
    }
}
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
  • 序言:七十年代末琳拭,一起剝皮案震驚了整個(gè)濱河市,隨后出現(xiàn)的幾起案子做葵,更是在濱河造成了極大的恐慌,老刑警劉巖心墅,帶你破解...
    沈念sama閱讀 219,427評(píng)論 6 508
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件酿矢,死亡現(xiàn)場(chǎng)離奇詭異榨乎,居然都是意外死亡,警方通過(guò)查閱死者的電腦和手機(jī)瘫筐,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 93,551評(píng)論 3 395
  • 文/潘曉璐 我一進(jìn)店門蜜暑,熙熙樓的掌柜王于貴愁眉苦臉地迎上來(lái),“玉大人策肝,你說(shuō)我怎么就攤上這事肛捍。” “怎么了之众?”我有些...
    開封第一講書人閱讀 165,747評(píng)論 0 356
  • 文/不壞的土叔 我叫張陵拙毫,是天一觀的道長(zhǎng)。 經(jīng)常有香客問(wèn)我酝枢,道長(zhǎng)恬偷,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 58,939評(píng)論 1 295
  • 正文 為了忘掉前任帘睦,我火速辦了婚禮袍患,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘竣付。我一直安慰自己诡延,他們只是感情好,可當(dāng)我...
    茶點(diǎn)故事閱讀 67,955評(píng)論 6 392
  • 文/花漫 我一把揭開白布古胆。 她就那樣靜靜地躺著肆良,像睡著了一般。 火紅的嫁衣襯著肌膚如雪逸绎。 梳的紋絲不亂的頭發(fā)上惹恃,一...
    開封第一講書人閱讀 51,737評(píng)論 1 305
  • 那天,我揣著相機(jī)與錄音棺牧,去河邊找鬼巫糙。 笑死,一個(gè)胖子當(dāng)著我的面吹牛颊乘,可吹牛的內(nèi)容都是我干的参淹。 我是一名探鬼主播,決...
    沈念sama閱讀 40,448評(píng)論 3 420
  • 文/蒼蘭香墨 我猛地睜開眼乏悄,長(zhǎng)吁一口氣:“原來(lái)是場(chǎng)噩夢(mèng)啊……” “哼浙值!你這毒婦竟也來(lái)了?” 一聲冷哼從身側(cè)響起檩小,我...
    開封第一講書人閱讀 39,352評(píng)論 0 276
  • 序言:老撾萬(wàn)榮一對(duì)情侶失蹤开呐,失蹤者是張志新(化名)和其女友劉穎,沒(méi)想到半個(gè)月后,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體负蚊,經(jīng)...
    沈念sama閱讀 45,834評(píng)論 1 317
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡神妹,尸身上長(zhǎng)有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 37,992評(píng)論 3 338
  • 正文 我和宋清朗相戀三年,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了家妆。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片鸵荠。...
    茶點(diǎn)故事閱讀 40,133評(píng)論 1 351
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡,死狀恐怖伤极,靈堂內(nèi)的尸體忽然破棺而出蛹找,到底是詐尸還是另有隱情,我是刑警寧澤哨坪,帶...
    沈念sama閱讀 35,815評(píng)論 5 346
  • 正文 年R本政府宣布庸疾,位于F島的核電站,受9級(jí)特大地震影響当编,放射性物質(zhì)發(fā)生泄漏届慈。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 41,477評(píng)論 3 331
  • 文/蒙蒙 一忿偷、第九天 我趴在偏房一處隱蔽的房頂上張望金顿。 院中可真熱鬧,春花似錦鲤桥、人聲如沸揍拆。這莊子的主人今日做“春日...
    開封第一講書人閱讀 32,022評(píng)論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽(yáng)嫂拴。三九已至,卻和暖如春贮喧,著一層夾襖步出監(jiān)牢的瞬間筒狠,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 33,147評(píng)論 1 272
  • 我被黑心中介騙來(lái)泰國(guó)打工箱沦, 沒(méi)想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留窟蓝,地道東北人。 一個(gè)月前我還...
    沈念sama閱讀 48,398評(píng)論 3 373
  • 正文 我出身青樓饱普,卻偏偏與公主長(zhǎng)得像,于是被迫代替她去往敵國(guó)和親状共。 傳聞我的和親對(duì)象是個(gè)殘疾皇子套耕,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 45,077評(píng)論 2 355