hubin преди 5 години
родител
ревизия
c8882dd5f9

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

@@ -58,14 +58,10 @@ public abstract class InjectionConfig {
 
 
     /**
     /**
      * 注入自定义 Map 对象
      * 注入自定义 Map 对象
+     *
+     * @param tableInfo 依据表相关信息,从三方获取到需要元数据,处理方法环境里面
      */
      */
-    public abstract void initMap();
-
-    /**
-     * 依据表相关信息, 从三方获取到需要元数据,处理方法环境里面
-     * @param tableInfo
-     */
-    public abstract void initTableMap(TableInfo tableInfo);
+    public abstract void initMap(TableInfo tableInfo);
 
 
     /**
     /**
      * 模板待渲染 Object Map 预处理<br>
      * 模板待渲染 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();
                 InjectionConfig injectionConfig = getConfigBuilder().getInjectionConfig();
                 if (null != injectionConfig) {
                 if (null != injectionConfig) {
-                    injectionConfig.initTableMap(tableInfo);
+                    injectionConfig.initMap(tableInfo);
                     objectMap.put("cfg", injectionConfig.getMap());
                     objectMap.put("cfg", injectionConfig.getMap());
                     List<FileOutConfig> focList = injectionConfig.getFileOutConfigList();
                     List<FileOutConfig> focList = injectionConfig.getFileOutConfigList();
                     if (CollectionUtils.isNotEmpty(focList)) {
                     if (CollectionUtils.isNotEmpty(focList)) {