feat: 项目集成mybatis-plus-generator 代码生成器

Closes https://gitee.com/youlaiorg/youlai-boot/issues/I80MH1
2024-04-10 22:35:07 +08:00
parent 58d30d56ca
commit 4b4bee4cd8
14 changed files with 1126 additions and 2 deletions
--- a/src/test/java/com/youlai/system/generator/FastAutoGeneratorTest.java
+++ b/src/test/java/com/youlai/system/generator/FastAutoGeneratorTest.java
@@ -0,0 +1,82 @@
+package com.youlai.system.generator;
+
+import com.baomidou.mybatisplus.generator.FastAutoGenerator;
+
+import com.baomidou.mybatisplus.generator.config.DataSourceConfig;
+import com.baomidou.mybatisplus.generator.config.OutputFile;
+import com.baomidou.mybatisplus.generator.config.builder.CustomFile;
+
+import java.sql.SQLException;
+import java.util.ArrayList;
+import java.util.Collections;
+import java.util.List;
+
+/**
+ * 代码交互式生成
+ * <p>
+ * 代码生成、MySQL表生成代码、自动代码生成
+ *
+ * @author Ray Hao
+ * @see <a href="https://baomidou.com/pages/981406/">代码生成器配置新</a>
+ * @since 2024/4/9
+ */
+public class FastAutoGeneratorTest {
+
+    private static final DataSourceConfig.Builder DATA_SOURCE_CONFIG = new DataSourceConfig
+            .Builder("jdbc:mysql://localhost:3306/youlai_boot?serverTimezone=Asia/Shanghai" , "root" , "123456");
+
+    /**
+     * 执行 run
+     */
+    public static void main(String[] args) {
+        FastAutoGenerator.create(DATA_SOURCE_CONFIG)
+                // 全局配置
+                .globalConfig((scanner, builder) -> {
+                    builder.outputDir(System.getProperty("user.dir") + "/src/main/java")
+                            .author("Ray Hao") // 设置作者
+                    ;
+                })
+                // 包配置
+                .packageConfig(builder -> builder
+                        .parent("com.youlai.system")
+                        .entity("model.entity")
+                        .mapper("mapper")
+                        .service("service")
+                        .serviceImpl("service.impl")
+                        .controller("controller")
+                        .pathInfo(Collections.singletonMap(OutputFile.xml, System.getProperty("user.dir") + "/src/main/resources/mapper"))
+                )
+                // 注入配置(设置扩展类的模板路径和包路径)
+                .injectionConfig(consumer -> {
+                    List<CustomFile> customFiles = new ArrayList<>();
+                    customFiles.add(new CustomFile.Builder().fileName("DTO.java").templatePath("/templates/dto.java.vm").packageName("model.dto").build());
+                    customFiles.add(new CustomFile.Builder().fileName("VO.java").templatePath("/templates/vo.java.vm").packageName("model.vo").build());
+                    customFiles.add(new CustomFile.Builder().fileName("BO.java").templatePath("/templates/bo.java.vm").packageName("model.bo").build());
+                    customFiles.add(new CustomFile.Builder().fileName("PageQuery.java").templatePath("/templates/pageQuery.java.vm").packageName("model.query").build());
+                    customFiles.add(new CustomFile.Builder().fileName("Form.java").templatePath("/templates/form.java.vm").packageName("model.form").build());
+                    customFiles.add(new CustomFile.Builder().fileName("Converter.java").templatePath("/templates/converter.java.vm").packageName("converter").build());
+                    consumer.customFile(customFiles);
+                })
+                // 策略配置
+                .strategyConfig((scanner, builder) -> {
+
+                            builder.entityBuilder()
+                                    .enableLombok() // 是否使用lombok
+                                    //.enableFileOverride() // 是否覆盖文件
+                                    .logicDeleteColumnName("deleted") // 逻辑删除字段名
+                            ;
+
+                            builder.mapperBuilder()
+                                    .enableBaseColumnList()
+                                    .enableBaseResultMap()
+                            ;
+
+                            builder.addTablePrefix("sys_") // 过滤移除表前缀 sys_user 表生成的实体类 User.java
+                                    .addInclude(scanner.apply("请输入表名，多个表名用,隔开"));
+                        }
+                )
+                .execute()
+
+        ;
+    }
+}