這就是Java代碼生成器的製做流程

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 獲取更多資訊

我的博客：https://felord.cn