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

左手的ㄟ右手 2023-09-26 16:28 8阅读 0赞

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

![图片][19959491c9b24c748d678c57559b3b09.png]

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

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

![图片][1aa42ceb936c498e9d42804e7eaa999a.png]

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

![图片][9712959cad2f4d8287e5da9394668c82.png]

执行结果如下：

![在这里插入图片描述][36b58053dd524c00bef3e3661c70cc7f.png]

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

--------------------

### 资源分享 ###

**下方这份完整的软件测试视频学习教程已经上传CSDN官方认证的二维码，朋友们如果需要可以自行免费领取 `【保证100%免费】`**

![在这里插入图片描述][b9a9eea1ed5c42429c5e383b00ba6e0c.png_pic_center]

![在这里插入图片描述][d96b58a511f84668b59f3f620964da07.png_pic_center]

[19959491c9b24c748d678c57559b3b09.png]: https://img-blog.csdnimg.cn/19959491c9b24c748d678c57559b3b09.png
[1aa42ceb936c498e9d42804e7eaa999a.png]: https://img-blog.csdnimg.cn/1aa42ceb936c498e9d42804e7eaa999a.png
[9712959cad2f4d8287e5da9394668c82.png]: https://img-blog.csdnimg.cn/9712959cad2f4d8287e5da9394668c82.png
[36b58053dd524c00bef3e3661c70cc7f.png]: https://img-blog.csdnimg.cn/36b58053dd524c00bef3e3661c70cc7f.png
[b9a9eea1ed5c42429c5e383b00ba6e0c.png_pic_center]: https://img-blog.csdnimg.cn/b9a9eea1ed5c42429c5e383b00ba6e0c.png#pic_center
[d96b58a511f84668b59f3f620964da07.png_pic_center]: https://img-blog.csdnimg.cn/d96b58a511f84668b59f3f620964da07.png#pic_center