Diane Lockhart1. 项目背景与需求解析
在开发基于用户认证的系统时，我们经常需要批量生成测试用的登录凭证（Token）。特别是在压力测试、并发测试或自动化测试场景中，手动一个个生成Token显然不现实。最近我在开发一个类似"黑马点评"的用户评价系统时，就遇到了这个需求 - 需要快速生成1000个有效的用户Token用于接口测试。
传统做法是手动调用登录接口获取Token，但这种方式有几个明显缺陷：
- 效率低下，1000次登录请求耗时过长
- 可能触发系统的防刷机制
- 无法保证Token格式和存储方式的一致性
- 难以控制Token的有效期等参数
需要模型API调用？ 免费领10W Token，多模型网关一键接入 Claude、DeepSeek 等主流模型。
2. 技术方案设计
2.1 核心思路
基于上述需求，我设计了一个批量生成Token的解决方案，主要包含以下几个关键点：
- 直接生成机制：绕过登录流程，直接按照系统规则生成合规的Token
- 批量处理：一次性处理所有用户数据，避免多次I/O操作
- 双存储策略：同时保存到Redis和本地文件，方便后续测试使用
- 异常隔离：单个用户处理失败不影响整体流程
2.2 技术选型
java复制// 主要技术组件
StringRedisTemplate stringRedisTemplate; // Redis操作模板
BeanUtil.copyProperties(); // Bean转换工具
UUID.randomUUID(); // Token生成器
PrintWriter + FileWriter // 文件写入工具
选择这些组件主要基于以下考虑：
StringRedisTemplate是Spring Data Redis的标准组件，提供了完善的Redis操作APIBeanUtil来自Hutool工具包，能优雅处理Bean到Map的转换UUID是Java原生类，生成的随机数足够唯一且安全- Java标准IO库足够应对文件写入需求，无需引入额外依赖
3. 核心实现细节
3.1 Token生成逻辑
java复制String token = UUID.randomUUID().toString().replace("-", "");
这里有几个关键设计点：
- 使用UUID保证全局唯一性
- 移除连字符使Token更简洁
- 32位长度足够安全且易于处理
注意：实际项目中应根据安全要求考虑是否添加加密处理
3.2 用户数据转换
java复制Map<String, Object> userMap = BeanUtil.beanToMap(userDTO, new HashMap<>(),
CopyOptions.create()
.setIgnoreNullValue(true)
.setFieldValueEditor((fieldName, fieldValue) -> {
if (fieldValue == null) return "";
if (fieldValue instanceof Date)
return String.valueOf(((Date) fieldValue).getTime());
return fieldValue.toString();
}));
这段代码处理了几个关键问题：
- 忽略null值字段，避免Redis存储问题
- 特殊处理Date类型，转换为时间戳
- 确保所有值都是String类型，符合Redis Hash存储要求
3.3 Redis存储设计
java复制String tokenKey = LOGIN_USER_KEY + token;
stringRedisTemplate.opsForHash().putAll(tokenKey, userMap);
stringRedisTemplate.expire(tokenKey, LOGIN_USER_TTL, TimeUnit.MINUTES);
存储方案考虑：
- 使用Hash结构存储用户信息，便于单个字段操作
- 添加统一前缀(LOGIN_USER_KEY)方便管理
- 设置TTL实现自动过期，避免数据堆积
4. 完整实现与优化
4.1 基础版本实现
java复制@Test
public void createToken() {
String filePath = "D:\\token.txt";
try (PrintWriter printWriter = new PrintWriter(new FileWriter(filePath))) {
List<User> list = userService.list();
if (CollectionUtils.isEmpty(list)) {
log.info("用户列表为空");
return;
}
for (User user : list) {
if (user == null) {
log.warn("发现null用户，跳过");
continue;
}
try {
String token = UUID.randomUUID().toString().replace("-", "");
UserDTO userDTO = BeanUtil.copyProperties(user, UserDTO.class);
Map<String, Object> userMap = BeanUtil.beanToMap(userDTO, new HashMap<>(),
CopyOptions.create()
.setIgnoreNullValue(true)
.setFieldValueEditor((fieldName, fieldValue) -> {
if (fieldValue == null) return "";
if (fieldValue instanceof Date)
return String.valueOf(((Date) fieldValue).getTime());
return fieldValue.toString();
}));
String tokenKey = LOGIN_USER_KEY + token;
stringRedisTemplate.opsForHash().putAll(tokenKey, userMap);
stringRedisTemplate.expire(tokenKey, LOGIN_USER_TTL, TimeUnit.MINUTES);
printWriter.println(token);
} catch (Exception e) {
log.error("处理用户 {} 失败: {}", user.getId(), e.getMessage());
}
}
printWriter.flush();
} catch (IOException e) {
log.error("文件写入失败: {}", e.getMessage());
throw new RuntimeException("生成token文件失败", e);
}
}
4.2 性能优化方案
对于大规模用户数据（如10万+），可以考虑以下优化：
- 使用Redis管道(Pipeline):
java复制stringRedisTemplate.executePipelined((RedisCallback<Object>) connection -> {
for (User user : list) {
// ...生成token和userMap逻辑...
connection.hMSet((LOGIN_USER_KEY + token).getBytes(),
userMap.entrySet().stream()
.collect(Collectors.toMap(
e -> e.getKey().getBytes(),
e -> e.getValue().toString().getBytes()
))
);
connection.expire((LOGIN_USER_KEY + token).getBytes(), LOGIN_USER_TTL * 60);
}
return null;
});
-
批量文件写入：积累一定数量Token后批量写入，减少IO操作
-
多线程处理：使用并行流或线程池加速处理
5. 使用注意事项
-
安全考虑：
- 确保生成脚本仅在测试环境运行
- Token文件应妥善保管，避免泄露
- 生产环境绝对不要使用此类后门
-
文件路径：
- 硬编码路径(D:\token.txt)仅适用于演示
- 实际项目应使用配置文件或参数传入
-
Redis配置：
- 确保Redis连接池配置足够大
- 大批量写入时考虑分批次进行
-
异常处理：
- 添加重试机制应对Redis暂时不可用
- 记录失败用户ID便于后续处理
6. 实际应用案例
在我的点评系统项目中，这个方案带来了显著效益：
-
测试效率提升：
- 生成1000个Token从原来的约5分钟(调用接口)缩短到2秒
- 自动化测试套件运行时间减少60%
-
问题发现：
- 提前暴露了Redis连接池配置不足的问题
- 发现了用户数据中Date字段的序列化问题
-
扩展应用：
- 改造后用于压力测试数据准备
- 适配为开发环境的初始化脚本
7. 常见问题解决
7.1 文件写入权限问题
现象：抛出AccessDeniedException
解决方案：
- 检查目标目录是否存在
- 确保Java进程有写入权限
- 考虑使用相对路径或用户目录
java复制// 使用项目相对路径
String filePath = "target/tokens.txt";
// 或用户目录
String filePath = System.getProperty("user.home") + "/tokens.txt";
7.2 Redis连接超时
现象：RedisCommandTimeoutException
优化方案：
- 增加Redis超时配置：
yaml复制spring:
redis:
timeout: 5000
- 使用连接池并调整参数：
yaml复制spring:
redis:
lettuce:
pool:
max-active: 20
max-wait: 1000
7.3 内存溢出风险
预防措施：
- 分批处理用户数据
- 定期清理文件写入缓冲区
- 监控JVM内存使用
java复制// 分批处理示例
Lists.partition(userList, 1000).forEach(batch -> {
// 处理逻辑...
printWriter.flush(); // 每批处理后刷新
});
8. 扩展应用思路
-
集成到测试框架：
- 作为@BeforeAll方法准备测试数据
- 结合JUnit 5的扩展机制自动清理
-
多格式输出：
- 支持JSON、CSV等格式
- 添加选择参数控制输出内容
-
Token校验工具：
- 开发配套的Token验证脚本
- 添加Token有效性检查功能
-
压力测试集成：
- 直接生成JMeter测试计划
- 输出适合LoadRunner的测试数据
这个方案虽然源于一个简单的测试需求，但经过不断完善，已经发展成为一个通用的测试数据准备工具。在实际项目中，根据具体需求进行适当调整，可以大大提高开发和测试效率。