Mybatis Plus 代碼生成器

最近在學習Mybatis Plus，剛剛寫好一個代碼生成器的工具類，拿出來給大家分享一下，相比Mybatis要方便很多，不需要配置文件

package com.utils;

import com.baomidou.mybatisplus.core.exceptions.MybatisPlusException;
import com.baomidou.mybatisplus.core.toolkit.StringUtils;
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 java.util.Scanner;

// 演示例子，執行 main 方法控制台輸入模塊表名回車自動生成對應項目目錄中
// MyBatis Plus(MP)
public class MPUtils {

 /**
 * <p>
 * 讀取控制台內容
 * </p>
 */
 public static String scanner(String tip) {
 Scanner scanner = new Scanner(System.in);
 StringBuilder help = new StringBuilder();
 help.append("請輸入" + tip + "：");
 System.out.println(help.toString());
 if (scanner.hasNext()) {
 String ipt = scanner.next();
 //MP提供的工具類，做非空判斷
 if (StringUtils.isNotEmpty(ipt)) {
 return ipt;
 }
 }
 throw new MybatisPlusException("請輸入正確的" + tip + "！");
 }

 public static void main(String[] args) {
 // 代碼生成器
 AutoGenerator mpg = new AutoGenerator();

 // 全局配置
 GlobalConfig gc = new GlobalConfig();
 // 獲取項目的絕對路徑
 String projectPath = System.getProperty("user.dir");
 // 指定文件生成路徑
 gc.setOutputDir(projectPath + "/src/main/java");
 // 作者（沒什么用。。。）
 gc.setAuthor("jobob");
 gc.setOpen(false);
 mpg.setGlobalConfig(gc);

 // 數據源配置
 DataSourceConfig dsc = new DataSourceConfig();
 // 基本連接字符串
 dsc.setUrl("jdbc:mysql:///testdb");
 dsc.setDriverName("com.mysql.jdbc.Driver");
 dsc.setUsername("root");
 dsc.setPassword("ok");
 mpg.setDataSource(dsc);

 // 包配置
 PackageConfig pc = new PackageConfig();
 // 要生成的包名（隸屬哪個模塊）
 pc.setModuleName(scanner("模塊名"));
 // 父級包名
 pc.setParent("com");
 mpg.setPackageInfo(pc);

 // 設置生成模板（默認就是這個，可以不寫）
 mpg.setTemplate(new TemplateConfig());

 // 策略配置
 StrategyConfig strategy = new StrategyConfig();
 // 表名生成策略
 strategy.setNaming(NamingStrategy.nochange);
 // 字段名生成策略
 strategy.setColumnNaming(NamingStrategy.nochange);
 // 是否使用Lombok優化代碼
 strategy.setEntityLombokModel(true);
 // controller類是否直接返回json
 strategy.setRestControllerStyle(true);
 // 要生成的表名（控制台輸入）
 strategy.setInclude(scanner("表名"));
 strategy.setSuperEntityColumns("id");
 strategy.setControllerMappingHyphenStyle(true);
 mpg.setStrategy(strategy);
 // 切換引擎
 mpg.setTemplateEngine(new FreemarkerTemplateEngine());
 // 執行代碼生成器
 mpg.execute();
 }

}