Mybatis的原始的執行方式
阿新 • • 發佈：2020-07-14
1. 前言
前幾天寫了篇關於Mybatis Plus程式碼生成器的文章，不少同學私下問我這個程式碼生成器是如何運作的，為什麼要用到一些模板引擎，所以今天來說明下程式碼生成器的流程。
2. 程式碼生成器的使用場景
我們在編碼中存在很多樣板程式碼，格式較為固定，結構隨著專案的迭代也比較穩定，而且數量巨大，這種程式碼寫多了也沒有什麼技術含量，在這種情況下程式碼生成器可以有效提高我們的效率，其它情況並不適於使用程式碼生成器。
3. 程式碼生成器的製作流程
首先我們要製作模板，把樣板程式碼的固定格式抽出來。然後把動態屬性繫結到模板中，就像做填空題一樣。所以在這個流程中模板引擎是最合適的。我們通過使用模板引擎的語法將資料動態地解析到靜態模板中去，然後匯出為程式設計中對應的檔案就行了。
另外模板引擎有著豐富的繫結資料的指令集，可以讓我們根據條件動態的繫結資料到模板中去。以Freemarker為例：
三元表示式：
${true ? 'checked': ''}
還有我們等下要用的遍歷列表：
<#list fields as field>
private ${field.fieldType} ${field.fieldName};
</#list>
在Java開發中我們常用的模板引擎有Freemarker、Velocity、Thymeleaf ，隨著Web開發中前後端分離的流行模板引擎的使用場景正在被壓縮，但是它依然是一門有用的技術。
4. 程式碼生成器演示
接下來，我們以Freemarker為例寫一個簡單的程式碼生成器，來生成POJO類。需要引入Freemarker的依賴。
<dependency>
<groupId>org.freemarker</groupId>
<artifactId>freemarker</artifactId>
<version>2.3.28</version>
</dependency>
4.1 模板製作
POJO的結構可以分為以下幾部分：
java.lang包無需匯入。
所以將這些規則封裝到配置類中：
public class JavaProperties {
// 包名
private final String pkg;
// 類名
private final String entityName;
// 屬性集合 需要改寫 equals hash 保證名字可不重複 型別可重複
private final Set<Field> fields = new LinkedHashSet<>();
// 匯入類的不重複集合
private final Set<String> imports = new LinkedHashSet<>();
public JavaProperties(String entityName, String pkg) {
this.entityName = entityName;
this.pkg = pkg;
}
public void addField(Class<?> type, String fieldName) {
// 處理 java.lang
final String pattern = "java.lang";
String fieldType = type.getName();
if (!fieldType.startsWith(pattern)) {
// 處理導包
imports.add(fieldType);
}
Field field = new Field();
// 處理成員屬性的格式
int i = fieldType.lastIndexOf(".");
field.setFieldType(fieldType.substring(i + 1));
field.setFieldName(fieldName);
fields.add(field);
}
public String getPkg() {
return pkg;
}
public String getEntityName() {
return entityName;
}
public Set<Field> getFields() {
return fields;
}
public Set<String> getImports() {
return imports;
}
/**
* 成員屬性封裝物件.
*/
public static class Field {
// 成員屬性型別
private String fieldType;
// 成員屬性名稱
private String fieldName;
public String getFieldType() {
return fieldType;
}
public void setFieldType(String fieldType) {
this.fieldType = fieldType;
}
public String getFieldName() {
return fieldName;
}
public void setFieldName(String fieldName) {
this.fieldName = fieldName;
}
/**
* 一個類的成員屬性 一個名稱只能出現一次
* 我們可以通過覆寫equals hash 方法 然後放入Set
*
* @param o 另一個成員屬性
* @return 比較結果
*/
@Override
public boolean equals(Object o) {
if (this == o) return true;
if (o == null || getClass() != o.getClass()) return false;
Field field = (Field) o;
return Objects.equals(fieldName, field.fieldName);
}
@Override
public int hashCode() {
return Objects.hash(fieldType, fieldName);
}
}
}
接著就是靜態模板entity.ftl
package ${pkg};
<#list imports as impt>
import ${impt};
</#list>
/**
* the ${entityName} type
* @author felord.cn
*/
public class ${entityName} {
<#list fields as field>
private ${field.fieldType} ${field.fieldName};
</#list>
}
這裡用到了Freemarker繫結資料的語法，比如List迭代渲染。
4.2 生成器編寫
Freemarker通過宣告配置並獲取模板物件freemarker.template，該物件的process方法可以將動態資料繫結到模板中並匯出為檔案，最終實現了程式碼生成器，核心程式碼如下：
/**
* 簡單的程式碼生成器.
*
* @param rootPath maven 的 java 目錄
* @param templatePath 模板存放的資料夾
* @param templateName 模板的名稱
* @param javaProperties 需要渲染物件的封裝
* @throws IOException the io exception
* @throws TemplateException the template exception
*/
public static void autoCodingJavaEntity(String rootPath,
String templatePath,
String templateName,
JavaProperties javaProperties) throws IOException, TemplateException {
// freemarker 配置
Configuration configuration = new Configuration(Configuration.DEFAULT_INCOMPATIBLE_IMPROVEMENTS);
configuration.setDefaultEncoding("UTF-8");
// 指定模板的路徑
configuration.setDirectoryForTemplateLoading(new File(templatePath));
// 根據模板名稱獲取路徑下的模板
Template template = configuration.getTemplate(templateName);
// 處理路徑問題
final String ext = ".java";
String javaName = javaProperties.getEntityName().concat(ext);
String packageName = javaProperties.getPkg();
String out = rootPath.concat(Stream.of(packageName.split("\\."))
.collect(Collectors.joining("/", "/", "/" + javaName)));
// 定義一個輸出流來匯出程式碼檔案
OutputStreamWriter outputStreamWriter = new OutputStreamWriter(new FileOutputStream(out));
// freemarker 引擎將動態資料繫結的模板並匯出為檔案
template.process(javaProperties, outputStreamWriter);
}
通過執行以下程式碼即可生成一個UserEntity的POJO：
// 路徑根據自己專案的特點調整
String rootPath = "C:\\Users\\felord\\IdeaProjects\\codegenerator\\src\\main\\java";
String packageName = "cn.felord.code";
String templatePath = "C:\\Users\\felord\\IdeaProjects\\codegenerator\\src\\main\\resources\\templates";
String templateName = "entity.ftl";
JavaProperties userEntity = new JavaProperties("UserEntity", packageName);
userEntity.addField(String.class, "username");
userEntity.addField(LocalDate.class, "birthday");
userEntity.addField(LocalDateTime.class, "addTime");
userEntity.addField(Integer.class, "gender");
userEntity.addField(Integer.class, "age");
autoCodingJavaEntity(rootPath, templatePath, templateName, userEntity);
生成的效果是不是跟手寫的差不多：
5. 總結
這就是大部分程式碼生成器的機制，希望可以解答一些網友的疑問。多多關注：碼農小胖哥 獲取更多幹貨，相關的DEMO可通過公眾號回覆codegen獲取。如果你有疑問可以通過微信MSW_623進行溝通。
關注公眾號：Felordcn 獲取更多資訊