所在位置:首页 → 编程语言 → Java → Jmeter实现RPC接口测试简洁化需求

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

发布: 更新时间:2022-11-12 20:21:22

根据需求,产品部分功能采用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”,设置输出路径和依赖库,如下图所示:

备注:选择第一种方式将依赖包打在里面。

将打好的RPC.jar包拷贝到JMeter_Home\lib\ext下,启动jmeter控制台打印该jar信息,如下图所示:

启动完成后添加Java请求,显示读取类名,配置相关参数,如下图所示:

执行结果如下:

之后便可以通过该jar进行RPC相关接口测试,而且保证其他测试人员无需写代码便可进行测试。

标签:, , ,
文章排行