Java 개발: 성능 테스트 및 로드 테스트 수행 방법

import org.apache.jmeter.config.Arguments;
import org.apache.jmeter.protocol.http.control.*;
import org.apache.jmeter.protocol.http.sampler.HTTPSamplerProxy;
import org.apache.jmeter.reporters.ResultCollector;
import org.apache.jmeter.save.SaveService;
import org.apache.jmeter.testelement.TestElement;
import org.apache.jmeter.threads.*;
import org.apache.jmeter.util.JMeterUtils;
import org.apache.jorphan.collections.HashTree;
import org.apache.jorphan.collections.ListedHashTree;

import java.io.*;
import java.util.concurrent.CountDownLatch;

public class PerformanceTest {
    private static final String JMETER_HOME = "D:/apache-jmeter-5.4.1";
    private static final String JMETER_PROPERTIES_FILE = JMETER_HOME + "/bin/jmeter.properties";
    private static final String JMETER_SCRIPT_FILE = "D:/test.jmx";
    private static final int THREAD_COUNT = 1000;
    private static final int LOOP_COUNT = 10;
    private static final String REPORT_FILE = "D:/report.txt";

    public static void main(String[] args) {
        // 初始化JMeter配置和环境
        JMeterUtils.setJMeterHome(JMETER_HOME);
        JMeterUtils.loadJMeterProperties(new FileReader(JMETER_PROPERTIES_FILE));
        JMeterUtils.initLogging();

        // 创建线程组
        ThreadGroup threadGroup = new ThreadGroup();
        threadGroup.setName("Thread Group");
        threadGroup.setNumThreads(THREAD_COUNT);
        threadGroup.setRampUp(1);
        threadGroup.setSamplerController(createSamplerLoopController());

        // 创建测试计划
        HashTree testPlanTree = new ListedHashTree();
        testPlanTree.add(testPlanTree.getTree(testPlanTree.add(threadGroup)));

        // 执行性能测试
        ResultCollector resultCollector = new ResultCollector();
        resultCollector.setFilename(REPORT_FILE);
        testPlanTree.add(testPlanTree.getArray()[0], resultCollector);

        SampleResult.setResponseDataEncoding("UTF-8");
        JMeterContextService.getContext().setSamplingStarted(true);

        JMeterThread thread = new JMeterThread(testPlanTree, threadGroup, new ListenerNotifier());
        thread.setInitialContext(JMeterContextService.getContext());
        thread.setThreadNum(0);
        thread.setThreadName("Thread 1");
        thread.setThreadGroup(threadGroup);
        thread.setThreadCounts(500);

        // 等待测试结果
        CountDownLatch countDownLatch = new CountDownLatch(1);
        thread.setCountDownLatch(countDownLatch);
        thread.run();

        try {
            countDownLatch.await();
        } catch (InterruptedException e) {
            e.printStackTrace();
        }

        System.exit(0);
    }

    private static LoopController createSamplerLoopController() {
        LoopController loopController = new LoopController();
        loopController.setLoops(LOOP_COUNT);
        loopController.setFirst(true);
        loopController.initialize();
        return loopController;
    }
}