Jmeter实现RPC接口测试简洁化需求

根据需求，产品部分功能采用thrift-RPC协议进行接口的增、删、改、查，前期采用Junit对其进行测试，为了提高RPC接口测试的简洁化和后期的性能测试需求，打算通过Jmeter的java类测试实现。

前期准备

开发提供了IDL描述的Thrift文件：dataService.thrift，其中定义了相关接口函数及数据类型，如下：

enum VStatus {
    ON,
    OFF
}

struct VData {
    1: required string tablename;
    2: required string time;
    3: required string jsons;
    4: optional VStatus status;
}

struct VCondition {
    1: required string tablename;
    2: required i64 starttime;
    3: required i64 endtime;
}

service DataService {
    string write(1: VData data);
    string read(1: VCondition condition);
}

在本地部署thrift.exe并设置环境变量，之后在dataService.thrift所在的目录下执行：

thrift -r -gen java dataService.thrift              //备注：我这里是用java语言；

执行成功后在该目录下产生gen-java文件夹，打开该文件夹至api目录下，生成4个文件：DataService.java、VCondition.java、VData.java和VStatus.java。

基于Jmeter的jar代码编写

创建Maven项目，编辑pom.xml添加依赖，如下所示：

<dependencies>
    <dependency>
      <groupId>org.apache.thrift</groupId>
      <artifactId>libthrift</artifactId>
      <version>0.11.0</version>
    </dependency>

    <dependency>
      <groupId>org.springframework.boot</groupId>
      <artifactId>spring-boot-starter</artifactId>
    </dependency>
    <dependency>
      <groupId>org.springframework.boot</groupId>
      <artifactId>spring-boot-starter-test</artifactId>
    </dependency>   
       <dependency>
           <groupId>org.apache.logging.log4j</groupId>
           <artifactId>log4j-api</artifactId>
           <version>2.8.2</version>
       </dependency>
       <dependency>
           <groupId>org.apache.logging.log4j</groupId>
           <artifactId>log4j-core</artifactId>
           <version>2.8.2</version>
       </dependency>           

   </dependencies>

之后手动加载本地Jmeter版本目录\Jmeter_Home\lib\ext中的ApacheJMeter_core.jar和ApacheJMeter_java.jar作为外部依赖包（注：直接maven加载相关版本的依赖包执行时存在冲突）。

将之前生成的4个java文件导入项目，设置包名为：com.test.thrift.api，在项目包名下创建ReadRPC.java类，根据Jmeter中java类要求需要继承 AbstractJavaSamplerClient，代码如下：

package com.test.RPCTest;

import com.test.thrift.api.DataService;
import com.test.thrift.api.VCondition;

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.apache.thrift.protocol.TBinaryProtocol;
import org.apache.thrift.protocol.TProtocol;
import org.apache.thrift.transport.TSocket;
import org.apache.thrift.transport.TTransport;

public class ReadRPC extends AbstractJavaSamplerClient{
    private TTransport tTransport;
    private DataService.Client dataService;
    private TProtocol tProtocol;
    private VCondition vCondition;

    @Override      
    public Arguments getDefaultParameters(){  //默认参数值，包括服务ip、端口、目标表、起止时间等
        Arguments arguments=new Arguments();
        arguments.addArgument("IP", "127.0.0.1");
        arguments.addArgument("Port", "8888");
        arguments.addArgument("TableName", "vle::test.trpc");
        arguments.addArgument("StartTime", "1539752449");
        arguments.addArgument("EndTime", "1539752449");
        return arguments;
    }
    
    @Override
    public void setupTest(JavaSamplerContext context) { //初始化参数，并通过TTransport模式TBinaryProtocol二进制协议与服务端建立连接
        // TODO Auto-generated method stub
        String ip=context.getParameter("IP");
        int port=context.getIntParameter("Port");
        String tableName=context.getParameter("TableName");
        long startTime=context.getLongParameter("StartTime");
        long endTime=context.getLongParameter("EndTime");
        tTransport=new TSocket(ip,port,30000);
        vCondition=new VCondition();
        try {
            tTransport.open();
            tProtocol=new tBinaryProtocol(tTransport);
            dataService=new DataService.Client(tProtocol);
            vCondition.setTablename(tableName);
            vCondition.setStarttime(startTime);
            vCondition.setEndtime(endTime);
        } catch (Exception e) {
            // TODO: handle exception
        }
        super.setupTest(context);
    }
    
    @Override
    public SampleResult runTest(JavaSamplerContext context) {
        // TODO Auto-generated method stub
        SampleResult sampleResult = new SampleResult();
        sampleResult.setDataEncoding("utf8");
        sampleResult.sampleStart();
        try {
            String msg=dataService.read(tCondition);//执行读操作，返回响应结果信息到msg
            
            if(null==msg){//为空判断
                sampleResult.setSuccessful(false);
                sampleResult.setResponseMessage("响应结果为空！");
                sampleResult.setSampleLabel("RPC读取");
                sampleResult.setResponseOK();
                sampleResult.setResponseData("响应结果为空！");
                sampleResult.setDataType(SampleResult.TEXT);
                return sampleResult;
            }
            else {
                sampleResult.setSuccessful(true);
                sampleResult.setSampleLabel("RPC读取");
                sampleResult.setResponseOK();
                sampleResult.setResponseData(msg);
                System.out.println("msg="+msg);
                sampleResult.setDataType(SampleResult.TEXT);
                return sampleResult;
            }
            
        } catch (Exception e) {
            // TODO: handle exception
            sampleResult.setSuccessful(false);
            e.printStackTrace();
        }
        finally{
            sampleResult.sampleEnd();
        }
        return sampleResult;
    }
    
    public void teardownTest(JavaSamplerContext context){
        tTransport.close();//关闭连接
        super.teardownTest(context);
    }

}

在main函数中调用ReadRPC相关函数，代码如下：

public static void main( String[] args )
    {
        JmeterRPC jrpc=new JmeterRPC();
        jrpc.read();

    }
    
    public void read(){
        Arguments arguments=new Arguments();
        arguments.addArgument("IP", "127.0.0.1");
        arguments.addArgument("Port", "8888");
        arguments.addArgument("TableName", "vle::test.trpc");
        arguments.addArgument("StartTime", "1539752449");
        arguments.addArgument("EndTime", "1539752449");
        JavaSamplerContext context=new JavaSamplerContext(arguments);
        ReadRPC rpc=new ReadRPC();
        rpc.setupTest(context);
        rpc.runTest(context);
        rpc.teardownTest(context);
    }

运行程序输出msg=[{“………相关响应数据…………..”}]。调试成功后保存后，选中项目右键“Export”->“Java”->“Runnable JAR file”，点击“Next”，设置输出路径和依赖库，如下图所示：