springboot项目组引入JMeter的实现步骤

目录

• 一、前言
  
• 二、springboot项目引入JMeter完成接口自动化操作
  

    5.3




    org.apache.jmeter
    ApacheJMeter_http
    ${jmeter.version}
    
        
            org.apache.logging.log4j
            log4j-slf4j-impl
        
    


    org.apache.jmeter
    ApacheJMeter_functions
    ${jmeter.version}


    org.apache.jmeter
    ApacheJMeter_jdbc
    ${jmeter.version}


    org.apache.jmeter
    ApacheJMeter_tcp
    ${jmeter.version}

//创建engine
StandardJMeterEngine engine = new StandardJMeterEngine();
//将hashTree配置注入engine中
engine.configure(hashTree); 
try {
 //执行engine开始运行
    engine.runTest();
} catch (JMeterEngineException e) {
    engine.stopTest();
    e.printStackTrace();
}

//初始化jmeter属性配置
private void initJMeterProperties() {
    if (!StringUtils.isEmpty(JMeterUtils.getJMeterProperties())){
        return;
    }
    try {
        InputStream inputStream = JMeterUtil.class.getResource("/jmeter/jmeter.properties").openStream();
        File tempFile = FileUtil.createTempFile(null);
        FileUtil.writeFromStream(inputStream,tempFile);
        //这里面loadJMeterProperties方法必须写成临时文件这样的形式，否则会获取不到jmeter.properties
        JMeterUtils.loadJMeterProperties(tempFile.getAbsolutePath());
        JMeterUtils.setJMeterHome(JMeterUtil.PATH);
        JMeterUtils.setLocale(LocaleContextHolder.getLocale());
    } catch (IOException e) {
        e.printStackTrace();
    }
}

public class CustomBackendListenerClient extends AbstractBackendListenerClient {

    private final List queue = new ArrayList<>();

    //获得控制台内容。
    private PrintStream oldPrintStream = System.out;
    private ByteArrayOutputStream bos = new ByteArrayOutputStream();

    private void setConsole() {
        System.setOut(new PrintStream(bos));
    }

    private String getConsole() {
        System.setOut(oldPrintStream);
        return bos.toString();
    }
 // engine执行之前会进行前置处理器
    @Override
    public void setupTest(BackendListenerContext context) throws Exception {
        setConsole();
        super.setupTest(context);
    }
 //engine执行中的处理
    @Override
    public void handleSampleResults(List sampleResults, BackendListenerContext context) {
     //结果集添加至集合中
        queue.addAll(sampleResults);
    }
 //engine结束后的后置处理器
    @Override
    public void teardownTest(BackendListenerContext context) throws Exception {
     //处理结果集中的数据并封装至JMeterRequestResult对象中
        RunJMeterRequestService runJMeterRequestService = CommonBeanFactory.getBean(RunJMeterRequestService.class);
        List jMeterRequestResults = Lists.newArrayList();
        String testId = context.getParameter("testId");
        queue.stream().forEach(result -> {
            setRequestResult(result, jMeterRequestResults);
        });
        queue.clear();
        runJMeterRequestService.addDebugResult(testId, jMeterRequestResults);
        super.teardownTest(context);
    }

    private void setRequestResult(SampleResult result, List jMeterRequestResults) {
        JMeterRequestResult metricResult = new JMeterRequestResult();
        Long respOnseTime= result.getEndTime() - result.getStartTime();
        metricResult.setUrl(result.getUrlAsString());
        metricResult.setResponseSize(((Integer) result.getBodySize()).toString());
        metricResult.setResponseTime(responseTime.toString());
        metricResult.setResponseResult(result.getResponseDataAsString());
        metricResult.setConsoleResult(getConsole());
        if (result instanceof HTTPSampleResult) {
            HTTPSampleResult res = (HTTPSampleResult) result;
            metricResult.setCOOKIE(res.getCOOKIEs());
            metricResult.setRequestMethod(res.getHTTPMethod());
        }
        metricResult.setRequestData(result.getSamplerData());
        metricResult.setResponseHeader(result.getResponseHeaders());
        metricResult.setRequestHeader(result.getRequestHeaders());
        metricResult.setStatusCode(result.getResponseCode());
        jMeterRequestResults.add(metricResult);
    }
}