Jmeter 二次封装

CallmeJack

     之前在老东家做性能测试的时候，一开始是使用LR的（服务端测试）。在LR里写vuser scripts 去进行
RPC协议的接口性能测试。但是后来觉得LR实在太重了，一般的机器消受不了。而且无法自动化的驱动测
试。所以后来引入了Jmeter。但是Jmeter也有它的缺点， 尤其是最后结果统计中那蛋疼的报表简直low到爆。
所以临走前搞了一次二次封装。增强jmeter的特性，可惜只做了比较少的一部分我就离职了。 这里我写出
来就当抛砖引玉了吧。

原理

其实原理十分简单，我们一般就是用Jmeter的GUI来完成工作么。现在我们不用GUI了，我们直接引用
jmeter的核心lib，直接调用jmeter的API来执行我们的操作。你需要引入maven的依赖。如下

1. <dependency>
   
2.         <groupId>com.lazerycode.jmeter</groupId>
   
3.         <artifactId>jmeter-maven-plugin</artifactId>
   
4.         <version>1.10.1</version>
   
5.         <exclusions>
   
6.             <exclusion>
   
7.                 <artifactId>ApacheJMeter_config</artifactId>
   
8.                 <groupId>org.apache.jmeter</groupId>
   
9.             </exclusion>
   
10.         </exclusions>
    
11.     </dependency>

复制代码

然后运行的时候，你还是需要装一个Jmeter的，这个我没绕过去。需要在代码中设置jmter的home。
以及一些其他的配置信息

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

复制代码

代码部分

现在我们来看看核心代码部分吧

1. Logger.info("测试方法: "+caseInfo.getName());
   
2.         Logger.info("起始并发数: "+caseInfo.getBeginThread());
   
3.         Logger.info("递增并发数: "+caseInfo.getAddThread());
   
4.         Logger.info("结束并发数: "+caseInfo.getEndThread());
   
5.         Logger.info("每个线程的循环次数: "+caseInfo.getLoop());
   
6.         //Logger.html_link("1111", "222");
   
7. 
   
8.         JMeterUtils.setJMeterHome(Constant.getJmeterHome());
   
9.         JMeterUtils.loadJMeterProperties(Constant.getJmeterHome()
   
10.                 + "/bin/jmeter.properties");
    
11.         JMeterUtils.setProperty("jmeter.save.saveservice.output_format", "xml");
    
12.         File log = new File(Constant.logFilePath() + "perfTest.log");
    
13. 
    
14.         JMeterUtils.setProperty(LoggingManager.LOG_FILE, log.getAbsolutePath());
    
15.         JMeterUtils.initLogging();// you can comment this line out to see extra
    
16.                                     // log messages of i.e. DEBUG level
    
17.         JMeterUtils.initLocale();
    
18. 
    
19.         // Initialize JMeter SaveService
    
20.         SaveService.loadProperties();
    
21. 
    
22.         JavaSampler javaSample = new JavaSampler();
    
23.         javaSample.setClassname(caseInfo.getName());
    
24. 
    
25.         // Loop Controller
    
26.         LoopController loopController = new LoopController();
    
27.         loopController.addTestElement(javaSample);
    
28.         loopController.setLoops(caseInfo.getLoop());
    
29.         loopController.setFirst(true);
    
30.         loopController.initialize();
    
31. 
    
32.         // Thread Group
    
33.         List<ThreadGroup> threadGrouplist = this.setupThreadGroup(caseInfo, loopController);
    
34. 
    
35.         // JMeter Test Plan, basic all u JOrphan HashTree
    
36.         TestPlan testPlan = new TestPlan("My Test Plan");
    
37.         testPlan.setSerialized(true);
    
38. 
    
39.         // JMeter Test Plan, basic all u JOrphan HashTree
    
40.         HashTree testPlanTree = new HashTree();
    
41. 
    
42.         // Construct Test Plan from previously initialized elements
    
43.         testPlanTree.add("TestPlan", testPlan);
    
44.         testPlanTree.add("LoopController", loopController);
    
45.         testPlanTree.add("JavaSampler", javaSample);
    
46. 
    
47.         int count = 1000;
    
48.         for(ThreadGroup threadGroup:threadGrouplist){
    
49.             testPlanTree.add("ThreadGroup"+count,threadGroup);
    
50.             count--;
    
51.         }
    
52. 
    
53. 
    
54.         // Store execution results
    
55.         MyResultCollector requestCollector = new MyResultCollector();
    
56.         requestCollector.setFilename(Constant.logFilePath()+"result.log");
    
57.         testPlanTree.add(testPlanTree.getArray()[0], requestCollector);
    
58.         // Run Test Plan
    
59.         StandardJMeterEngine jmeterEngine = new StandardJMeterEngine();
    
60.         jmeterEngine.configure(testPlanTree);
    
61.         jmeterEngine.run();
    
62. 
    
63.         Map<Integer,List<SampleResult>> sampleResultsMap = requestCollector.getSampleResults();
    
64.         for (Object key : sampleResultsMap.keySet()) {
    
65.             List<SampleResult> list = sampleResultsMap.get(key);
    
66.             /*for(SampleResult s : list){
    
67.                 System.out.println(s.getTime()+"fffff");
    
68.             }*/
    
69.             System.out.println(list.size()+"kkkkkkkkkkk");
    
70. 
    
71.         }
    
72.         PerfChartHelper.createChart_Jmeter(1, sampleResultsMap,caseInfo.getName());

复制代码

可以看到，我们在代码中设置循环，设置线程组，测试计划等等，熟悉jmeter的同学一定熟悉这些
概念。其中我们测试的java的接口么。所以用的是javasimpler

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

复制代码

其中需要给sample传一个测试类（如果是http接口，其实就不需要了）。下面我们看看这个测试类是
怎么定义的。

1. public class TestP extends AbstractJavaSamplerClient {
   
2.     private static AtomicInteger temp = new AtomicInteger();
   
3. 
   
4.     @Override
   
5.     public Arguments getDefaultParameters() {
   
6.         System.out.println("===========init parameters ========");
   
7.         Arguments arg = new Arguments();
   
8.         return arg;
   
9.     }
   
10.     public SampleResult runTest(JavaSamplerContext paramJavaSamplerContext) {
    
11.         SampleResult sr = new SampleResult();
    
12.         sr.sampleStart();
    
13.         //System.out.println(123);
    
14. 
    
15.             //此处是调用逻辑
    
16. 
    
17.         sr.setSuccessful(true);
    
18.         sr.sampleEnd();
    
19.         return sr;
    
20.     }
    
21. 
    
22. }

复制代码

可以看到我们只要继承jmeter定义的类AbstractJavaSamplerClient 就可以了。 jmeter也是有集合点这个概
念的。即便没有，现在你都能调用它的API了，你自己写一个实现嵌入进去也行。然后我们看结果收集，
这里我是自己进行了计算。

1. public class MyResultCollector extends ResultCollector {
   
2. 
   
3.     private static final long serialVersionUID = -8648350950445938218L;
   
4. 
   
5.     //private List<SampleResult> sampleResults;
   
6.     private Map<Integer,List<SampleResult>> threadResultsMap;
   
7. 
   
8.     public MyResultCollector() {
   
9.         threadResultsMap = new HashMap<Integer,List<SampleResult>>();
   
10.     }
    
11. 
    
12.     @Override
    
13.     public void sampleOccurred(SampleEvent e) {
    
14.         super.sampleOccurred(e);
    
15.         SampleResult r = e.getResult();
    
16.         int threadGroupName = Integer.parseInt(e.getThreadGroup());
    
17.         if(threadResultsMap.containsKey(threadGroupName)){
    
18.             List<SampleResult> sampleResults = threadResultsMap.get(threadGroupName);
    
19.             sampleResults.add(r);
    
20.         }else{
    
21.             List<SampleResult> sampleResults = new ArrayList<SampleResult>();
    
22.             sampleResults.add(r);
    
23.             threadResultsMap.put(threadGroupName, sampleResults);
    
24.         }
    
25.     }
    
26. 
    
27.     public Map<Integer,List<SampleResult>> getSampleResults() {
    
28.         return threadResultsMap;
    
29.     }
    
30. }

复制代码

首先是扩展Jmeter自己的结果收集器。

CallmeJack

1. public static void createChart_Jmeter(int type, Map<Integer, List<SampleResult>> sampleResultsMap,String className) throws IOException {
   
2.         ArrayList<double[]> tpsSeries = new ArrayList<double[]>();
   
3.         ArrayList<double[]> averageSeries = new ArrayList<double[]>();
   
4.         ArrayList<double[]> minSeries = new ArrayList<double[]>();
   
5.         ArrayList<double[]> maxSeries = new ArrayList<double[]>();
   
6.         ArrayList<double[]> time60 = new ArrayList<double[]>();
   
7.         ArrayList<double[]> time90 = new ArrayList<double[]>();
   
8.         ArrayList<double[]> time95 = new ArrayList<double[]>();
   
9. 
   
10.         //用treeSet排序
    
11.         TreeSet<Integer> tree = new TreeSet<Integer>();
    
12.         for(int thread : sampleResultsMap.keySet()) {
    
13.             tree.add(thread);
    
14.         }
    
15. 
    
16.         // 遍历每一个threadgroup的结果集，算出tps，平均相应时间并画图
    
17.         for(int thread : tree) {
    
18.             AggregatedParser aggregated = new AggregatedParser(sampleResultsMap.get(thread));
    
19. 
    
20.             double tps = aggregated.getTps();
    
21.             double averageTime = aggregated.getAverageTime();
    
22.             long maxTime = aggregated.getMaxTime();
    
23.             long minTime = aggregated.getMinTime();
    
24.             long per60Time = aggregated.getPercentTime(0.6);
    
25.             long per90Time = aggregated.getPercentTime(0.9);
    
26.             long per95Time = aggregated.getPercentTime(0.95);
    
27.             double errorRate = aggregated.getErrorRate();
    
28. 
    
29.             tpsSeries.add(new double[]{thread, tps});
    
30.             averageSeries.add(new double[]{thread, averageTime});
    
31.             maxSeries.add(new double[]{thread, maxTime});
    
32.             minSeries.add(new double[]{thread, minTime});
    
33.             time60.add(new double[]{thread, per60Time});
    
34.             time90.add(new double[]{thread, per90Time});
    
35.             time95.add(new double[]{thread, per95Time});
    
36.             Logger.info("在"+thread+"并发下错误率:"+errorRate);
    
37.         }
    
38. 
    
39.         Map<String, ArrayList<double[]>> timeMap = new HashMap<String, ArrayList<double[]>>();
    
40.         timeMap.put("average time", averageSeries);
    
41.         timeMap.put("min time", minSeries);
    
42.         timeMap.put("max time", maxSeries);
    
43.         timeMap.put("60% time", time60);
    
44.         timeMap.put("90% time", time90);
    
45.         timeMap.put("95% time", time95);
    
46. 
    
47. 
    
48.         JfreeChart tpsChart = new JfreeChart("thread_tps", "thread", "tps");
    
49.         tpsChart.addXYSeries("thread_tps_report", tpsSeries);
    
50.         File folder = new File(Constant.imagePath()+className);
    
51.         folder.mkdirs();
    
52.         File imgTps = new File(Constant.imagePath()+className, "_tps.png");
    
53.         tpsChart.createJfreeChartImage(imgTps, 800, 600);
    
54. 
    
55.         //在html report中增加图片链接
    
56.         //Logger.html_img(Constant.imagePath()+className+"/_tps.png");
    
57.         //echarts_Qps(tpsSeries);
    
58. 
    
59. 
    
60.         JfreeChart avgTimeChart = new JfreeChart("avgTime", "thread", "avg_time");
    
61.         avgTimeChart.addXYSeries("thread_avgTime", averageSeries);
    
62.         File imgAvgTime = new File(Constant.imagePath()+className, "_avg.png");
    
63.         avgTimeChart.createJfreeChartImage(imgAvgTime, 800, 600);
    
64. 
    
65.         //在html report中增加图片链接
    
66.         //Logger.html_img(Constant.imagePath()+className+"/_avg.png");
    
67. 
    
68.         //使用echart在html页面中花出tps和平均响应时间的图
    
69.         Integer id = tempCount.getAndAdd(1);
    
70.         Logger.html("<div id=\"chart_qps"+id+"\" style=\"width:800px;height:600px;\"></div>");
    
71.         Logger.html("<div id=\"chart_time"+id+"\" style=\"width:800px;height:600px;\"></div>");
    
72.         Logger.html("<script>");
    
73.         echarts_Qps(tpsSeries,id);
    
74.         echarts_Time(timeMap,id);
    
75.         Logger.html("</script>");

复制代码

然后增加结果收集算法。并画出图来。我分别用jfreechart和echart画图。

然后我们一下case控制

1. <suite name="rock" >
   
2.     <paras>
   
3.         <para host_ip="10.9.20.171"></para>
   
4.     </paras>
   
5.     <cases>
   
6.         <case name="performance.TestP" run="false">
   
7.             <perf begin_thread="100" end_thread="200" add_thread="20" loop="10"/>
   
8.             <para desc="根据订单ID查询订单"></para>
   
9.         </case>
   
10. 
    
11.         <case name="performance.TestP" run="false">
    
12.             <perf begin_thread="100" end_thread="200" add_thread="20" loop="10"/>
    
13.             <para desc="根据订单ID查询订单"></para>
    
14.         </case>
    
15. 
    
16.     </cases>
    
17. </suite>

复制代码

case里有我们想要控制的信息，例如起始的并发数，每次递增的并发数。结束的并发数等等。
其中name就是我们的测试类的全路径（我这里都是我自己写的demo类）。

效果图

现在我们来看看效果图吧，就是我们最后的结果


以及

梦想家

支持分享

Miss_love

感谢分享