本文是本系列文章的第七篇，前6篇訪問地址如下：
- 小書MybatisPlus第1篇-整合SpringBoot快速開始增刪改查
- 小書MybatisPlus第2篇-條件構造器的應用及總結
- 小書MybatisPlus第3篇-自定義SQL
- 小書MybatisPlus第4篇-表格分頁與下拉分頁查詢
- 小書MybatisPlus第5篇-Active Record模式精講
- 小書MybatisPlus第6篇-主鍵生成策略精講
一、代碼生成器的基礎實作原理
相信大部分的開發者都使用過或者聽說過“模板引擎”，它可以幫我們實作視圖與資料的分離，快速開發視圖頁面，并將模板整合結果用于在瀏覽器顯示，其核心實作原理就是：HTML模板頁面 + 頁面資料 = 輸出結果，頁面視圖輸出的程序就是通過模板引擎實作的，
代碼生成器的實作原理與模板引擎實作頁面渲染的邏輯幾乎是一致的，除了下面的幾個區別：
- 所謂模板：就是某語言的代碼 + 模板引擎語法的占位符，該占位符用來資料天換，所以代碼生成器的模板檔案不再專指HTML頁面模板檔案，可以是任何型別的代碼檔案，
- 模板引擎的輸出結果在專案中是輸出給瀏覽器進行頁面渲染的，但是對于代碼生成器而言，模板引擎的輸出結果是保存到磁盤檔案，
二、如何撰寫模板檔案
要撰寫模板檔案，首先我們要知道正常的代碼待如何書寫，比如下面的POJO代碼：
上面的POJO代碼寫成Freemarker模板檔案，就是下面的樣子：
package ${package.Entity};
<#list table.importPackages as pkg>
import ${pkg};
</#list>
<#if entityLombokModel>
import lombok.Data;
import lombok.EqualsAndHashCode;
</#if>
/**
* <p>
* ${table.comment!}
* </p>
*
* @author ${author}
* @since ${date}
*/
<#if entityLombokModel>
@Data
<#if superEntityClass??>
@EqualsAndHashCode(callSuper = true)
<#else>
@EqualsAndHashCode(callSuper = false)
</#if>
</#if>
public class ${entity} extends Model<${entity}> {
<#list table.fields as field>
private ${field.propertyType} ${field.propertyName};
</#list>
Mybatis Plus 代碼生成的模板檔案：https://gitee.com/baomidou/mybatis-plus/tree/3.0/mybatis-plus-generator/src/main/resources/templates
三、資料從哪里來？
有了模板檔案，我們想通過模板引擎生成代碼，下面的一個問題就是資料從哪里來？有了資料我們才能生成代碼
- 從配置中來，比如：package路徑等一些靜態化不經常變化的資訊，一個專案生成的代碼存放的包路徑通常不會經常變化，
- 從資料庫中來，比如:物體類名稱、物體類欄位名稱、物體類欄位型別等資訊，類似于逆向工程，通過資料庫表名、欄位名、欄位型別等資訊生成物體資訊，
3.1.以MySQL的INFORMATION_SCHEMA資訊獲取為例
我們的代碼自動生成是針對資料庫操作，所以首先要了解資料庫表的結構
SELECT column_name,data_type,is_nullable,character_maximum_length,column_comment
FROM INFORMATION_SCHEMA.COLUMNS
WHERE table_name='kpi_task'
AND table_schema='home'
ORDER BY ordinal_position
如上圖SQL查詢的是home庫，kpi_task表的資訊，如下：
- column_name作為表的欄位可以生成物體類的成員變數引數名稱（通常是駝峰標識規則）
- data_type，is_nullable，character_maximum_length可用于生成校驗規則，
- 注釋可以用于生成column_comment
四、Mybatis Plus代碼生成器的使用
如果上面的代碼生成器實作原理你都看懂了，下面的這些配置你也就不難理解了，
4.1.添加依賴
- 添加 代碼生成器 依賴
<dependency>
<groupId>com.baomidou</groupId>
<artifactId>mybatis-plus-generator</artifactId>
<version>3.3.2</version>
</dependency>
-
添加 模板引擎 依賴，MyBatis-Plus 支持 Velocity（默認）、Freemarker、Beetl，用戶可以選擇自己熟悉的模板引擎，
Velocity（默認）：
<dependency> <groupId>org.apache.velocity</groupId> <artifactId>velocity-engine-core</artifactId> <version>2.2</version> </dependency>Freemarker：
<dependency> <groupId>org.freemarker</groupId> <artifactId>freemarker</artifactId> <version>2.3.30</version> </dependency>Beetl：
<dependency> <groupId>com.ibeetl</groupId> <artifactId>beetl</artifactId> <version>3.1.8.RELEASE</version> </dependency>
注意！如果您選擇了非默認引擎，需要在 AutoGenerator 中 設定模板引擎，
AutoGenerator generator = new AutoGenerator();
// set freemarker engine
generator.setTemplateEngine(new FreemarkerTemplateEngine());
// set beetl engine
generator.setTemplateEngine(new BeetlTemplateEngine());
// set custom engine (reference class is your custom engine class)
generator.setTemplateEngine(new CustomTemplateEngine());
// other config
...
4.2.代碼生成配置
AutoGenerator 是 MyBatis-Plus 的代碼生成器，通過 AutoGenerator 可以快速生成 Entity、Mapper、Mapper XML、Service、Controller 等各個模塊的代碼，極大的提升了開發效率，
執行下面的這個測驗用例，Mybatis Plus就可以幫助我們實作以上各層的代碼生成到對應的package路徑下面
public class CodeGenerator {
@Test
public void startGenerator() {
//1、全域配置
GlobalConfig config = new GlobalConfig();
String projectPath = System.getProperty("user.dir");
config.setActiveRecord(true)//開啟AR模式
.setAuthor("zimug")//設定作者
.setOutputDir(projectPath + "/src/main/java")//生成路徑(一般在此專案的src/main/java下)
.setFileOverride(true)//第二次生成會把第一次生成的覆寫掉
.setOpen(true)//生成完畢后是否自動打開輸出目錄
//.setSwagger2(true)//物體屬性 Swagger2 注解
//.setIdType(IdType.AUTO)//主鍵策略
.setServiceName("%sService")//生成的service介面名字首字母是否為I，這樣設定就沒有I
.setBaseResultMap(true)//生成resultMap
.setBaseColumnList(true);//在xml中生成基礎列
//2、資料源配置
DataSourceConfig dataSourceConfig = new DataSourceConfig();
dataSourceConfig.setDbType(DbType.MYSQL)//資料庫型別
.setDriverName("com.mysql.jdbc.Driver")
.setUrl("jdbc:mysql://localhost:3306/zimug")
.setUsername("root")
.setPassword("zimug3456");
//3、策略配置
StrategyConfig strategyConfig = new StrategyConfig();
strategyConfig.setCapitalMode(true)//開啟全域大寫命名
.setNaming(NamingStrategy.underline_to_camel)//表名映射到物體的命名策略(下劃線到駝峰)
//表欄位映射屬性名策略(未指定按naming)
.setColumnNaming(NamingStrategy.underline_to_camel)
//.setTablePrefix("tb_")//表名前綴
//.setSuperEntityClass("你自己的父類物體,沒有就不用設定!")
//.setSuperEntityColumns("id");//寫于父類中的公共欄位
//.setSuperControllerClass("自定義繼承的Controller類全稱，帶包名,沒有就不用設定!")
.setRestControllerStyle(true) //生成 @RestController 控制器
.setEntityLombokModel(true)//使用lombok
.setInclude("sys_user","sys_role");//逆向工程使用的表
//4、包名策略配置
PackageConfig packageConfig = new PackageConfig();
packageConfig.setParent("com.zimug.example")//設定包名的parent
.setMapper("mapper")
.setService("service")
.setController("controller")
.setEntity("entity")
.setXml("mapper");//設定xml檔案的目錄
//5、整合配置
AutoGenerator autoGenerator = new AutoGenerator();
autoGenerator.setGlobalConfig(config)
.setDataSource(dataSourceConfig)
.setStrategy(strategyConfig)
.setPackageInfo(packageConfig);
//6、執行
autoGenerator.execute();
}
}
歡迎關注我的博客，里面有很多精品合集
- 本文轉載注明出處（必須帶連接，不能只轉文字）：字母哥博客，
覺得對您有幫助的話，幫我點贊、分享！您的支持是我不竭的創作動力！ ，另外，筆者最近一段時間輸出了如下的精品內容，期待您的關注，
- 《手摸手教你學Spring Boot2.0》
- 《Spring Security-JWT-OAuth2一本通》
- 《實戰前后端分離RBAC權限管理系統》
- 《實戰SpringCloud微服務從青銅到王者》
- 《VUE深入淺出系列》
轉載請註明出處，本文鏈接：https://www.uj5u.com/houduan/137116.html
標籤：Java
上一篇：資料結構--哈希表（Java）