Jmeter 二次封装 - 51Testing软件测试论坛

<dependency>
<groupId>com.lazerycode.jmeter</groupId>
<artifactId>jmeter-maven-plugin</artifactId>
<version>1.10.1</version>
<exclusions>
<exclusion>
<artifactId>ApacheJMeter_config</artifactId>
<groupId>org.apache.jmeter</groupId>
</exclusion>
</exclusions>
</dependency>

复制代码

JMeterUtils.setJMeterHome(Constant.getJmeterHome());
JMeterUtils.loadJMeterProperties(Constant.getJmeterHome()
+ "/bin/jmeter.properties");
JMeterUtils.setProperty("jmeter.save.saveservice.output_format", "xml");
File log = new File(Constant.logFilePath() + "perfTest.log");
private static String jmeterHome = "E:\\apache-jmeter-2.13";//jmeter的home目录

复制代码

Logger.info("测试方法: "+caseInfo.getName());
Logger.info("起始并发数: "+caseInfo.getBeginThread());
Logger.info("递增并发数: "+caseInfo.getAddThread());
Logger.info("结束并发数: "+caseInfo.getEndThread());
Logger.info("每个线程的循环次数: "+caseInfo.getLoop());
//Logger.html_link("1111", "222");
JMeterUtils.setJMeterHome(Constant.getJmeterHome());
JMeterUtils.loadJMeterProperties(Constant.getJmeterHome()
+ "/bin/jmeter.properties");
JMeterUtils.setProperty("jmeter.save.saveservice.output_format", "xml");
File log = new File(Constant.logFilePath() + "perfTest.log");
JMeterUtils.setProperty(LoggingManager.LOG_FILE, log.getAbsolutePath());
JMeterUtils.initLogging();// you can comment this line out to see extra
// log messages of i.e. DEBUG level
JMeterUtils.initLocale();
// Initialize JMeter SaveService
SaveService.loadProperties();
JavaSampler javaSample = new JavaSampler();
javaSample.setClassname(caseInfo.getName());
// Loop Controller
LoopController loopController = new LoopController();
loopController.addTestElement(javaSample);
loopController.setLoops(caseInfo.getLoop());
loopController.setFirst(true);
loopController.initialize();
// Thread Group
List<ThreadGroup> threadGrouplist = this.setupThreadGroup(caseInfo, loopController);
// JMeter Test Plan, basic all u JOrphan HashTree
TestPlan testPlan = new TestPlan("My Test Plan");
testPlan.setSerialized(true);
// JMeter Test Plan, basic all u JOrphan HashTree
HashTree testPlanTree = new HashTree();
// Construct Test Plan from previously initialized elements
testPlanTree.add("TestPlan", testPlan);
testPlanTree.add("LoopController", loopController);
testPlanTree.add("JavaSampler", javaSample);
int count = 1000;
for(ThreadGroup threadGroup:threadGrouplist){
testPlanTree.add("ThreadGroup"+count,threadGroup);
count--;
}
// Store execution results
MyResultCollector requestCollector = new MyResultCollector();
requestCollector.setFilename(Constant.logFilePath()+"result.log");
testPlanTree.add(testPlanTree.getArray()[0], requestCollector);
// Run Test Plan
StandardJMeterEngine jmeterEngine = new StandardJMeterEngine();
jmeterEngine.configure(testPlanTree);
jmeterEngine.run();
Map<Integer,List<SampleResult>> sampleResultsMap = requestCollector.getSampleResults();
for (Object key : sampleResultsMap.keySet()) {
List<SampleResult> list = sampleResultsMap.get(key);
/*for(SampleResult s : list){
System.out.println(s.getTime()+"fffff");
}*/
System.out.println(list.size()+"kkkkkkkkkkk");
}
PerfChartHelper.createChart_Jmeter(1, sampleResultsMap,caseInfo.getName());

复制代码

JavaSampler javaSample = new JavaSampler();
javaSample.setClassname(caseInfo.getName());

复制代码

public class TestP extends AbstractJavaSamplerClient {
private static AtomicInteger temp = new AtomicInteger();
@Override
public Arguments getDefaultParameters() {
System.out.println("===========init parameters ========");
Arguments arg = new Arguments();
return arg;
}
public SampleResult runTest(JavaSamplerContext paramJavaSamplerContext) {
SampleResult sr = new SampleResult();
sr.sampleStart();
//System.out.println(123);
//此处是调用逻辑
sr.setSuccessful(true);
sr.sampleEnd();
return sr;
}
}

复制代码

public class MyResultCollector extends ResultCollector {
private static final long serialVersionUID = -8648350950445938218L;
//private List<SampleResult> sampleResults;
private Map<Integer,List<SampleResult>> threadResultsMap;
public MyResultCollector() {
threadResultsMap = new HashMap<Integer,List<SampleResult>>();
}
@Override
public void sampleOccurred(SampleEvent e) {
super.sampleOccurred(e);
SampleResult r = e.getResult();
int threadGroupName = Integer.parseInt(e.getThreadGroup());
if(threadResultsMap.containsKey(threadGroupName)){
List<SampleResult> sampleResults = threadResultsMap.get(threadGroupName);
sampleResults.add(r);
}else{
List<SampleResult> sampleResults = new ArrayList<SampleResult>();
sampleResults.add(r);
threadResultsMap.put(threadGroupName, sampleResults);
}
}
public Map<Integer,List<SampleResult>> getSampleResults() {
return threadResultsMap;
}
}

复制代码

public static void createChart_Jmeter(int type, Map<Integer, List<SampleResult>> sampleResultsMap,String className) throws IOException {
ArrayList<double[]> tpsSeries = new ArrayList<double[]>();
ArrayList<double[]> averageSeries = new ArrayList<double[]>();
ArrayList<double[]> minSeries = new ArrayList<double[]>();
ArrayList<double[]> maxSeries = new ArrayList<double[]>();
ArrayList<double[]> time60 = new ArrayList<double[]>();
ArrayList<double[]> time90 = new ArrayList<double[]>();
ArrayList<double[]> time95 = new ArrayList<double[]>();
//用treeSet排序
TreeSet<Integer> tree = new TreeSet<Integer>();
for(int thread : sampleResultsMap.keySet()) {
tree.add(thread);
}
// 遍历每一个threadgroup的结果集，算出tps，平均相应时间并画图
for(int thread : tree) {
AggregatedParser aggregated = new AggregatedParser(sampleResultsMap.get(thread));
double tps = aggregated.getTps();
double averageTime = aggregated.getAverageTime();
long maxTime = aggregated.getMaxTime();
long minTime = aggregated.getMinTime();
long per60Time = aggregated.getPercentTime(0.6);
long per90Time = aggregated.getPercentTime(0.9);
long per95Time = aggregated.getPercentTime(0.95);
double errorRate = aggregated.getErrorRate();
tpsSeries.add(new double[]{thread, tps});
averageSeries.add(new double[]{thread, averageTime});
maxSeries.add(new double[]{thread, maxTime});
minSeries.add(new double[]{thread, minTime});
time60.add(new double[]{thread, per60Time});
time90.add(new double[]{thread, per90Time});
time95.add(new double[]{thread, per95Time});
Logger.info("在"+thread+"并发下错误率:"+errorRate);
}
Map<String, ArrayList<double[]>> timeMap = new HashMap<String, ArrayList<double[]>>();
timeMap.put("average time", averageSeries);
timeMap.put("min time", minSeries);
timeMap.put("max time", maxSeries);
timeMap.put("60% time", time60);
timeMap.put("90% time", time90);
timeMap.put("95% time", time95);
JfreeChart tpsChart = new JfreeChart("thread_tps", "thread", "tps");
tpsChart.addXYSeries("thread_tps_report", tpsSeries);
File folder = new File(Constant.imagePath()+className);
folder.mkdirs();
File imgTps = new File(Constant.imagePath()+className, "_tps.png");
tpsChart.createJfreeChartImage(imgTps, 800, 600);
//在html report中增加图片链接
//Logger.html_img(Constant.imagePath()+className+"/_tps.png");
//echarts_Qps(tpsSeries);
JfreeChart avgTimeChart = new JfreeChart("avgTime", "thread", "avg_time");
avgTimeChart.addXYSeries("thread_avgTime", averageSeries);
File imgAvgTime = new File(Constant.imagePath()+className, "_avg.png");
avgTimeChart.createJfreeChartImage(imgAvgTime, 800, 600);
//在html report中增加图片链接
//Logger.html_img(Constant.imagePath()+className+"/_avg.png");
//使用echart在html页面中花出tps和平均响应时间的图
Integer id = tempCount.getAndAdd(1);
Logger.html("<div id=\"chart_qps"+id+"\" style=\"width:800px;height:600px;\"></div>");
Logger.html("<div id=\"chart_time"+id+"\" style=\"width:800px;height:600px;\"></div>");
Logger.html("<script>");
echarts_Qps(tpsSeries,id);
echarts_Time(timeMap,id);
Logger.html("</script>");

复制代码

<suite name="rock" >
<paras>
<para host_ip="10.9.20.171"></para>
</paras>
<cases>
<case name="performance.TestP" run="false">
<perf begin_thread="100" end_thread="200" add_thread="20" loop="10"/>
<para desc="根据订单ID查询订单"></para>
</case>
<case name="performance.TestP" run="false">
<perf begin_thread="100" end_thread="200" add_thread="20" loop="10"/>
<para desc="根据订单ID查询订单"></para>
</case>
</cases>
</suite>

复制代码