Jmeter 插件开发

CallmeJack

一. 配置源码

下载源码包 地址
导入源码包，我这里使用的工具IntelliJ。直接File->open就行。导入后的项目目录如下图：

下载jar包。jar包下载下来的地址为：/lib

添加jar包。File->Project Structure->Libraries
install

运行 src->core->apache.jmeter->NewDriver

1. public static void main(String[] args) {
   
2. 
   
3.     Thread.currentThread().setContextClassLoader(loader);
   
4.     if (System.getProperty("log4j.configuration") == null) {// $NON-NLS-1$ $NON-NLS-2$
   
5.         File conf = new File(jmDir, "bin" + File.separator + "log4j.conf");// $NON-NLS-1$ $NON-NLS-2$
   
6.         System.setProperty("log4j.configuration", "file:" + conf);
   
7.     }
   
8. 
   
9.     try {
   
10.         Class<?> initialClass = loader.loadClass("org.apache.jmeter.JMeter");// $NON-NLS-1$
    
11.         Object instance = initialClass.newInstance();
    
12.         Method startup = initialClass.getMethod("start", new Class[] { new String[0].getClass() });// $NON-NLS-1$
    
13.         startup.invoke(instance, new Object[] { args });
    
14.     } catch(Throwable e){
    
15.         e.printStackTrace();
    
16.         System.err.println("JMeter home directory was detected as: "+jmDir);
    
17.     }
    
18. }

复制代码

从上述代码中我们可以看出，其实之际上调用的是src->core->apache.jmeter->Jmeter 的start() 方法。
而当我们打开Jmeter的Gui时，实际上调用的用的是src->core->apache.jmeter->Jmeter 的startGui() 方法

1. private void startGui(String testFile) {
   
2.         String jMeterLaf = LookAndFeelCommand.getJMeterLaf();
   
3.         try {
   
4.             UIManager.setLookAndFeel(jMeterLaf);
   
5.         } catch (Exception ex) {
   
6.             log.warn("Could not set LAF to:"+jMeterLaf, ex);
   
7.         }
   
8. 
   
9.         PluginManager.install(this, true);
   
10. 
    
11.         JMeterTreeModel treeModel = new JMeterTreeModel();
    
12.         JMeterTreeListener treeLis = new JMeterTreeListener(treeModel);
    
13.         final ActionRouter instance = ActionRouter.getInstance();
    
14.         instance.populateCommandMap();  //这个方法会去寻找<your project>/lib/ext 下所有的jar
    
15.         treeLis.setActionHandler(instance);
    
16.         // NOTUSED: GuiPackage guiPack =
    
17.         GuiPackage.getInstance(treeLis, treeModel);
    
18.         MainFrame main = new MainFrame(treeModel, treeLis);
    
19.         ComponentUtil.centerComponentInWindow(main, 80);
    
20.         main.setVisible(true);
    
21.         instance.actionPerformed(new ActionEvent(main, 1, ActionNames.ADD_ALL));
    
22.         if (testFile != null) {
    
23.             try {
    
24.                 File f = new File(testFile);
    
25.                 log.info("Loading file: " + f);
    
26.                 FileServer.getFileServer().setBaseForScript(f);
    
27. 
    
28.                 HashTree tree = SaveService.loadTree(f);
    
29. 
    
30.                 GuiPackage.getInstance().setTestPlanFile(f.getAbsolutePath());
    
31. 
    
32.                 Load.insertLoadedTree(1, tree);
    
33.             } catch (ConversionException e) {
    
34.                 log.error("Failure loading test file", e);
    
35.                 JMeterUtils.reportErrorToUser(SaveService.CEtoString(e));
    
36.             } catch (Exception e) {
    
37.                 log.error("Failure loading test file", e);
    
38.                 JMeterUtils.reportErrorToUser(e.toString());
    
39.             }
    
40.         } else {
    
41.             JTree jTree = GuiPackage.getInstance().getMainFrame().getTree();
    
42.             TreePath path = jTree.getPathForRow(0);
    
43.             jTree.setSelectionPath(path);
    
44.             FocusRequester.requestFocus(jTree);
    
45.         }
    
46.     }

复制代码

二.开发sampler

JMeter加载插件的机制比较简单，扫描扩展下的的所有实现了JMeterGUIComponent和TestBean接口的
类，然后进行初始化。

1. ClassFinder.findClassesThatExtend(
   
2.     JMeterUtils.getSearchPaths(),
   
3.     new Class[] {JMeterGUIComponent.class, TestBean.class }
   

复制代码

在getSearchPaths()在这个方法中，它寻找的路径是/lib/ext,所以要确保你开发的插件的jar存在与这
个路径中。

1. public static String[] getSearchPaths() {
   
2.         String p = JMeterUtils.getPropDefault("search_paths", null); // $NON-NLS-1$
   
3.         String[] result = new String[1];
   
4. 
   
5.         if (p != null) {
   
6.             String[] paths = p.split(";"); // $NON-NLS-1$
   
7.             result = new String[paths.length + 1];
   
8.             System.arraycopy(paths, 0, result, 1, paths.length);
   
9.         }
   
10.         result[0] = getJMeterHome() + "/lib/ext"; // $NON-NLS-1$
    
11.         return result;
    
12.     }

复制代码

1. jmeter提供了example。src/examples，所以在写之前 可以多参考一下
   
2. 
   
3. 在src/components/sampler/gui 下新建SmaplerGui，继承AbstractSamplerGui 类。 ```java package org.apache.jmeter.sampler.gui;
   
4. import com.btcc.fix.TestUser;
   
5. import net.sourceforge.jdatepicker.JDateComponentFactory;
   
6. import net.sourceforge.jdatepicker.JDatePanel;
   
7. import net.sourceforge.jdatepicker.impl.UtilDateModel;
   
8. import org.apache.jmeter.gui.util.HorizontalPanel;
   
9. import org.apache.jmeter.gui.util.VerticalPanel;
   
10. import org.apache.jmeter.sampler.FixSmapler;
    
11. import org.apache.jmeter.sampler.MessageProvider;
    
12. import org.apache.jmeter.samplers.gui.AbstractSamplerGui;
    
13. import org.apache.jmeter.testelement.TestElement;
    
14. import org.apache.jorphan.logging.LoggingManager;
    
15. import org.apache.log.Logger;
    
16. import quickfix.btcc.Message;
    
17. 
    
18. import javax.swing.;
    
19. import java.awt.;
    
20. import java.awt.event.*;
    
21. import java.text.SimpleDateFormat;
    
22. import java.util.Date;
    
23. 
    
24. import static javax.swing.BorderFactory.createTitledBorder;
    
25. 
    
26. /**
    
27. 
    
28. Created by sara on 12/10/2016.
    
29. */
    
30. public class FixSmaplerGui extends AbstractSamplerGui {
    
31. 
    
32. private static final Logger log = LoggingManager.getLoggerForClass();
    
33. 
    
34. // 布局定义
    
35. private JTextField requestNameText = null;
    
36. private JTextField hostText = null;
    
37. private JTextField portText = null;
    
38. private JCheckBox sslCheckBox = null;
    
39. 
    
40. public FixSmaplerGui(){
    
41. super();
    
42. init();
    
43. }
    
44. 
    
45. //用这个方法将fixSmapler的数据设置到你的gui中
    
46. @Override
    
47. public void configure(TestElement element) {
    
48. 
    
49. super.configure(element);
    
50. FixSmapler fixSmapler = (FixSmapler)element;
    
51. hostText.setText(fixSmapler.getHost());
    
52. portText.setText(fixSmapler.getPort());
    
53. sslCheckBox.setSelected(fixSmapler.getSsl());
    
54. 
    
55. }
    
56. 
    
57. private void init() {
    
58. setLayout(new BorderLayout(0, 5));
    
59. setBorder(makeBorder());
    
60. add(createDataPanel(),BorderLayout.CENTER);
    
61. }
    
62. 
    
63. // 这个方法是用来创建你自己的元素布局的
    
64. private JPanel createDataPanel(){
    
65. final JPanel settingPanel = new VerticalPanel(5, 0);
    
66. settingPanel.setBorder(makeBorder());
    
67. settingPanel.add(makeTitlePanel());
    
68. 
    
69. // host
    
70. JPanel hostPanel = new HorizontalPanel();
    
71. hostPanel.setBorder(createTitledBorder("Host/port"));
    
72. 
    
73. JLabel hostLable = new JLabel("Host:");
    
74. this.hostText = new JTextField(15);
    
75. hostLable.setLabelFor(hostText);
    
76. hostPanel.add(hostLable);
    
77. hostPanel.add(hostText);
    
78. 
    
79. JLabel portLable = new JLabel("Port:");
    
80. portText = new JTextField(15);
    
81. portLable.setLabelFor(portText);
    
82. hostPanel.add(portLable);
    
83. hostPanel.add(portText);
    
84. 
    
85. JLabel sslLabel = new JLabel("SSL");
    
86. sslCheckBox = new JCheckBox();
    
87. sslLabel.setLabelFor(sslCheckBox);
    
88. hostPanel.add(sslLabel);
    
89. hostPanel.add(sslCheckBox);
    
90. settingPanel.add(hostPanel);
    
91. return settingPanel;
    
92. 
    
93. }
    
94. 
    
95. // 创建新的sampler。并且将它传给你创建的modifyTestElement(TestElement)方法。
    
96. @Override
    
97. public TestElement createTestElement() {
    
98. 
    
99. TestElement sampler = new FixSmapler();
    
100. modifyTestElement(sampler);
     
101. return sampler;
     
102. }
     
103. 
     
104. // 这个方法应该返回代表的component的title/name的名字。fix_sampler_title必须被写进jmeter 的messages.properties文件中。
     
105. @Override
     
106. public String getLabelResource() {
     
107. // TODO Auto-generated method stub
     
108. return "fix_sampler_title"; // $NON-NLS-1$
     
109. }
     
110. 
     
111. // 这个方法是用来将数据从你的gui传到TestElement.这个方法和configure()方法在逻辑上是相反的
     
112. @Override
     
113. public void modifyTestElement(TestElement element) {
     
114. super.configureTestElement(element);
     
115. FixSmapler smapler = (FixSmapler) element;
     
116. smapler.setHost(hostText.getText());
     
117. smapler.setPort(portText.getText());
     
118. smapler.setSsl(sslCheckBox.isSelected());
     
119. 
     
120. }
     
121. 
     
122. // 这个方法是用来在你创建新的sampler时，清除数据的。
     
123. @Override
     
124. public void clearGui() {
     
125. super.clearGui();
     
126. hostText.setText("");
     
127. portText.setText("");
     
128. sslCheckBox.setSelected(false);
     
129. 
     
130. }
     
131. }

复制代码

CallmeJack

1. 当是在gui上创建一个sampler，其调用的顺序是**clearGui()->createTestElement()->modifyTestElement()->configure()**
   
2. 
   
3. 2. 在src/components/sampler 下新建Smapler，继承AbstractSampler 类。
   
4. ```java
   
5. import com.btcc.fix.FixClient;
   
6. import com.btcc.fix.TestClient;
   
7. import org.apache.jmeter.samplers.AbstractSampler;
   
8. import org.apache.jmeter.samplers.Entry;
   
9. import org.apache.jmeter.samplers.SampleResult;
   
10. import org.apache.jmeter.testelement.property.BooleanProperty;
    
11. import org.apache.jmeter.testelement.property.IntegerProperty;
    
12. import org.apache.jmeter.testelement.property.StringProperty;
    
13. import org.apache.jorphan.logging.LoggingManager;
    
14. import org.apache.log.Logger;
    
15. 
    
16. import java.util.concurrent.CountDownLatch;
    
17. import java.util.concurrent.atomic.AtomicInteger;
    
18. 
    
19. 
    
20. /**
    
21. * Created by sara on 12/10/2016.
    
22. */
    
23. public class FixSmapler extends AbstractSampler {
    
24. 
    
25.     private static final Logger log = LoggingManager.getLoggerForClass();
    
26. 
    
27. 
    
28.     private static final String HOST = "Fix.host"; //$NON-NLS-1$
    
29.     private static final String PORT = "Fix.port"; //$NON-NLS-1$
    
30.     private static final String SSL = "Fix.ssl"; //$NON-NLS-1$
    
31. 
    
32.     private static TestClient client = null;
    
33.     private static Message request  = null;
    
34. 
    
35. 
    
36.     private static AtomicInteger classCount = new AtomicInteger(0); // keep track of classes created
    
37. 
    
38.     public FixSmapler() {
    
39.        classCount.incrementAndGet();
    
40.         trace("FixSmapler()");
    
41.     }
    
42. 
    
43.     // 这个方法主要是用来执行和收集执行后的result数据的
    
44.     @Override
    
45.     public SampleResult sample(Entry entry) {
    
46.         trace("sample()");
    
47.         SampleResult res = new SampleResult();
    
48.         boolean isOK = false;
    
49. 
    
50.         TestClient c = getClient();
    
51.         // 设置结果的名称
    
52.         res.setSampleLabel(getTitle());
    
53. 
    
54.         // 设置request数据
    
55.         res.setSamplerData(request.toString());
    
56.         res.sampleStart();
    
57. 
    
58.         final CountDownLatch responseLatch = new CountDownLatch(1);
    
59. 
    
60.         StringBuffer sb = new StringBuffer();
    
61.         FixClient.Callback callback = new FixClient.Callback() {
    
62.             @Override
    
63.             public void onMessage(quickfix.Message message) {
    
64. 
    
65.                 String[] fds = message.toString().split("\\001");
    
66. 
    
67.                 for(String fd:fds){
    
68.                     sb.append(fd).append("\r\n");
    
69.                 }
    
70. 
    
71.                 responseLatch.countDown();
    
72.             }
    
73.         };
    
74. 
    
75.         try{
    
76. 
    
77.             int count = 0;
    
78.             while (!c.connect()){
    
79.                 Thread.sleep(1000);
    
80. 
    
81.                 count++;
    
82. 
    
83.                 if(count >=10){
    
84.                     throw new Exception("Fix server can't connect");
    
85.                 }
    
86.             }
    
87. 
    
88.             c.send(request,callback);
    
89.             responseLatch.await();
    
90.             //设置response数据
    
91.             res.setResponseMessage(sb.toString());
    
92.             byte[] bytes = sb.toString().getBytes();
    
93.             res.setResponseData(bytes);
    
94.             res.setDataType(SampleResult.TEXT);
    
95.             res.setResponseCodeOK();
    
96.             res.setResponseMessage("OK");
    
97.             isOK = true;
    
98.         }catch (Exception e){
    
99.             log.error("send fail: ".concat(e.toString()));
    
100.             res.setResponseCode("500");// $NON-NLS-1$
     
101.             res.setResponseData(e.toString());
     
102.         }finally {
     
103.             c.removeCallback(callback);
     
104.             c.disconnect();
     
105.         }
     
106. 
     
107.         res.sampleEnd();
     
108. 
     
109.         res.setSuccessful(isOK);
     
110.         return  res;
     
111.     }

复制代码

执行 ant install
安装过后，你直接运行NewDriver 和 /bin/jmeter 都是可以的。
截图

梦想家

支持分享

Miss_love

感谢分享