Jmeter Java请求

一、为什么要使用Java请求

现有的post 、 get 无法对加解密及鉴权的接口进行测试，需要进行封装，然后再请求，怎么进行封装？通常使用的方式是使用httpclient 工具类型来操作，下面举一个简单的Java自定义测试代码例子,使用Java编译器编写测试代码

二、完成以下几部曲即可

1、打开IDE（这里我使用的是 eclipse Luna 4.3），创建一个Java工程项目名为enterprise

2、从Jmeter的安装目录lib/ext中拷贝两个文件ApacheJMeter_core.jar（必须）和ApacheJMeter_java.jar（必须）到enterprise项目中，然后引入这两个JAR包；如果需要在eclipse中调试还需要拷贝以下三个工具包可选avalon-framework-4.1.4.jar(jmeter3.2中名称是framework)、jorphan.jar、logkit-2.0.jar。

3、创建一个测试类CipherTypeRSA 继承AbstractJavaSamplerClient存在于ApacheJMeter_java.jar这个JAR包中或实现JavaSamplerClient 接口，两种方式其中一种引用即可调用

4、需要实现四个方法，分别是：

三、直接上具体代码

说明：以下代码已脱敏处理，直接复制是跑不通的，主要是领悟其用法

package com.hcb.enterprise;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;
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.wlqq.codec.enums.CipherType;
import org.wlqq.codec.exception.CodecHttpException;
import org.wlqq.codec.http.client.CodecHttpTemplate;
import org.wlqq.codec.http.client.HttpTemplateBuilder;

public class CipherTypeRSA extends AbstractJavaSamplerClient  {  
    private String label;
    private String resultData;  
    private String url;
    private String host;
    private String uri;
    private String appId;
    private String privateKey;
    Map<String, String> map = new HashMap<>();
    private static final CodecHttpTemplate CODEC_HTTP_TEMPLATE = HttpTemplateBuilder.create().build();

    public Arguments getDefaultParameters() {  
        Arguments params = new Arguments(); 
        params.addArgument("label","exec name");
        params.addArgument("appId", "open-enterprise");  
        params.addArgument("privateKey", "MIICdQIBADANBgkqhkiG9w0BAQEFAASCAl8wggJbAgEAAoGBAJPq9N65uR+bHxtUKFv4YgSV5Voa6jLKGhYwPGqAItthp");  
        params.addArgument("host","http://enterprise-api.dev-ag.56qq.com");
        params.addArgument("uri","/server/servicetask/list");
        params.addArgument("content","");
        return params;  
    }  
    
  //每个线程测试前执行一次，做一些初始化工作；  
    public void setupTest(JavaSamplerContext arg0) {  
        Iterator<String> it =arg0.getParameterNamesIterator();
        while(it.hasNext()){
            String key=it.next();
            String value = arg0.getParameter(key);
            if("host".equals(key)){
                host = arg0.getParameter(key);
            }else if("label".equals(key)){
                label = arg0.getParameter(key);
            }else if("uri".equals(key)){
                uri = arg0.getParameter(key);
            }else if("appId".equals(key)){
                appId = arg0.getParameter(key);
            }else if("privateKey".equals(key)){
                privateKey = arg0.getParameter(key);
            }else if("content".equals(key)){
                getContent(arg0.getParameter(key));
            }else{
                map.put(key, value);
            }
            url = host + uri;
        }
    }  
      
    public SampleResult runTest(JavaSamplerContext arg0) {         
        SampleResult sr = new SampleResult();  
        try {  
            sr.setSampleLabel(label);
            sr.sampleStart();// jmeter 开始统计响应时间标记  
            resultData = doPostSign(url, map, appId, privateKey, CipherType.RSA);
            if (resultData != null && resultData.length() > 0) {  
                sr.setResponseData(resultData, null);  //将结果输出到结果树当中
                sr.setDataType(SampleResult.TEXT);    //请求头中会多个类型输出 Data type ("text"|"bin"|""): text
            }  
            //System.out.println(resultData);  //会输出到控制台，使用Jmeter 会输出到控制台或日志里面
            sr.setSuccessful(true);  
        } catch (Throwable e) {  
            sr.setSuccessful(false);  
            sr.setResponseData("Check the input parameters is correct !",null);  
            e.printStackTrace();
        } finally {  
            sr.sampleEnd(); //Jmeter 结束统计响应时间标记  
        }  
        return sr;  
    }  
  
    public void teardownTest(JavaSamplerContext arg0) {  
       
    }  
/**
    public static void main(String[] args) {  
        Arguments params = new Arguments();  
        params.addArgument("appId", "open-enterprise");  
        params.addArgument("privateKey", "MIICdQIBADANBgkqhkiG9w0BAQEFAASCAl8wggJbAgEAAoGBAJPq9N65uR+bHxtUKFv4YgSV5Voa6jLKGhYwPGqAItthp");  
        params.addArgument("host","http://enterprise-api.dev-ag.com");
        params.addArgument("uri","/server/company/get-company-by-name");
        params.addArgument("content", "fullName=XX科技有限公司");
        JavaSamplerContext arg0 = new JavaSamplerContext(params);  
        CipherTypeRSA test = new CipherTypeRSA();  
        test.setupTest(arg0);  
        test.runTest(arg0); 
        test.teardownTest(arg0);
    }  */
    
    private  String doPostSign(String url, Map<String, String> params, String appId, String privateKey,
            CipherType cipherType) throws CodecHttpException {
        if (params != null) {
            params.put("__APPID__", appId);
        }
        String result = CODEC_HTTP_TEMPLATE.doPostSign(url, params, appId, privateKey, cipherType);
        return result;
    }
    
    // 拼接的字符串转换成键值对存进map 集合中 content = "a=123&b=&c=234";
    private  void getContent(String content){
        if(!content.isEmpty()&&!content.startsWith("&")){
            String[] list = content.split("&");
            for(String s:list){
                Boolean es = s.endsWith("=");
                if(es){
                    map.put(s.substring(0,s.length()-1),"");
                }else{
                    String[] mapList = s.split("=");
                    map.put(mapList[0],mapList[1]);
                }
            }
        }
    }
}  

四、 把Java项目导出成.jar，并放于apache-jmeter-3.0libext文件夹里面，Java项目lib的依赖包在打成.jar 时生成，所以不用单独处理

导出

五、创建Jmeter java 请求

六、查看响应结果

 

综上按照以上写个Java请求是没有什么问题的，也没有什么难度，关键的需要准备一个工具类（Java请求加解密、鉴权等的工具类）

补充：封装完成的结果函数

private void setSampleResultFailed(SampleResult sr, int status, String result) {
   sr.setSuccessful(false);
   sr.setResponseCode(String.valueOf(status));
   sr.setResponseData(result, "UTF-8");    //把响应结果返回到jmeter中
   sr.setResponseOK();
}

private void setSampleResultSuccess(SampleResult sr, int status, String result) {
   sr.setSuccessful(true);
   sr.setResponseCode(String.valueOf(status));
   sr.setResponseData(result, "UTF-8");
   sr.setResponseOK();
}