Sfoglia il codice sorgente

修改,自动页面的时候,缺少表相关的数据,导致无法从数据库获取

lht 5 anni fa
parent
commit
38188f30ba

+ 4 - 2
mybatis-plus-generator/src/main/java/com/baomidou/mybatisplus/generator/InjectionConfig.java

@@ -22,6 +22,7 @@ import com.baomidou.mybatisplus.generator.config.FileOutConfig;
 import com.baomidou.mybatisplus.generator.config.IFileCreate;
 import com.baomidou.mybatisplus.generator.config.builder.ConfigBuilder;
 
+import com.baomidou.mybatisplus.generator.config.po.TableInfo;
 import lombok.Data;
 import lombok.experimental.Accessors;
 
@@ -56,9 +57,10 @@ public abstract class InjectionConfig {
     private IFileCreate fileCreate;
 
     /**
-     * 注入自定义 Map 对象
+     * 根据表相关数据 注入自定义Map对象
+     * @param tableInfo
      */
-    public abstract void initMap();
+    public abstract void initMap(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();
+                    injectionConfig.initMap(tableInfo);
                     objectMap.put("cfg", injectionConfig.getMap());
                     List<FileOutConfig> focList = injectionConfig.getFileOutConfigList();
                     if (CollectionUtils.isNotEmpty(focList)) {