Hanserwei
40c05838f7
- 新增 QueryTool 类，提供 findAll 和 findAllByIdIn 方法用于查询用户 - 在 ChatClientConfiguration 中注册 QueryTool 为默认工具 - 创建 TestDataController，提供生成测试用户数据的接口- 新增 UserService 和 UserRepository，实现用户数据的批量插入和查询功能 - 将 ChatMessageDTO 从 model 包移动到 dto 包，优化包结构 -为 UserEntity 添加 createTime 和 updateTime 字段，完善实体类审计信息 - 新增 RedisConfig 配置类，为后续 Redis 功能做准备
29 lines
960 B
Java
29 lines
960 B
Java
package com.hanserwei.snailsai.controller;
|
|
|
|
import com.hanserwei.snailsai.entity.UserEntity;
|
|
import com.hanserwei.snailsai.service.UserService;
|
|
import org.springframework.web.bind.annotation.PostMapping;
|
|
import org.springframework.web.bind.annotation.RequestParam;
|
|
import org.springframework.web.bind.annotation.RestController;
|
|
|
|
import java.util.List;
|
|
|
|
@RestController
|
|
public class TestDataController {
|
|
|
|
private final UserService userService;
|
|
|
|
public TestDataController(UserService userService) {
|
|
this.userService = userService;
|
|
}
|
|
|
|
/**
|
|
* POST /api/test/generate-users?count=100
|
|
* 插入假数据用于分页测试
|
|
*/
|
|
@PostMapping("/api/test/generate-users")
|
|
public String generateTestData(@RequestParam(defaultValue = "100") int count) {
|
|
List<UserEntity> insertedUsers = userService.insertDummyUsers(count);
|
|
return String.format("成功插入了 %d 条假数据！", insertedUsers.size());
|
|
}
|
|
} |