JMeter-SSE响应数据自动化

结构图

背景：

需要写一个JMeter脚本来进行自动化测试，主要是通过接口调用一些东西，同时要对响应的数据进行处理，包括不限于错误信息的输出。

1.SSE(摘录)

SSE（Server-Sent Events）是一种基于HTTP协议、允许服务器主动向客户端推送实时更新的技术‌。它特别适用于单向数据流的实时场景，例如消息通知、AI对话流式响应等，通过保持长连接实现持续数据传输。

2.实现思路

2.1 用户自定义变量组件

主要是用来统一更换和维护环境变量的，比如线上、线下环境host的切换

2.2 HTTP信息头管理

设置https请求的信息头，比如token，数据格式等等

2.3 CSV 数据文件设置

参数如下图设置

变量名称：appId,query,appName,needFiles,file

对应vars中的变量，变量值是根据分隔带JMeter自动处理的，变量名称 数量和分隔后的变量值 数量不对应也没影响，两者缺少的值会忽略或者设置为空值

2.4 HTTP请求组件

需要设置消息体数据，请求url等

下面三个插件，归类到HTTP请求的子目录下：如图

2.4.1 JSR223 预处理程序

主要来处理一下请求中消息体数据中的一个参数,fileParam

根据csv文件中的标记来确定fileParam的具体值，如下代码

import org.json.JSONObject;
import org.json.JSONArray;
import java.util.ArrayList;//判断是否需要 files 参数
private Boolean needFiles(String str) {if("1".equals(str)) {return true;}else {return false;}
}//设置 files 参数
private void setFiles(String fileParams) {try{//将fileParams转为json格式JSONObject jsonResponse = new JSONObject(fileParams);// 提取各个参数String filename = jsonResponse.optString("xxx", "");String fileHash = jsonResponse.optString("xxx", "");Integer filesize = jsonResponse.optInt("xxx", 0);  // Integer类型String extension = jsonResponse.optString("xxx", "");String mimeType = jsonResponse.optString("xxx", "");// 创建 JSON 对象数组（List<Map> 格式）JSONArray fileParamsArray = new JSONArray();JSONObject fileObj = new JSONObject();fileObj.put("xxx", xxx);fileObj.put("xxx", xxx);fileObj.put("xxx", xxx);fileObj.put("xxx", xxx);fileObj.put("xxx", xxx);fileParamsArray.put(fileObj);// 存入 vars（JSON 字符串）vars.put("fileParam", fileParamsArray.toString());} catch (Exception e) {log.error("设置 files 参数 失败!", e);prev.setSuccessful(false);}	}    try {String str = vars.get("needFiles");	String fileParams = vars.get("file");if(needFiles(str)){//需要文件参数setFiles(fileParams);}else{//不需要文件参数，设置为空JSONArray fileParamsArray = new JSONArray();vars.put("fileParam",fileParamsArray.toString());	}
} catch (Exception e) {log.error("判断是否需要文件参数 失败!", e);prev.setSuccessful(false);
}

2.4.2 JSR223 后置处理程序

对SSE响应的数据进行处理和判断，确定好哪个数据是一次请求结束的标记

下面代码是根据 event 含有 message_end 字段来做执行成功的标记
error字段来错失败的标记，同时进入断言

下面的代码是逐行匹配、逐个处理 SSE 事件，适合实时响应场景

import org.apache.jmeter.samplers.SampleResult;
import java.util.regex.Pattern;
import java.util.regex.Matcher;
import org.json.JSONObject;
import org.json.JSONException;// 判断是否是流式响应
private Boolean isStreamingResponse(String response) {return response.contains("data: {");
}// 非流式响应处理
private void handleNonStreamingResponse(String response, SampleResult prev) throws Exception {JSONObject jsonResponse = new JSONObject(response);String msg = jsonResponse.get("msg");if ("智能体不存在".equals(msg)) {vars.put("response_type", "智能体不存在");vars.put("error_msg", msg);vars.put("isExist", "false");} else {vars.put("response_type", "非流式响应,未知错误");vars.put("error_msg", msg);vars.put("isExist", "false");}
}// 流式响应处理
private void handleStreamingResponse(String response, SampleResult prev) throws Exception {vars.put("response_type", "streaming");// 正则表达式，确保匹配完整JSONPattern pattern = Pattern.compile("data:\\s*(\\{[^{}]*\\})", Pattern.DOTALL)Matcher matcher = pattern.matcher(response);while (matcher.find()) {try {String eventJson = matcher.group(1).trim();// 处理可能的UTF-8 BOM头if (eventJson.startsWith("\uFEFF")) {eventJson = eventJson.substring(1);}JSONObject jsonResponse = new JSONObject(eventJson);String eventType = jsonResponse.optString("event");if ("message_end".equals(eventType)) {vars.put("response_type", "智能体执行成功");vars.put("error_msg", "智能体执行成功");vars.put("isExist", "true");} else if ("error".equals(eventType)) {              vars.put("response_type", "智能体执行失败");String errorMsg = jsonResponse.get("message");vars.put("error_msg", errorMsg);vars.put("isExist", "false");}} catch (JSONException e) {log.warn("SSE事件JSON解析失败，跳过该事件: " + vars.get("appName"));}}
}SampleResult prev = ctx.getPreviousResult();String response = prev.getResponseDataAsString();//添加APPID信息
vars.put("APPID",vars.get("appId"));//每次重置isExist的值，避免上次结果影响本次
vars.put("isExist", "true");try {if (!isStreamingResponse(response)) {handleNonStreamingResponse(response, prev);} else {handleStreamingResponse(response, prev);}
} catch (Exception e) {log.error("处理响应失败!", e);prev.setSuccessful(false);
}

handleStreamingResponse方法：
对每次符合要求的数据进行处理和判断

// 正则表达式，确保匹配完整JSON
Pattern pattern = Pattern.compile(“data:\s*(\{[^{}]*\})”, Pattern.DOTALL)
Matcher matcher = pattern.matcher(response);

data:\s*
匹配字符串 “data:”，后面跟 0个或多个空白字符（\s* 包括空格、换行符等）。

(\{[^{}]*\})
\{ 匹配左花括号 {（{ 需要转义）。
[^{}]* 匹配 任意字符（除了 { 和 }）0次或多次，确保匹配的是 单层花括号 的内容。
\} 匹配右花括号 }。
() 表示捕获组，最终提取的是花括号内的内容。

Pattern.DOTALL
让 . 匹配 包括换行符在内的所有字符，确保多行文本也能被正确匹配。
.
Matcher matcher = pattern.matcher(response);
用编译好的正则模式 pattern 去匹配输入的字符串 response。
matcher 对象可以用于查找、提取符合正则规则的部分。

while (matcher.find()) {
try {
String eventJson = matcher.group(1).trim();

matcher.find() 每次找到一个匹配项后，会移动内部指针，直到所有匹配项被遍历完。

matcher.group(1) 提取正则中 第一个捕获组（即 ({[^{}]*}) 匹配的 {…} 部分）。

2.4.3 JSR JSR223 Assertion

进行断言处理，处理需要输出的信息

// 断言
if ("false".equals(vars.get("isExist"))) {// 获取智能体名称String appName = vars.get("appName");// 获取具体失败类型String respone_type = vars.get("response_type");// 获取error_messageString error_message = vars.get("error_msg");// 获取APPIDString appId = vars.get("APPID");//执行失败AssertionResult.setFailure(true); // 标记断言失败AssertionResult.setFailureMessage(appName + "\n\t" + " 智能体ID:" + appId + "\n" + "\t 错误原因:"+ respone_type + "\n" + "\t error_message:" + error_message);// 添加到标签列//prev.setSampleLabel(vars.get("respone_type")) // 修改响应消息为message变量的内容prev.setResponseMessage(respone_type);// 添加调试信息log.info("智能体名称:", appName);log.info("智能体ID:", appId);log.info("错误原因:", respone_type);log.info("error_message:", error_message);
}