201206030/novel
1
0
Fork 0
mirror of https://github.com/201206030/novel.git synced 2025-04-27 07:30:50 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

test(Generator.java): 禁用 service 和 controller 的代码生成

Browse Source
This commit is contained in:
xiongxiaoyang 2022-05-12 07:45:59 +08:00
parent 6951a0aad4
commit f7e0b68442
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/test/java/io/github/xxyopen/novel/generator/Generator.java
View File

@ -3,6 +3,7 @@ package io.github.xxyopen.novel.generator;
import com.baomidou.mybatisplus.generator.FastAutoGenerator; import com.baomidou.mybatisplus.generator.FastAutoGenerator;
import com.baomidou.mybatisplus.generator.config.OutputFile; import com.baomidou.mybatisplus.generator.config.OutputFile;
import com.baomidou.mybatisplus.generator.config.TemplateType;
import java.util.Arrays; import java.util.Arrays;
import java.util.Collections; import java.util.Collections;
@ -39,7 +40,7 @@ public class Generator {
public static void main(String[] args) { public static void main(String[] args) {
// 传入需要生成的表名，多个用英文逗号分隔，所有用 all 表示 // 传入需要生成的表名，多个用英文逗号分隔，所有用 all 表示
genCode("all"); genCode("sys_user");
} }
@ -49,6 +50,7 @@ public class Generator {
*/ */
private static void genCode(String tables) { 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) 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 -> { .globalConfig(builder -> {
builder.author(USERNAME) // 设置作者 builder.author(USERNAME) // 设置作者
@ -58,8 +60,8 @@ public class Generator {
.fileOverride() // 覆盖已生成文件 .fileOverride() // 覆盖已生成文件
.commentDate("yyyy/MM/dd") .commentDate("yyyy/MM/dd")
.outputDir(PROJECT_PATH + JAVA_PATH); // 指定输出目录 .outputDir(PROJECT_PATH + JAVA_PATH); // 指定输出目录
}) })
// 包配置
.packageConfig(builder -> builder.parent(BASE_PACKAGE) // 设置父包名 .packageConfig(builder -> builder.parent(BASE_PACKAGE) // 设置父包名
.entity("dao.entity") .entity("dao.entity")
.service("service") .service("service")
@ -67,6 +69,11 @@ public class Generator {
.mapper("dao.mapper") .mapper("dao.mapper")
.controller("controller.front") .controller("controller.front")
.pathInfo(Collections.singletonMap(OutputFile.mapperXml, PROJECT_PATH + RESOURCE_PATH + "/mapper"))) .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)) // 设置需要生成的表名 .strategyConfig(builder -> builder.addInclude(getTables(tables)) // 设置需要生成的表名
.controllerBuilder() .controllerBuilder()
.enableRestStyle() .enableRestStyle()