diff --git a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/config/GlobalConfig.java b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/config/GlobalConfig.java index 4a79d41e..4acdfc34 100644 --- a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/config/GlobalConfig.java +++ b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/config/GlobalConfig.java @@ -233,6 +233,13 @@ public class GlobalConfig implements Serializable { customConfig.put(key, value); } + public Map getCustomConfig() { + return customConfig; + } + + public void setCustomConfig(Map customConfig) { + this.customConfig = customConfig; + } // === 分项配置 === /** diff --git a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/ControllerGenerator.java b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/ControllerGenerator.java index 9e6d104d..ec59a22d 100644 --- a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/ControllerGenerator.java +++ b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/ControllerGenerator.java @@ -71,7 +71,7 @@ public class ControllerGenerator implements IGenerator { params.put("javadocConfig", globalConfig.getJavadocConfig()); params.put("withSwagger", globalConfig.isEntityWithSwagger()); params.put("swaggerVersion", globalConfig.getSwaggerVersion()); - + params.putAll(globalConfig.getCustomConfig()); globalConfig.getTemplateConfig().getTemplate().generate(params, templatePath, controllerJavaFile); System.out.println("Controller ---> " + controllerJavaFile); diff --git a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/EntityGenerator.java b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/EntityGenerator.java index 28600499..e4b12817 100644 --- a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/EntityGenerator.java +++ b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/EntityGenerator.java @@ -72,7 +72,7 @@ public class EntityGenerator implements IGenerator { params.put("entityConfig", entityConfig); params.put("packageConfig", packageConfig); params.put("javadocConfig", globalConfig.getJavadocConfig()); - + params.putAll(globalConfig.getCustomConfig()); globalConfig.getTemplateConfig().getTemplate().generate(params, templatePath, entityJavaFile); System.out.println("Entity ---> " + entityJavaFile); diff --git a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/MapperGenerator.java b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/MapperGenerator.java index baa73240..12bd7d61 100644 --- a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/MapperGenerator.java +++ b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/MapperGenerator.java @@ -69,7 +69,7 @@ public class MapperGenerator implements IGenerator { params.put("mapperConfig", mapperConfig); params.put("packageConfig", packageConfig); params.put("javadocConfig", globalConfig.getJavadocConfig()); - + params.putAll(globalConfig.getCustomConfig()); globalConfig.getTemplateConfig().getTemplate().generate(params, templatePath, mapperJavaFile); System.out.println("Mapper ---> " + mapperJavaFile); diff --git a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/MapperXmlGenerator.java b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/MapperXmlGenerator.java index 7c18c0eb..ad6fbad3 100644 --- a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/MapperXmlGenerator.java +++ b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/MapperXmlGenerator.java @@ -66,7 +66,7 @@ public class MapperXmlGenerator implements IGenerator { Map params = new HashMap<>(2); params.put("table", table); params.put("packageConfig", packageConfig); - + params.putAll(globalConfig.getCustomConfig()); globalConfig.getTemplateConfig().getTemplate().generate(params, templatePath, mapperXmlFile); System.out.println("MapperXML ---> " + mapperXmlFile); diff --git a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/ServiceGenerator.java b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/ServiceGenerator.java index 34d156ed..3a7ce530 100644 --- a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/ServiceGenerator.java +++ b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/ServiceGenerator.java @@ -69,7 +69,7 @@ public class ServiceGenerator implements IGenerator { params.put("serviceConfig", serviceConfig); params.put("packageConfig", packageConfig); params.put("javadocConfig", globalConfig.getJavadocConfig()); - + params.putAll(globalConfig.getCustomConfig()); globalConfig.getTemplateConfig().getTemplate().generate(params, templatePath, serviceJavaFile); System.out.println("Service ---> " + serviceJavaFile); diff --git a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/ServiceImplGenerator.java b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/ServiceImplGenerator.java index 71ec4610..1d9a74a0 100644 --- a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/ServiceImplGenerator.java +++ b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/ServiceImplGenerator.java @@ -69,7 +69,7 @@ public class ServiceImplGenerator implements IGenerator { params.put("packageConfig", packageConfig); params.put("serviceImplConfig", serviceImplConfig); params.put("javadocConfig", globalConfig.getJavadocConfig()); - + params.putAll(globalConfig.getCustomConfig()); globalConfig.getTemplateConfig().getTemplate().generate(params, templatePath, serviceImplJavaFile); System.out.println("ServiceImpl ---> " + serviceImplJavaFile); diff --git a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/TableDefGenerator.java b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/TableDefGenerator.java index 1bf83d06..420c6c2a 100644 --- a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/TableDefGenerator.java +++ b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/TableDefGenerator.java @@ -69,7 +69,7 @@ public class TableDefGenerator implements IGenerator { params.put("packageConfig", packageConfig); params.put("tableDefConfig", tableDefConfig); params.put("javadocConfig", globalConfig.getJavadocConfig()); - + params.putAll(globalConfig.getCustomConfig()); globalConfig.getTemplateConfig().getTemplate().generate(params, templatePath, tableDefJavaFile); System.out.println("TableDef ---> " + tableDefJavaFile);