xiaoe/test
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

用例步骤表触发JMeter调用接口

Browse Source
This commit is contained in:
liangdaliang
2025-02-14 16:10:23 +08:00
parent 186665beae
commit 7103c9eab0
7 changed files with 1030 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
test-common/pom.xml
View File

@@ -52,13 +52,13 @@
<groupId>org.apache.commons</groupId>
<artifactId>commons-lang3</artifactId>
</dependency>
<!-- JSON工具类 -->
<dependency>
<groupId>com.fasterxml.jackson.core</groupId>
<artifactId>jackson-databind</artifactId>
</dependency>
<!-- 阿里JSON解析器 -->
<dependency>
<groupId>com.alibaba.fastjson2</groupId>
@@ -119,6 +119,35 @@
<artifactId>jakarta.servlet-api</artifactId>
</dependency>
<!-- jmeter包 -->
<dependency>
<groupId>org.apache.jmeter</groupId>
<artifactId>ApacheJMeter_core</artifactId>
<version>5.4.3</version>
</dependency>
<dependency>
<groupId>org.apache.jmeter</groupId>
<artifactId>ApacheJMeter_components</artifactId>
<version>5.4.3</version>
</dependency>
<dependency>
<groupId>org.apache.jmeter</groupId>
<artifactId>ApacheJMeter_http</artifactId>
<version>5.4.3</version>
</dependency>
<dependency>
<groupId>junit</groupId>
<artifactId>junit</artifactId>
<version>4.13.1</version>
<scope>test</scope>
</dependency>
<dependency>
<groupId>com.google.code.gson</groupId>
<artifactId>gson</artifactId>
<version>2.10.1</version>
</dependency>
</dependencies>
</project>
</project>

421
test-common/src/main/java/com/test/common/utils/JMeterUtil.java Normal file
View File

@@ -0,0 +1,421 @@
package com.test.common.utils;
import com.google.gson.Gson;
import com.google.gson.reflect.TypeToken;
import org.apache.jmeter.config.Arguments;
import org.apache.jmeter.config.gui.ArgumentsPanel;
import org.apache.jmeter.control.LoopController;
import org.apache.jmeter.control.gui.LoopControlPanel;
import org.apache.jmeter.control.gui.TestPlanGui;
import org.apache.jmeter.engine.StandardJMeterEngine;
import org.apache.jmeter.protocol.http.control.Header;
import org.apache.jmeter.protocol.http.control.HeaderManager;
import org.apache.jmeter.protocol.http.control.gui.HttpTestSampleGui;
import org.apache.jmeter.protocol.http.gui.HeaderPanel;
import org.apache.jmeter.protocol.http.sampler.HTTPSamplerProxy;
import org.apache.jmeter.reporters.ResultCollector;
import org.apache.jmeter.reporters.Summariser;
import org.apache.jmeter.samplers.SampleSaveConfiguration;
import org.apache.jmeter.save.SaveService;
import org.apache.jmeter.testelement.TestElement;
import org.apache.jmeter.testelement.TestPlan;
import org.apache.jmeter.testelement.property.BooleanProperty;
import org.apache.jmeter.testelement.property.ObjectProperty;
import org.apache.jmeter.testelement.property.StringProperty;
import org.apache.jmeter.threads.ThreadGroup;
import org.apache.jmeter.threads.gui.ThreadGroupGui;
import org.apache.jmeter.util.JMeterUtils;
import org.apache.jorphan.collections.HashTree;
import org.w3c.dom.Document;
import org.w3c.dom.Element;
import org.w3c.dom.Node;
import org.w3c.dom.NodeList;
import javax.xml.parsers.DocumentBuilder;
import javax.xml.parsers.DocumentBuilderFactory;
import java.io.File;
import java.io.FileOutputStream;
import java.lang.reflect.Type;
import java.net.MalformedURLException;
import java.net.URL;
import java.net.URLEncoder;
import java.nio.charset.StandardCharsets;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
/**
* @author liangdaliang
* @Description：JMeterter工具类
* @date 2025-02-12 9:12
*/
public class JMeterUtil {
/**
* 生成jmeter测试执行计划并获取执行结果
* @param id
* @param url
* @param port
* @param method
* @param paramJson
* @param requestHeader
* @param jmeterHomePath
* @return
*/
public static String getJmeterResult(Long id, String url, int port, String method, String paramJson, String requestHeader, String jmeterHomePath) {
String result = "";
try {
// 1. 初始化 JMeter
JMeterUtils.loadJMeterProperties(jmeterHomePath + "/bin/jmeter.properties");
JMeterUtils.setJMeterHome(jmeterHomePath);
JMeterUtils.initLocale();
// 2. 创建测试计划
TestPlan testPlan = createTestPlan("Dynamic Test Plan");
// 3. 创建线程组
ThreadGroup threadGroup = createThreadGroup(1, 1);
// 4. 创建 HTTP 请求
HTTPSamplerProxy httpSampler = null;
if ("POST".equals(method)) {
httpSampler = createPostHttpSampler(
url, // 请求地址
port,
paramJson // 请求参数
);
} else {
httpSampler = createGetHttpSampler(
url, // 请求地址
port,
paramJson // 请求参数
);
}
// 5. 创建请求头管理器
Map<String, String> headerMap = new HashMap<>();
if (!StringUtils.isEmpty(requestHeader)) {
Gson gson = new Gson();
// 定义 Map 的类型
Type type = new TypeToken<Map<String, String>>() {}.getType();
// 将 JSON 字符串转换为 Map
headerMap = gson.fromJson(requestHeader, type);
}
HeaderManager headerManager = createHeaderManager(headerMap);
// 6. 获取结果：如汇总报告、查看结果树
String replayLogPath = jmeterHomePath + "/replay_result"+ id +".log";
List<ResultCollector> resultCollector = getResultCollector(replayLogPath);
// 7. 构建测试树
HashTree testPlanTree = new HashTree();
HashTree threadGroupTree = testPlanTree.add(testPlan);
threadGroupTree.add(threadGroup);
threadGroupTree.add(httpSampler);
threadGroupTree.add(headerManager);
threadGroupTree.add(resultCollector);
// 8. 保存测试计划为 JMX 文件
File jmxFile = new File(jmeterHomePath + "/dynamic_test_plan"+ id +".jmx");
SaveService.saveTree(testPlanTree, new FileOutputStream(jmxFile));
// 9. 运行 JMeter 测试
File resultFile = new File(jmeterHomePath + "/replay_result"+ id +".log");
if (resultFile.exists()) {
resultFile.delete();
}
StandardJMeterEngine jmeter = new StandardJMeterEngine();
jmeter.configure(testPlanTree);
jmeter.run();
while (jmeter.isActive()) {
System.out.println("JMeter isActive！");
Thread.sleep(1000); // 每隔 1 秒检查一次
}
// 10. 获取响应结果
result = getResultMessageFromFile(jmeterHomePath + "/replay_result"+ id +".log");
System.out.println("JMeter 测试执行完成！");
} catch (Exception e) {
e.printStackTrace();
}
return result;
}
/**
* 创建测试计划
* @param name
* @return
*/
private static TestPlan createTestPlan(String name) {
TestPlan testPlan = new TestPlan(name);
testPlan.setProperty(TestElement.TEST_CLASS, TestPlan.class.getName());
testPlan.setProperty(TestElement.GUI_CLASS, TestPlanGui.class.getName());
testPlan.setUserDefinedVariables((Arguments) new ArgumentsPanel().createTestElement());
return testPlan;
}
/**
* 创建线程组
* @param numThreads
* @param loops
* @return
*/
private static ThreadGroup createThreadGroup(int numThreads, int loops) {
ThreadGroup threadGroup = new ThreadGroup();
threadGroup.setName("Thread Group");
threadGroup.setNumThreads(numThreads);
threadGroup.setRampUp(1);
threadGroup.setSamplerController(createLoopController(loops));
threadGroup.setProperty(TestElement.TEST_CLASS, ThreadGroup.class.getName());
threadGroup.setProperty(TestElement.GUI_CLASS, ThreadGroupGui.class.getName());
return threadGroup;
}
/**
* 创建轮询控制器
* @param loops
* @return
*/
private static LoopController createLoopController(int loops) {
LoopController loopController = new LoopController();
loopController.setLoops(loops);
loopController.setProperty(TestElement.TEST_CLASS, LoopController.class.getName());
loopController.setProperty(TestElement.GUI_CLASS, LoopControlPanel.class.getName());
loopController.initialize();
return loopController;
}
/**
* 创建get请求方式
* @param urlString
* @param port
* @param requestJson
* @return
* @throws MalformedURLException
*/
private static HTTPSamplerProxy createGetHttpSampler(String urlString, int port, String requestJson) throws MalformedURLException {
URL url = new URL(urlString);
// 提取域名（不包括端口号）
String domain = url.getHost();
// 提取路径（不包括查询参数和片段标识符）
String path = url.getPath();
HTTPSamplerProxy httpSampler = new HTTPSamplerProxy();
httpSampler.setName("HTTP Request Test");
httpSampler.setProperty(TestElement.TEST_CLASS, HTTPSamplerProxy.class.getName());
httpSampler.setProperty(TestElement.GUI_CLASS, HttpTestSampleGui.class.getName());
httpSampler.setDomain(domain); // 提取域名
httpSampler.setPath(path); // 提取路径
httpSampler.setPort(port);
httpSampler.setMethod("GET");
httpSampler.setConnectTimeout("5000");
httpSampler.setUseKeepAlive(true);
// 添加请求参数
Map<String, String> parameters = new HashMap<>();
if (!StringUtils.isEmpty(requestJson)) {
Gson gson = new Gson();
// 定义 Map 的类型
Type type = new TypeToken<Map<String, String>>() {}.getType();
// 将 JSON 字符串转换为 Map
parameters = gson.fromJson(requestJson, type);
path += "?";
path += mapToQueryString(parameters);
httpSampler.setPath(path); // 提取路径
}
return httpSampler;
}
/**
* 构建get请求查询参数
* @param params
* @return
*/
public static String mapToQueryString(Map<String, String> params) {
StringBuilder queryString = new StringBuilder();
for (Map.Entry<String, String> entry : params.entrySet()) {
if (!queryString.isEmpty()) {
queryString.append("&");
}
// 对键和值进行 URL 编码
String encodedKey = URLEncoder.encode(entry.getKey(), StandardCharsets.UTF_8);
String encodedValue = URLEncoder.encode(entry.getValue(), StandardCharsets.UTF_8);
queryString.append(encodedKey).append("=").append(encodedValue);
}
return queryString.toString();
}
/**
* 创建post请求方式
* @param urlString
* @param port
* @param requestJson
* @return
* @throws MalformedURLException
*/
private static HTTPSamplerProxy createPostHttpSampler(String urlString, int port, String requestJson) throws MalformedURLException {
URL url = new URL(urlString);
// 提取域名（不包括端口号）
String domain = url.getHost();
// 提取路径（不包括查询参数和片段标识符）
String path = url.getPath();
// int portTest = url.getPort();
HTTPSamplerProxy httpSampler = new HTTPSamplerProxy();
httpSampler.setName("POST JSON Request");
httpSampler.setProperty(TestElement.TEST_CLASS, HTTPSamplerProxy.class.getName());
httpSampler.setProperty(TestElement.GUI_CLASS, HttpTestSampleGui.class.getName());
httpSampler.setDomain(domain);
httpSampler.setPath(path);
httpSampler.setPort(port);
httpSampler.setMethod("POST");
httpSampler.setConnectTimeout("5000");
httpSampler.setUseKeepAlive(true);
// 设置请求头为 JSON
httpSampler.setPostBodyRaw(true); // 使用原始请求体
httpSampler.addNonEncodedArgument("", requestJson, "");
return httpSampler;
}
/**
* 创建http请求头管理器
* @param headerMap
* @return
*/
private static HeaderManager createHeaderManager(Map<String, String> headerMap) {
HeaderManager headerManager = new HeaderManager();
headerManager.setName("HTTP Header Manager");
headerManager.setProperty(TestElement.TEST_CLASS, HeaderManager.class.getName());
headerManager.setProperty(TestElement.GUI_CLASS, HeaderPanel.class.getName());
// 添加默认 Content-Type 请求头
if (!headerMap.containsKey("Content-Type")) {
Header contentTypeHeader = new Header();
contentTypeHeader.setName("Content-Type");
contentTypeHeader.setValue("application/json; charset=UTF-8");
headerManager.add(contentTypeHeader);
}
// 遍历 headerMap，动态添加请求头
for (Map.Entry<String, String> entry : headerMap.entrySet()) {
Header header = new Header();
header.setName(entry.getKey()); // 请求头名称
header.setValue(entry.getValue()); // 请求头值
headerManager.add(header);
}
return headerManager;
}
/***
* 监听结果树
* @param replayLogPath 将结果保存到文件中，这个是文件的路径
* @return
*/
private static List<ResultCollector> getResultCollector(String replayLogPath) {
// 察看结果数
List<ResultCollector> resultCollectors = new ArrayList<>();
Summariser summariser = new Summariser("速度");
ResultCollector resultCollector = new ResultCollector(summariser);
resultCollector.setProperty(new BooleanProperty("ResultCollector.error_logging", false));
resultCollector.setProperty(new ObjectProperty("saveConfig", getSampleSaveConfig()));
resultCollector.setProperty(new StringProperty("TestElement.gui_class", "org.apache.jmeter.visualizers.ViewResultsFullVisualizer"));
resultCollector.setProperty(new StringProperty("TestElement.name", "察看结果树"));
resultCollector.setProperty(new StringProperty("TestElement.enabled", "true"));
resultCollector.setProperty(new StringProperty("filename", replayLogPath));
resultCollectors.add(resultCollector);
// 结果汇总
// ResultCollector resultTotalCollector = new ResultCollector();
// resultTotalCollector.setProperty(new BooleanProperty("ResultCollector.error_logging", false));
// resultTotalCollector.setProperty(new ObjectProperty("saveConfig", getSampleSaveConfig()));
// resultTotalCollector.setProperty(new StringProperty("TestElement.gui_class", "org.apache.jmeter.visualizers.SummaryReport"));
// resultTotalCollector.setProperty(new StringProperty("TestElement.name", "汇总报告"));
// resultTotalCollector.setProperty(new StringProperty("TestElement.enabled", "true"));
// resultTotalCollector.setProperty(new StringProperty("filename", ""));
// resultCollectors.add(resultTotalCollector);
return resultCollectors;
}
/**
* 配置采样结果信息
* @return
*/
private static SampleSaveConfiguration getSampleSaveConfig() {
SampleSaveConfiguration sampleSaveConfiguration = new SampleSaveConfiguration();
sampleSaveConfiguration.setTime(true);
sampleSaveConfiguration.setLatency(true);
sampleSaveConfiguration.setTimestamp(true);
sampleSaveConfiguration.setSuccess(true);
sampleSaveConfiguration.setLabel(true);
sampleSaveConfiguration.setCode(true);
sampleSaveConfiguration.setMessage(true);
sampleSaveConfiguration.setThreadName(true);
sampleSaveConfiguration.setDataType(true);
sampleSaveConfiguration.setEncoding(true);
sampleSaveConfiguration.setAssertions(true);
sampleSaveConfiguration.setSubresults(true);
sampleSaveConfiguration.setResponseData(true);
sampleSaveConfiguration.setSamplerData(true);
sampleSaveConfiguration.setAsXml(true);
sampleSaveConfiguration.setFieldNames(true);
sampleSaveConfiguration.setResponseHeaders(true);
sampleSaveConfiguration.setRequestHeaders(false);
sampleSaveConfiguration.setAssertionResultsFailureMessage(true);
//sampleSaveConfiguration.setsserAtionsResultsToSave(0); assertionsResultsToSave
sampleSaveConfiguration.setBytes(true);
sampleSaveConfiguration.setSentBytes(true);
sampleSaveConfiguration.setUrl(true);
sampleSaveConfiguration.setThreadCounts(true);
sampleSaveConfiguration.setIdleTime(true);
sampleSaveConfiguration.setConnectTime(true);
return sampleSaveConfiguration;
}
/**
* 从结果文件读取响应内容
* @param filePath
* @return
*/
private static String getResultMessageFromFile(String filePath) {
File file = new File(filePath);
if (!file.exists()) {
return null;
}
try {
// 1. 创建DocumentBuilderFactory对象
DocumentBuilderFactory factory = DocumentBuilderFactory.newInstance();
// 2. 创建DocumentBuilder对象
DocumentBuilder builder = factory.newDocumentBuilder();
// 3. 解析XML文件，获取Document对象
Document document = builder.parse(new File(filePath));
// 4. 获取所有<httpSample>节点
NodeList httpSampleList = document.getElementsByTagName("httpSample");
Node httpSampleNode = httpSampleList.item(0);
if (httpSampleNode.getNodeType() == Node.ELEMENT_NODE) {
Element httpSampleElement = (Element) httpSampleNode;
// 提取子节点<responseData>的内容
NodeList responseDataList = httpSampleElement.getElementsByTagName("responseData");
if (responseDataList.getLength() > 0) {
Element responseDataElement = (Element) responseDataList.item(0);
return responseDataElement.getTextContent();
}
}
} catch (Exception e) {
e.printStackTrace();
}
return null;
}
public static void main(String[] args) {
String paramJson = "{\"md5Value\":\"value1\",\"fileName\":\"value2\"}";
paramJson = "{\"name\":\"John\",\"city\":\"New York\",\"age\":30}";
String requestHeader = "{\"Content-Type\":\"application/json; charset=UTF-8\",\"Authorization\":\"Bearer token123\"}";
String jmeterHomePath = "D:/apache-jmeter-5.4.3";
// String result = getJmeterResult("http://127.0.0.1:3000/api/items/add", 3000, "POST", paramJson, requestHeader, jmeterHomePath);
String result = getJmeterResult(123L, "http://127.0.0.1:3000/api/items/user", 3000, "GET", paramJson, requestHeader, jmeterHomePath);
System.out.println(result);
}
}