JMeter 参考文档 #

$ java -version
java version "1.8.0_301"
Java(TM) SE Runtime Environment (build 1.8.0_301-b09)
Java HotSpot(TM) 64-Bit Server VM (build 25.301-b09, mixed mode)

<dependencies>
    <!-- JMeter 核心依赖 -->
    <dependency>
        <groupId>org.apache.jmeter</groupId>
        <artifactId>ApacheJMeter_core</artifactId>
        <version>5.6.3</version>
    </dependency>
    <!-- HTTP 请求依赖 -->
    <dependency>
        <groupId>org.apache.jmeter</groupId>
        <artifactId>ApacheJMeter_http</artifactId>
        <version>5.6.3</version>
    </dependency>
    <!-- 测试元件依赖 -->
    <dependency>
        <groupId>org.apache.jmeter</groupId>
        <artifactId>ApacheJMeter_components</artifactId>
        <version>5.6.3</version>
    </dependency>
</dependencies>

import org.apache.jmeter.engine.StandardJMeterEngine;
import org.apache.jmeter.testelement.TestPlan;
import org.apache.jmeter.threads.ThreadGroup;
import org.apache.jmeter.protocol.http.sampler.HTTPSamplerProxy;
import org.apache.jmeter.config.Arguments;
import org.apache.jmeter.config.gui.ArgumentsPanel;
import org.apache.jmeter.samplers.Sampler;
import org.apache.jmeter.visualizers.ViewResultsFullVisualizer;
import org.apache.jmeter.reporters.ResultCollector;
import org.apache.jmeter.save.SaveService;
import org.apache.jmeter.util.JMeterUtils;
import org.apache.jmeter.reporters.Summariser;
import org.apache.jorphan.collections.HashTree;

import java.io.File;
import java.io.FileOutputStream;

public class JMeterJavaExample {
    public static void main(String[] args) throws Exception {
        // 1. 初始化 JMeter
        JMeterUtils.loadJMeterProperties("/path/to/jmeter/bin/jmeter.properties");
        JMeterUtils.setJMeterHome("/path/to/jmeter");
        JMeterUtils.initLocale();

        // 2. 创建测试计划
        TestPlan testPlan = new TestPlan("Java 测试计划");
        testPlan.setProperty(TestElement.TEST_CLASS, TestPlan.class.getName());
        testPlan.setProperty(TestElement.GUI_CLASS, TestPlanGui.class.getName());

        // 3. 创建线程组
        ThreadGroup threadGroup = new ThreadGroup();
        threadGroup.setName("测试线程组");
        threadGroup.setNumThreads(10);
        threadGroup.setRampUp(5);
        threadGroup.setLoopCount(1);
        threadGroup.setProperty(TestElement.TEST_CLASS, ThreadGroup.class.getName());
        threadGroup.setProperty(TestElement.GUI_CLASS, ThreadGroupGui.class.getName());

        // 4. 创建 HTTP 请求
        HTTPSamplerProxy httpSampler = new HTTPSamplerProxy();
        httpSampler.setName("HTTP 请求");
        httpSampler.setDomain("example.com");
        httpSampler.setPort(80);
        httpSampler.setPath("/api/test");
        httpSampler.setMethod("GET");
        httpSampler.setProperty(TestElement.TEST_CLASS, HTTPSamplerProxy.class.getName());
        httpSampler.setProperty(TestElement.GUI_CLASS, HttpTestSampleGui.class.getName());

        // 5. 创建结果收集器
        Summariser summariser = new Summariser("");
        String logFile = "result.jtl";
        ResultCollector logger = new ResultCollector(summariser);
        logger.setFilename(logFile);

        // 6. 构建测试树
        HashTree testPlanTree = new HashTree();
        testPlanTree.add(testPlan);
        HashTree threadGroupTree = testPlanTree.add(testPlan, threadGroup);
        threadGroupTree.add(httpSampler);
        threadGroupTree.add(logger);

        // 7. 运行测试
        StandardJMeterEngine jmeter = new StandardJMeterEngine();
        jmeter.configure(testPlanTree);
        jmeter.run();

        // 8. 保存测试计划
        SaveService.saveTree(testPlanTree, new FileOutputStream("test-plan.jmx"));
        System.out.println("测试完成，结果已保存到 " + logFile);
    }
}

import org.apache.jmeter.samplers.AbstractSampler;
import org.apache.jmeter.samplers.Entry;
import org.apache.jmeter.samplers.SampleResult;
import org.apache.jmeter.testelement.TestElement;

public class CustomSampler extends AbstractSampler {
    private static final String SAMPLE_MESSAGE = "SampleMessage";

    public CustomSampler() {
        setName("自定义采样器");
    }

    @Override
    public SampleResult sample(Entry entry) {
        SampleResult result = new SampleResult();
        result.setSampleLabel(getName());
        result.setSamplerData(getSampleMessage());
        
        result.sampleStart(); // 开始计时
        
        try {
            // 执行自定义逻辑
            Thread.sleep(100);
            String response = "自定义响应: " + getSampleMessage();
            
            result.setResponseData(response, "UTF-8");
            result.setDataType(SampleResult.TEXT);
            result.setResponseCodeOK();
            result.setResponseMessage("OK");
            result.setSuccessful(true);
        } catch (Exception e) {
            result.setSuccessful(false);
            result.setResponseMessage(e.getMessage());
            result.setResponseCode("500");
        } finally {
            result.sampleEnd(); // 结束计时
        }
        
        return result;
    }

    public String getSampleMessage() {
        return getPropertyAsString(SAMPLE_MESSAGE);
    }

    public void setSampleMessage(String message) {
        setProperty(SAMPLE_MESSAGE, message);
    }
}