Browse Source

生成器加回 MODULE_NAME 开放配置 config

hubin 6 năm trước cách đây
mục cha
commit
ba8b550111

+ 2 - 0
mybatis-plus-generator/src/main/java/com/baomidou/mybatisplus/generator/config/ConstVal.java

@@ -29,6 +29,8 @@ import com.baomidou.mybatisplus.core.toolkit.StringPool;
  */
 public interface ConstVal {
 
+    String MODULE_NAME = "ModuleName";
+
     String ENTITY = "Entity";
     String SERVICE = "Service";
     String SERVICE_IMPL = "ServiceImpl";

+ 1 - 0
mybatis-plus-generator/src/main/java/com/baomidou/mybatisplus/generator/config/builder/ConfigBuilder.java

@@ -246,6 +246,7 @@ public class ConfigBuilder {
     private void handlerPackage(TemplateConfig template, String outputDir, PackageConfig config) {
         // 包信息
         packageInfo = new HashMap<>(6);
+        packageInfo.put(ConstVal.MODULE_NAME, config.getModuleName());
         packageInfo.put(ConstVal.ENTITY, joinPackage(config.getParent(), config.getEntity()));
         packageInfo.put(ConstVal.MAPPER, joinPackage(config.getParent(), config.getMapper()));
         packageInfo.put(ConstVal.XML, joinPackage(config.getParent(), config.getXml()));

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

@@ -216,6 +216,7 @@ public abstract class AbstractTemplateEngine {
             objectMap.put("controllerMappingHyphen", StringUtils.camelToHyphen(tableInfo.getEntityPath()));
         }
         objectMap.put("restControllerStyle", config.getStrategyConfig().isRestControllerStyle());
+        objectMap.put("config", config);
         objectMap.put("package", config.getPackageInfo());
         GlobalConfig globalConfig = config.getGlobalConfig();
         objectMap.put("author", globalConfig.getAuthor());