From f7e0b684420cdf79b85f0860f203d5d15fa2b044 Mon Sep 17 00:00:00 2001
From: xiongxiaoyang <773861846@qq.com>
Date: Thu, 12 May 2022 07:45:59 +0800
Subject: [PATCH] =?UTF-8?q?test(Generator.java):=20=E7=A6=81=E7=94=A8=20se?=
 =?UTF-8?q?rvice=20=E5=92=8C=20controller=20=E7=9A=84=E4=BB=A3=E7=A0=81?=
 =?UTF-8?q?=E7=94=9F=E6=88=90?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../io/github/xxyopen/novel/generator/Generator.java  | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/src/test/java/io/github/xxyopen/novel/generator/Generator.java b/src/test/java/io/github/xxyopen/novel/generator/Generator.java
index 8316cc4..f136da4 100644
--- a/src/test/java/io/github/xxyopen/novel/generator/Generator.java
+++ b/src/test/java/io/github/xxyopen/novel/generator/Generator.java
@@ -3,6 +3,7 @@ package io.github.xxyopen.novel.generator;
 
 import com.baomidou.mybatisplus.generator.FastAutoGenerator;
 import com.baomidou.mybatisplus.generator.config.OutputFile;
+import com.baomidou.mybatisplus.generator.config.TemplateType;
 
 import java.util.Arrays;
 import java.util.Collections;
@@ -39,7 +40,7 @@ public class Generator {
     public static void main(String[] args) {
 
         // 传入需要生成的表名，多个用英文逗号分隔，所有用 all 表示
-        genCode("all");
+        genCode("sys_user");
 
     }
 
@@ -49,6 +50,7 @@ public class Generator {
      */
     private static void genCode(String tables) {
 
+        // 全局配置
         FastAutoGenerator.create(String.format("jdbc:mysql://%s:%s/%s?useUnicode=true&characterEncoding=utf-8&useSSL=false&serverTimezone=Asia/Shanghai", DATABASE_IP, DATABASE_PORT, DATABASE_NAME), DATABASE_USERNAME, DATABASE_PASSWORD)
                 .globalConfig(builder -> {
                     builder.author(USERNAME) // 设置作者
@@ -58,8 +60,8 @@ public class Generator {
                             .fileOverride() // 覆盖已生成文件
                             .commentDate("yyyy/MM/dd")
                             .outputDir(PROJECT_PATH + JAVA_PATH); // 指定输出目录
-
                 })
+                // 包配置
                 .packageConfig(builder -> builder.parent(BASE_PACKAGE) // 设置父包名
                         .entity("dao.entity")
                         .service("service")
@@ -67,6 +69,11 @@ public class Generator {
                         .mapper("dao.mapper")
                         .controller("controller.front")
                         .pathInfo(Collections.singletonMap(OutputFile.mapperXml, PROJECT_PATH + RESOURCE_PATH + "/mapper")))
+                // 模版配置
+                .templateConfig(builder -> builder.disable(TemplateType.SERVICE)
+                        .disable(TemplateType.SERVICEIMPL)
+                        .disable(TemplateType.CONTROLLER))
+                // 策略配置
                 .strategyConfig(builder -> builder.addInclude(getTables(tables)) // 设置需要生成的表名
                         .controllerBuilder()
                         .enableRestStyle()