From c00fbffd4cd3e7a9a80457e8ab6ee4ab740c1740 Mon Sep 17 00:00:00 2001 From: SimonSun Date: Sun, 12 Aug 2018 09:10:16 +0800 Subject: [PATCH 1/2] =?UTF-8?q?=E9=BB=98=E8=AE=A4=E7=94=9F=E6=88=90model?= =?UTF-8?q?=E3=80=81mapper=E3=80=81repository=E5=92=8Cservice=E3=80=82?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/main/java/com/simon/common/code/CodeGenerator.java | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/src/main/java/com/simon/common/code/CodeGenerator.java b/src/main/java/com/simon/common/code/CodeGenerator.java index 55b9bd6..1e61372 100644 --- a/src/main/java/com/simon/common/code/CodeGenerator.java +++ b/src/main/java/com/simon/common/code/CodeGenerator.java @@ -122,10 +122,9 @@ public class CodeGenerator { */ private static void genCodeByCustomModelName(String tableName, String modelName) { genModelAndMapper(tableName, modelName); - //genRepository(tableName, modelName); - //genService(tableName, modelName); + genRepository(tableName, modelName); + genService(tableName, modelName); //genController(tableName, modelName); - } -- Gitee From 7c3eef3b9a3cb0298233e809ce7b0a0377b3b196 Mon Sep 17 00:00:00 2001 From: SimonSun Date: Sun, 12 Aug 2018 09:23:52 +0800 Subject: [PATCH 2/2] =?UTF-8?q?=E7=94=9F=E6=88=90=E7=9A=84=E4=BB=A3?= =?UTF-8?q?=E7=A0=81=E6=A8=A1=E5=9D=97=E9=85=8D=E7=BD=AE=E3=80=82?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../com/simon/common/code/CodeGenerator.java | 30 +++++++++++++++---- src/main/resources/code-gen.properties | 4 ++- 2 files changed, 28 insertions(+), 6 deletions(-) diff --git a/src/main/java/com/simon/common/code/CodeGenerator.java b/src/main/java/com/simon/common/code/CodeGenerator.java index 1e61372..f5efe48 100644 --- a/src/main/java/com/simon/common/code/CodeGenerator.java +++ b/src/main/java/com/simon/common/code/CodeGenerator.java @@ -51,11 +51,14 @@ public class CodeGenerator { private static String PACKAGE_PATH_SERVICE_IMPL;//生成的Service实现存放路径 private static String PACKAGE_PATH_CONTROLLER;//生成的Controller存放路径 + private static String GEN_MODULES;//要生成的模块 + protected static String AUTHOR;//@author protected static final String CREATE = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss").format(new Date());//@date + private static Properties prop; static { - Properties prop = new Properties(); + prop = new Properties(); try { prop.load(CodeGenerator.class.getResourceAsStream("/code-gen.properties")); JDBC_URL = prop.getProperty("jdbc_url"); @@ -79,6 +82,8 @@ public class CodeGenerator { PACKAGE_PATH_SERVICE = packageConvertPath(SERVICE_PACKAGE); PACKAGE_PATH_SERVICE_IMPL = packageConvertPath(SERVICE_IMPL_PACKAGE); PACKAGE_PATH_CONTROLLER = packageConvertPath(CONTROLLER_PACKAGE); + + GEN_MODULES = prop.getProperty("gen_modules"); } catch (IOException e) { e.printStackTrace(); } @@ -121,10 +126,25 @@ public class CodeGenerator { * @param modelName 自定义的 Model 名称 */ private static void genCodeByCustomModelName(String tableName, String modelName) { - genModelAndMapper(tableName, modelName); - genRepository(tableName, modelName); - genService(tableName, modelName); - //genController(tableName, modelName); + if(StringUtils.isEmpty(GEN_MODULES)){ + genModelAndMapper(tableName, modelName); + genRepository(tableName, modelName); + genService(tableName, modelName); + //genController(tableName, modelName); + }else{ + if(GEN_MODULES.toLowerCase().contains("modelandmapper")){ + genModelAndMapper(tableName, modelName); + } + if(GEN_MODULES.toLowerCase().contains("repository")){ + genRepository(tableName, modelName); + } + if(GEN_MODULES.toLowerCase().contains("service")){ + genService(tableName, modelName); + } + if(GEN_MODULES.toLowerCase().contains("controller")){ + genController(tableName, modelName); + } + } } diff --git a/src/main/resources/code-gen.properties b/src/main/resources/code-gen.properties index 6f17f0c..dc6edc5 100644 --- a/src/main/resources/code-gen.properties +++ b/src/main/resources/code-gen.properties @@ -12,4 +12,6 @@ resources_path=/src/test/resources # ĿƣԼĿ޸ base_package=com.simon # Mapperӿڵȫ޶(ڶᵽĺļ̳нӿMapper) -mapper_interface_reference=com.simon.common.mapper.MyMapper \ No newline at end of file +mapper_interface_reference=com.simon.common.mapper.MyMapper +# Ҫɵģ飬ѡֵmodelAndMapper,repository,service,controllerָmodelAndMapper,repository,service +gen_modules=modelAndMapper,repository,service \ No newline at end of file -- Gitee