diff --git a/mybatis-plus-generator/src/test/java/com/baomidou/mybatisplus/test/generator/MysqlGenerator.java b/mybatis-plus-generator/src/test/java/com/baomidou/mybatisplus/test/generator/MysqlGenerator.java index 02623b0ebb..8b1aad5453 100644 --- a/mybatis-plus-generator/src/test/java/com/baomidou/mybatisplus/test/generator/MysqlGenerator.java +++ b/mybatis-plus-generator/src/test/java/com/baomidou/mybatisplus/test/generator/MysqlGenerator.java @@ -15,11 +15,7 @@ */ package com.baomidou.mybatisplus.test.generator; -import java.util.ArrayList; -import java.util.Collections; -import java.util.HashMap; -import java.util.List; -import java.util.Map; +import java.util.*; import com.baomidou.mybatisplus.annotation.DbType; import com.baomidou.mybatisplus.annotation.FieldFill; @@ -34,6 +30,7 @@ import com.baomidou.mybatisplus.generator.config.converts.MySqlTypeConvert; import com.baomidou.mybatisplus.generator.config.po.TableFill; import com.baomidou.mybatisplus.generator.config.po.TableInfo; +import com.baomidou.mybatisplus.generator.config.querys.MySqlQuery; import com.baomidou.mybatisplus.generator.config.rules.IColumnType; import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy; import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine; @@ -90,6 +87,19 @@ public IColumnType processTypeConvert(GlobalConfig globalConfig, String fieldTyp return super.processTypeConvert(globalConfig, fieldType); } }) + .setDbQuery(new MySqlQuery() { + + /** + * 重写父类预留查询自定义字段
+ * 这里查询的 SQL 对应父类 tableFieldsSql 的查询字段,默认不能满足你的需求请重写它
+ * 模板中调用: table.fields 获取所有字段信息, + * 然后循环字段获取 field.customMap 从 MAP 中获取注入字段如下 NULL 或者 PRIVILEGES + */ + @Override + public String[] fieldCustom() { + return new String[]{"NULL", "PRIVILEGES"}; + } + }) .setDriverName(Driver.class.getName()) .setUsername("root") .setPassword("1q2w3e4r")