jmeter登录接口生成一批token并写入csv文件

背景：大部分项目真实的业务接口都是需要token鉴权的，想对一批核心业务接口进行并发压测，必然要先生成一批token给这些接口并发循环调用。

基本的思路是这样的：一批手机号csv文件 -》登录接口循环读取csv文件并生成token -》每次调用生成的token写入到存放token的csv文件中 -》其他所有接口读取csv文件中的token -》生成汇总报告。

jmeter 测试计划的整体结构图：

核心步骤：

（一）配置手机号csv数据文件设置

（二）添加json提取器，提取token变量

（三）添加Bean shell后置处理器

这段BeanShell的脚本代码就是将登录接口获得的token写入到csv文件中。

尤其注意，此处fw.write("${token}")这样写会报文件写入异常，只能用vars.get("token")，我这边压测的环境会报错。

代码粘贴如下：

import java.io.File;

import java.io.FileWriter;

import java.io.IOException;

// 获取token变量值

String token = vars.get("token");

// 文件路径 - 注意使用双反斜杠或正斜杠

String filePath = "D:\\jmeter-jmx\\压测\\生成号生成的token.csv";

try {

FileWriter fw = new FileWriter(filePath, true); // true表示追加模式

fw.write(token + "\n");

fw.close();

} catch (IOException e) {

log.error("写入文件失败: " + e.getMessage());

}

复制代码

import java.io.File;
import java.io.FileWriter;
import java.io.IOException;

// 获取token变量值
String token = vars.get("token");

// 文件路径 - 注意使用双反斜杠或正斜杠
String filePath = "D:\\jmeter-jmx\\压测\\生成号生成的token.csv"; 

try {
    FileWriter fw = new FileWriter(filePath, true); // true表示追加模式
    fw.write(token + "\n");
    fw.close();
} catch (IOException e) {
    log.error("写入文件失败: " + e.getMessage());
}

（四）读取csv中的token供其他所有需要鉴权压测的接口使用

经过前面的步骤已经生成了token备用，此时可以禁用手机号登录接口。

注意是在线程组下添加csv数据文件，这样只需加载一次csv数据文件就能被线程组下的接口共用。

因为我的token文件首行没有标题，所以不忽略首行。

查询用户信息接口要鉴权，所以请求头要带上token。

（五）配置压测参数和观察结果汇总

用10个线程持续压测10s