使用程式碼生成器自定義Entity的部分註解

2022-05-21 16:00:13

程式碼生成器自定義Entity部分註解

對mybatis-plus程式碼生成器的entity模板進行部分自定義

此文章觀看前提：已經能用mybatis-plus自動生成程式碼

前因

因為原有模板自動生成的entity為下面這樣，表欄位為name，age，number，但我想生成圖2的樣子，在每個欄位前面新增個註解@ApiModelProperty()

查詢了我的程式碼生成器那個設定類

debug了最後要生成模板的兩句，找到了我的entity的生成模板為/templates/entity.java.ftl(因為模板引擎是freemarker，所以字尾為ftl)

隨後在templates/entity.java.ftl裡

找到模板生成的程式碼，然後在屬性模板前面加上這個程式碼

<#if swagger2>
    @ApiModelProperty(value = "${field.propertyName}",dataType = "${field.propertyType}",required = true)
</#if>

還有一點

記得在設定類中設定開啟 swagger2 模式

程式碼生成器最新版設定方案

因為最近寫專案使用到了MyBatis-Plus，所以去官網找到了一個程式碼生成器的新版，現在網上能查到的都是之前的舊版生成器程式碼，官方提供的快速生成沒有開啟Lombok、RestController還有很多自定義類名的策略。

新版對依賴版本有要求，適用版本：mybatis-plus-generator 3.5.1 及其以上版本，對歷史版本不相容！

<dependency>
    <groupId>com.baomidou</groupId>
    <artifactId>mybatis-plus-generator</artifactId>
    <version>3.5.1</version>
</dependency>
<!--對MyBatisPlus程式碼生成器進行支援的模板引擎-->
<dependency>
     <groupId>org.freemarker</groupId>
     <artifactId>freemarker</artifactId>
     <version>2.3.30</version>
</dependency>
<dependency>
     <groupId>com.baomidou</groupId>
     <artifactId>mybatis-plus-boot-starter</artifactId>
     <version>3.5.1</version>
</dependency>

本著喜新厭舊的初心，我根據官方檔案設定了一個新版常用的快速生成器，修改資料庫設定並設計好表之後填入表名直接執行使用就行了，如果要設定詳細的策略請去官網查檔案設定。MyBatis程式碼生成器官方檔案

import com.baomidou.mybatisplus.generator.FastAutoGenerator;
import com.baomidou.mybatisplus.generator.config.OutputFile;
import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;
import xyz.tylt.controller.BaseController;
import java.util.Collections;
/*
 * MyBatisPlus程式碼生成器
 */
public class CodeGenerator {
    public static void main(String[] args) {
        String drive = "D://";
        String xmlpath = drive + "xyz//resource//mapper";
        String projectPath = System.getProperty("user.dir");
        FastAutoGenerator.create("jdbc:mysql://localhost:3306/databasename?useSSl=ture&serverTimezone=UTC&useUnicode=true&characterEncoding=UTF-8", "root", "1234")
                .globalConfig(builder -> {
                    builder.author("tylt6688")// 設定作者
                            .enableSwagger() // 開啟 swagger 模式
                            .fileOverride() // 覆蓋已生成檔案
                            .outputDir(drive); // 指定輸出目錄
                })
                .packageConfig(builder -> {
                    builder.parent("xyz.tylt") // 設定父包名
//                            .moduleName("system") // 設定父包模組名
                            .pathInfo(Collections.singletonMap(OutputFile.mapperXml, xmlpath)); // 設定mapperXml生成路徑
                })
                //實體策略設定
                .strategyConfig(builder -> {
                    builder.entityBuilder()
                            //開啟Lombok，預設生成@Get，@Set，可以手動換成@Data
                            .enableLombok();
                })
                //控制類策略設定
                .strategyConfig(builder -> {
                    builder.controllerBuilder()
                             //如果沒有父類別請註釋掉
                            .superClass(BaseController.class)
                            .enableRestStyle();
                })
                //服務類策略設定
                .strategyConfig(builder -> {
                    builder.serviceBuilder()
                            .formatServiceFileName("%sService")
                            .formatServiceImplFileName("%sServiceImp");
                })
                .strategyConfig(builder -> {
                    builder.addInclude("tb_teachplan")// 設定資料庫中表名，按其中屬性內容進行生成
//                            .addInclude("tb_floor")
//                            .addInclude("tb_room")
                            .addTablePrefix("tb_", "c_", "sys_"); // 設定過濾表字首進行生成
                })
                .templateEngine(new FreemarkerTemplateEngine()) // 使用Freemarker引擎模板，預設的是Velocity引擎模板
                .execute();
    }
}

以上為個人經驗，希望能給大家一個參考，也希望大家多多支援it145.com。