jmeter之自定义java请求性能测试

软件测试君

发布于 2019-07-10 17:40:08

1.8K0

发布于 2019-07-10 17:40:08

文章被收录于专栏：测试人生测试人生

写在前面

最近个人情绪不稳定、心情很不美丽，特别懒，没有博客和技术文章更新，这个文章也是迟到了一周多的时间，啰嗦完了，开始正题吧。

我们在实际做性能测试的时候，一般会采用jmeter、LR进行测试，但是很少有同学用jmeter的测试Java请求，也可以这么说，一般是开发自己测试自己编写代码性能时用的多。

准备工作

自己编写一个接口（我用springboot随便写了一个，公号内回复接口服务demo即可）
新建一个maven工程
pom文件引用ApacheJMeter_core.ja、ApacheJMeter_java.jar

写代码前该知道的

1、创建类的时候需要去继承AbstractJavaSamplerClient抽象类或者是实现JavaSamplerClient接口，我在这里只继承AbstractJavaSamplerClient抽象类，需要复写的方法有

public SampleResult runTest(JavaSamplerContext context) {}
public void setupTest(JavaSamplerContext context) {}
public void teardownTest(JavaSamplerContext context) {}
public Arguments getDefaultParameters() {}

2、针对复写方法说明：

//这个方法由Jmeter在进行添加javaRequest时第一个运行，它决定了你要在GUI中默认显示出哪些属性。
public Arguments getDefaultParameters() {}
//这个方法相当于loadrunner中的init，我们可以用它来进行一些初始化的动作。
public void setupTest(JavaSamplerContext context) {}
//这个方法相当于loadrunner中的action，我们的核心测试代码就在这里了。
public SampleResult runTest(JavaSamplerContext context) {}
//这个方法相当于loadrunner中的end，收尾的工作可以由它来做。
public void teardownTest(JavaSamplerContext context) {}

示例代码

package com.jmeter.demo;

import org.apache.http.HttpEntity;

import org.apache.http.NameValuePair;

import org.apache.http.client.ClientProtocolException;

import org.apache.http.client.config.RequestConfig;

import org.apache.http.client.entity.UrlEncodedFormEntity;

import org.apache.http.client.methods.CloseableHttpResponse;

import org.apache.http.client.methods.HttpPost;

import org.apache.http.impl.client.CloseableHttpClient;

import org.apache.http.impl.client.HttpClients;

import org.apache.http.message.BasicNameValuePair;

import org.apache.http.util.EntityUtils;

import org.apache.jmeter.config.Arguments;

import org.apache.jmeter.protocol.java.sampler.AbstractJavaSamplerClient;

import org.apache.jmeter.protocol.java.sampler.JavaSamplerContext;

import org.apache.jmeter.samplers.SampleResult;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import java.io.IOException;

import java.util.ArrayList;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

/**

 * @author rongrong

 * jmeter之自定义java请求性能测试

 */

public class JmeterForJavaTest extends AbstractJavaSamplerClient {

    private static final Logger log = LoggerFactory.getLogger(AbstractJavaSamplerClient.class);

    //测试结果

    private SampleResult sr;

    private static String lable = "放大招了奥！";

    /**

     * 初始化性能测试时，每个测试线程

     *

     * @param context

     */

    @Override

    public void setupTest(JavaSamplerContext context) {

        log.debug(getClass().getName() + ": setupTest");

    }

    /**

     * 主要设置传入的参数

     *

     * @return

     */

    @Override

    public Arguments getDefaultParameters() {

        Arguments arguments = new Arguments();

        arguments.addArgument("url", "http://localhost:8888/login?");

        arguments.addArgument("userName", "rongrong");

        arguments.addArgument("passWord", "123456");

        return arguments;

    }

    /**

     * 为性能测试的运行体

     *

     * @param context

     * @return

     */

    @Override

    public SampleResult runTest(JavaSamplerContext context) {

        String url = context.getParameter("url");

        String userName = context.getParameter("userName");

        String passWord = context.getParameter("passWord");

        String sbResultData = null;

        sr = new SampleResult();

        sr.setSampleLabel(lable);

        //记录测试时间及结果

        sr.sampleStart();

        //执行请求

        sbResultData = executRequest(url, userName, passWord, sbResultData);

        sr.setSuccessful(true);

        sr.setResponseCode("200");

        sr.setResponseCodeOK();

        sr.setRequestHeaders("这是个header呀，但是我就没写呀");

        sr.setResponseMessage("这块是自己瞎写的呀，测试是不是该严谨些呢");

        sr.setContentType("application/json 没想到这块我也要皮一下呢");

        //将数据打印到查看结果树当中

        sr.setResponseData(sbResultData, null);

        sr.setDataType(SampleResult.TEXT);

        return sr;

    }

    /**

     * 模拟接口请求

     *

     * @param url

     * @param userName

     * @param passWord

     * @param sbResultData

     * @return

     */

    private String executRequest(String url, String userName, String passWord, String sbResultData) {

        CloseableHttpClient httpClient = null;

        HttpPost httpPost = null;

        try {

            httpClient = HttpClients.createDefault();

            RequestConfig requestConfig = RequestConfig.custom().setSocketTimeout(20000).setConnectTimeout(20000).build();

            httpPost = new HttpPost(url);

            httpPost.setConfig(requestConfig);

            List<NameValuePair> ps = new ArrayList<NameValuePair>();

            Map<String, String> params = new HashMap<String, String>();

            params.put("userName", userName);

            params.put("passWord", passWord);

            for (String pKey : params.keySet()) {

                ps.add(new BasicNameValuePair(pKey, params.get(pKey)));

            }

            httpPost.setEntity(new UrlEncodedFormEntity(ps));

            CloseableHttpResponse response = httpClient.execute(httpPost);

            HttpEntity httpEntity = response.getEntity();

            sbResultData = EntityUtils.toString(httpEntity, "utf-8");

        } catch (ClientProtocolException e) {

            e.printStackTrace();

        } catch (IOException e) {

            e.printStackTrace();

        } finally {

            try {

                if (httpPost != null) {

                    httpPost.releaseConnection();

                }

                if (httpClient != null) {

                    httpClient.close();

                }

            } catch (IOException e) {

                e.printStackTrace();

            }

        }

        return sbResultData;

    }

    @Override

    public void teardownTest(JavaSamplerContext context) {

        log.debug(getClass().getName() + ": teardownTest");

    }

}

jmeter运行java代码