Browse Source

调整 github pr 1709 修改

hubin 5 years ago
parent
commit
91a922a017

+ 10 - 3
mybatis-plus-generator/src/main/java/com/baomidou/mybatisplus/generator/InjectionConfig.java

@@ -57,11 +57,18 @@ public abstract class InjectionConfig {
     private IFileCreate fileCreate;
 
     /**
-     * 注入自定义 Map 对象
+     * 注入自定义 Map 对象,针对所有表的全局参数
+     */
+    public abstract void initMap();
+
+    /**
+     * 依据表相关信息,从三方获取到需要元数据,处理方法环境里面
      *
-     * @param tableInfo 依据表相关信息,从三方获取到需要元数据,处理方法环境里面
+     * @param tableInfo
      */
-    public abstract void initMap(TableInfo tableInfo);
+    public void initTableMap(TableInfo tableInfo) {
+        // 子类重写注入表对应补充信息
+    }
 
     /**
      * 模板待渲染 Object Map 预处理<br>

+ 1 - 1
mybatis-plus-generator/src/main/java/com/baomidou/mybatisplus/generator/engine/AbstractTemplateEngine.java

@@ -72,7 +72,7 @@ public abstract class AbstractTemplateEngine {
                 // 自定义内容
                 InjectionConfig injectionConfig = getConfigBuilder().getInjectionConfig();
                 if (null != injectionConfig) {
-                    injectionConfig.initMap(tableInfo);
+                    injectionConfig.initTableMap(tableInfo);
                     objectMap.put("cfg", injectionConfig.getMap());
                     List<FileOutConfig> focList = injectionConfig.getFileOutConfigList();
                     if (CollectionUtils.isNotEmpty(focList)) {