最近准备给jmeter增加一个web页面，不想通过上传jmx文件的方式，而是直接根据所填的信息生成jmx文件，下面展示一个最简单的http请求，其他的组件可以参考下面的组装。
import org.apache.jmeter.config.Arguments;
import org.apache.jmeter.control.LoopController; import org.apache.jmeter.engine.JMeterEngine; import org.apache.jmeter.engine.JMeterEngineException; 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.sampler.HTTPSamplerProxy; import org.apache.jmeter.protocol.http.util.HTTPArgument; import org.apache.jmeter.reporters.ResultCollector; 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.*; import org.apache.jmeter.util.JMeterUtils; import org.apache.jorphan.collections.HashTree; import org.apache.jorphan.collections.ListedHashTree; import org.testng.annotations.Test; import org.apache.jmeter.threads.ThreadGroup; import java.io.File; import java.io.IOException; import java.util.*; public class JmeterTest { static{ { JMeterUtils.setJMeterHome(new File("").getAbsolutePath()); JMeterUtils.loadJMeterProperties(new File("src/main/resources/jmeter/jmeter.properties").getAbsolutePath()); JMeterUtils.setProperty("saveservice_properties", "src/main/resources/jmeter/saveservice.properties"); JMeterUtils.setProperty("user_properties", "src/main/resources/jmeter/user.properties"); JMeterUtils.setProperty("upgrade_properties", "src/main/resources/jmeter/upgrade.properties"); JMeterUtils.setProperty("system_properties", "src/main/resources/jmeter/system.properties"); // JMeterUtils.setProperty("proxy.cert.directory", new File("").getAbsolutePath()); JMeterUtils.setLocale(Locale.CHINESE); JMeterUtils.initLocale(); } } @Test public void test() throws JMeterEngineException, IOException { File file = new File("D:\\jmeter_jmx\\HTTP Request.jmx"); HashTree hashTree = SaveService.loadTree(file); JMeterEngine engine = new StandardJMeterEngine(); engine.configure(hashTree); ((StandardJMeterEngine) engine).run(); } @Test public void test04() throws IOException { String url = "172.16.2.143"; String port = "8080"; String api = "/submit/job"; String request = "{}"; //TestPlan TestPlan testPlan = new TestPlan("Test Plan"); testPlan.setFunctionalMode(false); testPlan.setSerialized(false); testPlan.setTearDownOnShutdown(true); testPlan.setProperty(TestElement.TEST_CLASS, TestPlan.class.getName()); testPlan.setProperty(TestElement.GUI_CLASS,"TestPlanGui"); testPlan.setProperty(new BooleanProperty(TestElement.ENABLED, true)); testPlan.setProperty(new StringProperty("TestPlan.comments", "")); testPlan.setProperty(new StringProperty("TestPlan.user_define_classpath", "")); Arguments arguments = new Arguments(); testPlan.setProperty(new TestElementProperty("TestPlan.user_defined_variables",arguments)); ThreadGroup threadGroup = new ThreadGroup(); threadGroup.setNumThreads(1); threadGroup.setRampUp(1); threadGroup.setDelay(0); threadGroup.setDuration(0); threadGroup.setProperty(new StringProperty(ThreadGroup.ON_SAMPLE_ERROR, "continue")); threadGroup.setScheduler(false); threadGroup.setName("Group1"); threadGroup.setProperty(TestElement.TEST_CLASS, ThreadGroup.class.getName()); threadGroup.setProperty(TestElement.GUI_CLASS,"ThreadGroup"); threadGroup.setProperty(new BooleanProperty(TestElement.ENABLED, true)); LoopController loopController = new LoopController(); loopController.setLoops(0); loopController.setProperty(new BooleanProperty("LoopController.continue_forever",false)); loopController.setProperty(new StringProperty("TestElement.gui_class","org.apache.jmeter.control.gui.LoopControlPanel")); loopController.setProperty(new StringProperty("TestElement.test_class","org.apache.jmeter.control.LoopController")); loopController.setProperty(new StringProperty("TestElement.name","Loop Controller")); loopController.setProperty(new StringProperty("TestElement.enabled","true")); loopController.setProperty(new StringProperty("LoopController.loops","1")); threadGroup.setProperty(new TestElementProperty("ThreadGroup.main_controller",loopController)); HTTPSamplerProxy httpSamplerProxy = new HTTPSamplerProxy(); Arguments HTTPsamplerArguments = new Arguments(); HTTPArgument httpArgument = new HTTPArgument(); httpArgument.setProperty(new BooleanProperty("HTTPArgument.always_encode",false)); httpArgument.setProperty(new StringProperty("Argument.value", request)); httpArgument.setProperty(new StringProperty("Argument.metadata","=")); ArrayList<TestElementProperty> list1 = new ArrayList<>(); list1.add(new TestElementProperty("",httpArgument)); HTTPsamplerArguments.setProperty(new CollectionProperty("Arguments.arguments",list1)); httpSamplerProxy.setProperty(new TestElementProperty("HTTPsampler.Arguments",HTTPsamplerArguments)); httpSamplerProxy.setProperty(new StringProperty("HTTPSampler.domain", url)); httpSamplerProxy.setProperty(new StringProperty("HTTPSampler.port", port)); httpSamplerProxy.setProperty(new StringProperty("HTTPSampler.protocol", "http")); httpSamplerProxy.setProperty(new StringProperty("HTTPSampler.path", api)); httpSamplerProxy.setProperty(new StringProperty("HTTPSampler.method", "POST")); httpSamplerProxy.setProperty(new StringProperty("HTTPSampler.contentEncoding", "")); httpSamplerProxy.setProperty(new BooleanProperty("HTTPSampler.follow_redirects", true)); httpSamplerProxy.setProperty(new BooleanProperty("HTTPSampler.postBodyRaw", true)); httpSamplerProxy.setProperty(new BooleanProperty("HTTPSampler.auto_redirects", false)); httpSamplerProxy.setProperty(new BooleanProperty("HTTPSampler.use_keepalive", true)); httpSamplerProxy.setProperty(new BooleanProperty("HTTPSampler.DO_MULTIPART_POST", false)); httpSamplerProxy.setProperty(new StringProperty("TestElement.gui_class", "org.apache.jmeter.protocol.http.control.gui.HttpTestSampleGui")); httpSamplerProxy.setProperty(new StringProperty("TestElement.test_class", "org.apache.jmeter.protocol.http.sampler.HTTPSamplerProxy")); httpSamplerProxy.setProperty(new StringProperty("TestElement.name", "HTTP Request")); httpSamplerProxy.setProperty(new StringProperty("TestElement.enabled", "true")); httpSamplerProxy.setProperty(new BooleanProperty("HTTPSampler.postBodyRaw", true)); httpSamplerProxy.setProperty(new StringProperty("HTTPSampler.embedded_url_re", "")); httpSamplerProxy.setProperty(new StringProperty("HTTPSampler.connect_timeout", "")); httpSamplerProxy.setProperty(new StringProperty("HTTPSampler.response_timeout", "")); ResultCollector resultCollector = new ResultCollector(); HeaderManager headerManager = new HeaderManager(); 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(false); sampleSaveConfiguration.setAssertions(true); sampleSaveConfiguration.setSubresults(true); sampleSaveConfiguration.setResponseData(false); sampleSaveConfiguration.setSamplerData(false); sampleSaveConfiguration.setAsXml(false); sampleSaveConfiguration.setFieldNames(true); sampleSaveConfiguration.setResponseHeaders(false); sampleSaveConfiguration.setRequestHeaders(false); //sampleSaveConfiguration.setAssertionResultsFailureMessage(true); responseDataOnError 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); resultCollector.setProperty(new BooleanProperty("ResultCollector.error_logging",false)); resultCollector.setProperty(new ObjectProperty("saveConfig",new SampleSaveConfiguration())); resultCollector.setProperty(new StringProperty("TestElement.gui_class","org.apache.jmeter.visualizers.ViewResultsFullVisualizer")); resultCollector.setProperty(new StringProperty("TestElement.name","View Results Tree")); resultCollector.setProperty(new StringProperty("TestElement.enabled","true")); resultCollector.setProperty(new StringProperty("filename","")); ArrayList<TestElementProperty> list2 = new ArrayList<>(); Header header = new Header(); header.setProperty(new StringProperty("Header.name","Content-Type")); header.setProperty(new StringProperty("Header.value","application/json")); TestElementProperty HeaderElement = new TestElementProperty("",header); list2.add(HeaderElement); headerManager.setProperty(new CollectionProperty("HeaderManager.headers",list2)); headerManager.setProperty(new StringProperty("TestElement.test_class","org.apache.jmeter.protocol.http.control.HeaderManager")); headerManager.setProperty(new StringProperty("TestElement.name","HTTP Header Manager")); headerManager.setProperty(new StringProperty("TestElement.enabled","true")); headerManager.setProperty(new StringProperty("TestElement.gui_class","org.apache.jmeter.protocol.http.gui.HeaderPanel")); ListedHashTree hashTreeResultCollectorAndHeaderManager = new ListedHashTree(); hashTreeResultCollectorAndHeaderManager.add(resultCollector); hashTreeResultCollectorAndHeaderManager.add(headerManager); ListedHashTree hashTreeHTTPSamplerProxy = new ListedHashTree(); hashTreeHTTPSamplerProxy.add(httpSamplerProxy,hashTreeResultCollectorAndHeaderManager); ListedHashTree hashTreeThreadGroup = new ListedHashTree(); hashTreeThreadGroup.add(threadGroup,hashTreeHTTPSamplerProxy); ListedHashTree hashTreeTestPlan = new ListedHashTree(); hashTreeTestPlan.add(testPlan,hashTreeThreadGroup); //可以将生成的hashTree保存成一个jmx文件 与Jmeter自己生成的做对比 //SaveService.saveTree(hashTreeTestPlan, new FileOutputStream("D:\\jmeter_jmx\\test.jmx")); JMeterEngine engine = new StandardJMeterEngine(); //engine.configure(SaveService.loadTree(new File("D:\\jmeter_jmx\\test.jmx"))); engine.configure(hashTreeTestPlan); ((StandardJMeterEngine) engine).run(); } }
上方代码中Test04生成的jmx文件如下所示
<?xml version="1.0" encoding="UTF-8"?> <jmeterTestPlan version="1.2" properties="5.0" jmeter="5.2.1"> <hashTree> <TestPlan guiclass="TestPlanGui" testclass="TestPlan" testname="Test Plan" enabled="true"> <boolProp name="TestPlan.functional_mode">false</boolProp> <boolProp name="TestPlan.serialize_threadgroups">false</boolProp> <boolProp name="TestPlan.tearDown_on_shutdown">true</boolProp> <stringProp name="TestPlan.comments"></stringProp> <stringProp name="TestPlan.user_define_classpath"></stringProp> <elementProp name="TestPlan.user_defined_variables" elementType="Arguments"> <collectionProp name="Arguments.arguments"/> </elementProp> </TestPlan> <hashTree> <ThreadGroup guiclass="ThreadGroup" testclass="ThreadGroup" testname="Group1" enabled="true"> <intProp name="ThreadGroup.num_threads">1</intProp> <intProp name="ThreadGroup.ramp_time">1</intProp> <longProp name="ThreadGroup.delay">0</longProp> <longProp name="ThreadGroup.duration">0</longProp> <stringProp name="ThreadGroup.on_sample_error">continue</stringProp> <boolProp name="ThreadGroup.scheduler">false</boolProp> <elementProp name="ThreadGroup.main_controller" elementType="LoopController" guiclass="LoopControlPanel" testclass="LoopController" testname="Loop Controller" enabled="true"> <boolProp name="LoopController.continue_forever">false</boolProp> <stringProp name="LoopController.loops">1</stringProp> </elementProp> </ThreadGroup> <hashTree> <HTTPSamplerProxy guiclass="HttpTestSampleGui" testclass="HTTPSamplerProxy" testname="HTTP Request" enabled="true"> <elementProp name="HTTPsampler.Arguments" elementType="Arguments"> <collectionProp name="Arguments.arguments"> <elementProp name="" elementType="HTTPArgument"> <boolProp name="HTTPArgument.always_encode">false</boolProp> <stringProp name="Argument.value">{}</stringProp> <stringProp name="Argument.metadata">=</stringProp> </elementProp> </collectionProp> </elementProp> <stringProp name="HTTPSampler.domain">172.16.2.143</stringProp> <stringProp name="HTTPSampler.port">8080</stringProp> <stringProp name="HTTPSampler.protocol">http</stringProp> <stringProp name="HTTPSampler.path">/submit/job</stringProp> <stringProp name="HTTPSampler.method">POST</stringProp> <stringProp name="HTTPSampler.contentEncoding"></stringProp> <boolProp name="HTTPSampler.follow_redirects">true</boolProp> <boolProp name="HTTPSampler.postBodyRaw">true</boolProp> <boolProp name="HTTPSampler.auto_redirects">false</boolProp> <boolProp name="HTTPSampler.use_keepalive">true</boolProp> <boolProp name="HTTPSampler.DO_MULTIPART_POST">false</boolProp> <stringProp name="HTTPSampler.embedded_url_re"></stringProp> <stringProp name="HTTPSampler.connect_timeout"></stringProp> <stringProp name="HTTPSampler.response_timeout"></stringProp> </HTTPSamplerProxy> <hashTree> <ResultCollector guiclass="ViewResultsFullVisualizer" testname="View Results Tree" enabled="true"> <boolProp name="ResultCollector.error_logging">false</boolProp> <objProp> <name>saveConfig</name> <value class="SampleSaveConfiguration"> <time>true</time> <latency>true</latency> <timestamp>true</timestamp> <success>true</success> <label>true</label> <code>true</code> <message>true</message> <threadName>true</threadName> <dataType>true</dataType> <encoding>false</encoding> <assertions>true</assertions> <subresults>true</subresults> <responseData>false</responseData> <samplerData>false</samplerData> <xml>false</xml> <fieldNames>true</fieldNames> <responseHeaders>false</responseHeaders> <requestHeaders>false</requestHeaders> <responseDataOnError>false</responseDataOnError> <saveAssertionResultsFailureMessage>true</saveAssertionResultsFailureMessage> <assertionsResultsToSave>0</assertionsResultsToSave> <bytes>true</bytes> <sentBytes>true</sentBytes> <url>true</url> <threadCounts>true</threadCounts> <idleTime>true</idleTime> <connectTime>true</connectTime> </value> </objProp> <stringProp name="filename"></stringProp> </ResultCollector> <hashTree/> <HeaderManager guiclass="HeaderPanel" testclass="HeaderManager" testname="HTTP Header Manager" enabled="true"> <collectionProp name="HeaderManager.headers"> <elementProp name="" elementType="Header"> <stringProp name="Header.name">Content-Type</stringProp> <stringProp name="Header.value">application/json</stringProp> </elementProp> </collectionProp> </HeaderManager> <hashTree/> </hashTree> </hashTree> </hashTree> </hashTree> </jmeterTestPlan>
如果需要添加其他元素的，可以参考jmeter生成jmx的结构，在此基础上自行拼装