Browse Source

新增渲染对象 MAP 信息预处理注入

hubin 6 years ago
parent
commit
686f5b234f

+ 9 - 0
mybatis-plus-generator/src/main/java/com/baomidou/mybatisplus/generator/InjectionConfig.java

@@ -59,4 +59,13 @@ public abstract class InjectionConfig {
      * 注入自定义 Map 对象
      */
     public abstract void initMap();
+
+    /**
+     * 模板待渲染 Object Map 预处理<br>
+     * com.baomidou.mybatisplus.generator.engine.AbstractTemplateEngine
+     * 方法: getObjectMap 结果处理
+     */
+    public Map<String, Object> prepareObjectMap(Map<String, Object> objectMap) {
+        return objectMap;
+    }
 }

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

@@ -234,7 +234,7 @@ public abstract class AbstractTemplateEngine {
         objectMap.put("superServiceImplClass", getSuperClassName(config.getSuperServiceImplClass()));
         objectMap.put("superControllerClassPackage", config.getSuperControllerClass());
         objectMap.put("superControllerClass", getSuperClassName(config.getSuperControllerClass()));
-        return objectMap;
+        return config.getInjectionConfig().prepareObjectMap(objectMap);
     }