Jelajahi Sumber

Merge pull request #1709 from luohongtu/3.0

自定前端页面生成的时候,需要通过表名,获取到相关数据库配置前端设置
qmdx 5 tahun lalu
induk
melakukan
fcedaaa3cc

+ 7 - 0
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;
 
@@ -60,6 +61,12 @@ public abstract class InjectionConfig {
      */
     public abstract void initMap();
 
+    /**
+     * 依据表相关信息, 从三方获取到需要元数据,处理方法环境里面
+     * @param tableInfo
+     */
+    public abstract void initTableMap(TableInfo tableInfo);
+
     /**
      * 模板待渲染 Object Map 预处理<br>
      * com.baomidou.mybatisplus.generator.engine.AbstractTemplateEngine

+ 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.initTableMap(tableInfo);
                     objectMap.put("cfg", injectionConfig.getMap());
                     List<FileOutConfig> focList = injectionConfig.getFileOutConfigList();
                     if (CollectionUtils.isNotEmpty(focList)) {