fix: 代码生成测试问题修复

youlai-codegen/src/main/java/com/youlai/codegen/controller/CodegenController.java

@@ -66,9 +66,9 @@ public class CodegenController {
     @GetMapping("/{tableName}/config")
     public Result<GenConfigForm> getGenConfigFormData(
             @Parameter(description = "表名", example = "sys_user") @PathVariable String tableName,
-            @Parameter(description = "数据源", example = "youlai_system") @RequestParam String datasourceKey
+            @Parameter(description = "数据源Key", example = "youlai_system") @RequestParam String dsKey
     ) {
-        GenConfigForm formData = genConfigService.getGenConfigFormData(tableName,datasourceKey);
+        GenConfigForm formData = genConfigService.getGenConfigFormData(tableName,dsKey);
         return Result.success(formData);
     }
 

youlai-codegen/src/main/java/com/youlai/codegen/mapper/DatabaseMapper.java

@@ -25,9 +25,11 @@ public interface DatabaseMapper extends BaseMapper {
 
     Page<TablePageVO> getTablePage(Page<TablePageVO> page, TablePageQuery queryParams);
 
+    @DS("#datasourceKey")
+    TableMetaData getTableMetadata(String tableName,String datasourceKey);
 
-    List<ColumnMetaData> getTableColumns(String tableName);
+    @DS("#datasourceKey")
+    List<ColumnMetaData> getTableColumns(String tableName,String datasourceKey);
 
 
-    TableMetaData getTableMetadata(String tableName);
 }

youlai-codegen/src/main/java/com/youlai/codegen/model/entity/GenConfig.java

@@ -1,10 +1,11 @@
 package com.youlai.codegen.model.entity;
 
-import com.baomidou.mybatisplus.annotation.TableName;
+import com.baomidou.mybatisplus.annotation.*;
-import com.youlai.common.base.BaseEntity;
 import lombok.Getter;
 import lombok.Setter;
 
+import java.time.LocalDateTime;
+
 /**
  * 代码生成基础配置
  *
@@ -14,7 +15,13 @@ import lombok.Setter;
 @TableName(value = "gen_config")
 @Getter
 @Setter
-public class GenConfig extends BaseEntity {
+public class GenConfig  {
+
+    /**
+     * 主键ID
+     */
+    @TableId(type = IdType.AUTO)
+    private Long id;
 
     /**
      * 表名
@@ -50,4 +57,17 @@ public class GenConfig extends BaseEntity {
      * 作者
      */
     private String author;
+
+    /**
+     * 创建时间
+     */
+    @TableField(fill = FieldFill.INSERT)
+    private LocalDateTime createTime;
+
+    /**
+     * 更新时间
+     */
+    @TableField(fill = FieldFill.INSERT_UPDATE)
+    private LocalDateTime updateTime;
+
 }

youlai-codegen/src/main/java/com/youlai/codegen/model/entity/GenFieldConfig.java

@@ -1,14 +1,14 @@
 package com.youlai.codegen.model.entity;
 
-import com.baomidou.mybatisplus.annotation.TableField;
+import com.baomidou.mybatisplus.annotation.*;
-import com.baomidou.mybatisplus.annotation.TableName;
 import com.fasterxml.jackson.annotation.JsonIgnore;
-import com.youlai.common.base.BaseEntity;
 import com.youlai.codegen.enums.FormTypeEnum;
 import com.youlai.codegen.enums.QueryTypeEnum;
 import lombok.Getter;
 import lombok.Setter;
 
+import java.time.LocalDateTime;
+
 /**
  * 字段生成配置实体
  *
@@ -18,7 +18,13 @@ import lombok.Setter;
 @TableName(value = "gen_field_config")
 @Getter
 @Setter
-public class GenFieldConfig extends BaseEntity {
+public class GenFieldConfig {
+
+    /**
+     * 主键ID
+     */
+    @TableId(type = IdType.AUTO)
+    private Long id;
 
 
     /**
@@ -102,4 +108,17 @@ public class GenFieldConfig extends BaseEntity {
      * 字典类型
      */
     private String dictType;
+
+
+    /**
+     * 创建时间
+     */
+    @TableField(fill = FieldFill.INSERT)
+    private LocalDateTime createTime;
+
+    /**
+     * 更新时间
+     */
+    @TableField(fill = FieldFill.INSERT_UPDATE)
+    private LocalDateTime updateTime;
 }

youlai-codegen/src/main/java/com/youlai/codegen/model/query/TablePageQuery.java

@@ -19,11 +19,11 @@ import java.util.List;
 @Setter
 public class TablePageQuery extends BasePageQuery {
 
-    @Schema(description="关键字(表名)")
+    @Schema(description="关键字(表名)",example = "sys_user")
     private String keywords;
 
-    @Schema(description="数据源Key")
+    @Schema(description="数据源Key",example = "master")
-    private String datasourceKey;
+    private String dsKey;
 
     /**
      * 排除的表名

youlai-codegen/src/main/java/com/youlai/codegen/service/impl/CodegenServiceImpl.java

@@ -133,9 +133,9 @@ public class CodegenServiceImpl implements CodegenService {
             previewVO.setFileName(fileName);
 
             /* 2. 生成文件路径 */
-            // 包名：com.youlai.boot
+            // 包名：com.youlai.mall
             String packageName = genConfig.getPackageName();
-            // 模块名：system
+            // 模块名：mall-order
             String moduleName = genConfig.getModuleName();
             // 子包名：controller
             String subpackageName = templateConfig.getSubpackageName();

youlai-codegen/src/main/java/com/youlai/codegen/service/impl/DatabaseServiceImpl.java

@@ -32,7 +32,7 @@ public class DatabaseServiceImpl implements DatabaseService {
      * @return 分页结果
      */
     @Override
-    @DS("#queryParams.datasourceKey")
+    @DS("#queryParams.dsKey")
     public Page<TablePageVO> getTablePage(TablePageQuery queryParams) {
         // 设置排除的表
         List<String> excludeTables = codegenProperties.getExcludeTables();

youlai-codegen/src/main/java/com/youlai/codegen/service/impl/GenConfigServiceImpl.java

@@ -4,6 +4,7 @@ import cn.hutool.core.collection.CollectionUtil;
 import cn.hutool.core.lang.Assert;
 import cn.hutool.core.util.StrUtil;
 import com.baomidou.dynamic.datasource.annotation.DS;
+import com.baomidou.dynamic.datasource.toolkit.DynamicDataSourceContextHolder;
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.youlai.common.core.exception.BusinessException;
@@ -66,9 +67,8 @@ public class GenConfigServiceImpl extends ServiceImpl<GenConfigMapper, GenConfig
         boolean hasGenConfig = genConfig != null;
         // 如果没有代码生成配置，则根据表的元数据生成默认配置
         if (genConfig == null) {
-            TableMetaData tableMetadata = this.getTableMetadata(tableName, datasourceKey);
+            TableMetaData tableMetadata = databaseMapper.getTableMetadata(tableName, datasourceKey);
             Assert.isTrue(tableMetadata != null, "未找到表元数据");
-
             genConfig = new GenConfig();
             genConfig.setTableName(tableName);
 
@@ -91,10 +91,16 @@ public class GenConfigServiceImpl extends ServiceImpl<GenConfigMapper, GenConfig
         List<GenFieldConfig> genFieldConfigs = new ArrayList<>();
 
         // 获取表的列
-        List<ColumnMetaData> tableColumns = this.getTableColumns(tableName, datasourceKey);
+        List<ColumnMetaData> tableColumns = databaseMapper.getTableColumns(tableName, datasourceKey);
         if (CollectionUtil.isNotEmpty(tableColumns)) {
+            try {
+                DynamicDataSourceContextHolder.push("master");
                 // 查询字段生成配置
-            List<GenFieldConfig> fieldConfigList = this.getGenFieldConfigs(tableName);
+                List<GenFieldConfig> fieldConfigList = genFieldConfigService.list(new LambdaQueryWrapper<GenFieldConfig>()
+                        .eq(GenFieldConfig::getConfigId, genConfig.getId())
+                        .orderByAsc(GenFieldConfig::getFieldSort)
+                );
+
                 Integer maxSort = fieldConfigList.stream()
                         .map(GenFieldConfig::getFieldSort)
                         .filter(Objects::nonNull)
@@ -123,6 +129,9 @@ public class GenConfigServiceImpl extends ServiceImpl<GenConfigMapper, GenConfig
                     }
                     genFieldConfigs.add(fieldConfig);
                 }
+            } finally {
+                DynamicDataSourceContextHolder.poll();
+            }
         }
         // 对genFieldConfigs按照fieldSort排序
         genFieldConfigs = genFieldConfigs.stream()
@@ -135,30 +144,6 @@ public class GenConfigServiceImpl extends ServiceImpl<GenConfigMapper, GenConfig
         return genConfigForm;
     }
 
-    @DS("master")
-    public List<GenFieldConfig> getGenFieldConfigs(String tableName) {
-        GenConfig genConfig = this.getOne(new LambdaQueryWrapper<>(GenConfig.class)
-                .eq(GenConfig::getTableName, tableName)
-        );
-        if (genConfig == null) {
-            return null;
-        }
-        return genFieldConfigService.list(new LambdaQueryWrapper<GenFieldConfig>()
-                .eq(GenFieldConfig::getConfigId, genConfig.getId())
-                .orderByAsc(GenFieldConfig::getFieldSort)
-        );
-    }
-
-    @DS("#datasourceKey")
-    public TableMetaData getTableMetadata(String tableName, String datasourceKey) {
-        return databaseMapper.getTableMetadata(tableName);
-    }
-
-    @DS("#datasourceKey")
-    public List<ColumnMetaData> getTableColumns(String tableName, String datasourceKey) {
-        return databaseMapper.getTableColumns(tableName);
-    }
-
     /**
      * 创建默认字段配置
      *

youlai-codegen/src/main/resources/bootstrap-dev.yml

@@ -27,7 +27,7 @@ spring:
         master:
           type: com.alibaba.druid.pool.DruidDataSource
           driver-class-name: com.mysql.cj.jdbc.Driver
-          url: jdbc:mysql://localhost:3306/youlai_generator?zeroDateTimeBehavior=convertToNull&useUnicode=true&characterEncoding=UTF-8&serverTimezone=Asia/Shanghai&autoReconnect=true&allowMultiQueries=true
+          url: jdbc:mysql://localhost:3306/youlai_codegen?zeroDateTimeBehavior=convertToNull&useUnicode=true&characterEncoding=UTF-8&serverTimezone=Asia/Shanghai&autoReconnect=true&allowMultiQueries=true
           username: root
           password: 123456
         youlai_system: # 系统库

youlai-codegen/src/main/resources/bootstrap.yml

@@ -1,6 +1,3 @@
-server:
-  port: 8810
-
 spring:
   application:
     name: youlai-codegen

youlai-common/common-core/src/main/java/com/youlai/common/core/config/RestTemplateConfig.java

@@ -24,8 +24,10 @@ public class RestTemplateConfig {
     @Bean
     public ClientHttpRequestFactory simpleClientHttpRequestFactory() {
         SimpleClientHttpRequestFactory factory = new SimpleClientHttpRequestFactory();
-        factory.setReadTimeout(5000);//单位 ms
+        // 读取超时时间 5s
-        factory.setConnectTimeout(5000);//单位 ms
+        factory.setReadTimeout(5000);
+        // 连接超时时间 5s
+        factory.setConnectTimeout(5000);
         return factory;
     }