From 6a3ace95da3d5317f85a10500f5be19cb36f4df3 Mon Sep 17 00:00:00 2001 From: LeeNan <925966998@qq.com> Date: Tue, 10 Oct 2023 11:52:52 +0800 Subject: [PATCH] =?UTF-8?q?=E6=8F=90=E4=BE=9B=E4=BA=86GlobalConfig?= =?UTF-8?q?=E7=9A=84customConfig=E7=9A=84getter/setter,=E4=BB=A3=E7=A0=81?= =?UTF-8?q?=E7=94=9F=E6=88=90=E5=99=A8=E5=8F=82=E6=95=B0=E5=88=97=E8=A1=A8?= =?UTF-8?q?=E5=8A=A0=E5=85=A5customConfig?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../java/com/mybatisflex/codegen/config/GlobalConfig.java | 7 +++++++ .../codegen/generator/impl/ControllerGenerator.java | 2 +- .../codegen/generator/impl/EntityGenerator.java | 2 +- .../codegen/generator/impl/MapperGenerator.java | 2 +- .../codegen/generator/impl/MapperXmlGenerator.java | 2 +- .../codegen/generator/impl/ServiceGenerator.java | 2 +- .../codegen/generator/impl/ServiceImplGenerator.java | 2 +- .../codegen/generator/impl/TableDefGenerator.java | 2 +- 8 files changed, 14 insertions(+), 7 deletions(-) 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..63c52008 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.put("customConfig", 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..927dff3f 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.put("customConfig", 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..143bbfb5 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.put("customConfig", 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..9d314964 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.put("customConfig", 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..1eedd5a2 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.put("customConfig", 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..a465dffc 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.put("customConfig", 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..0ebca1aa 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.put("customConfig", globalConfig.getCustomConfig()); globalConfig.getTemplateConfig().getTemplate().generate(params, templatePath, tableDefJavaFile); System.out.println("TableDef ---> " + tableDefJavaFile);