使用Jmeter测试thrift接口

2023-06-02,,

术语描述

jmeter:一款性能压力测试工具,支持多种协议,java 、http 等,但是不支持thrift

thrift:跨语言的RPC调用框架,提供编译器,可以将thrift接口生成不同语言的接口服务类

此处以一个Jmeter测试thrift接口的为案例记录下如何做!

案例描述:

一个thrift接口 ,入参String, 返回结果为入参String ,接口逻辑,打印入参内容。

操作步骤概述:

1、准备thrift接口文件HelloWorld.thrift

2、使用thrift.exe  将HelloWorld.thrift 编译生成 java 文件

3、创建maven项目,添加依赖包,添加步骤2生成的java文件

4、创建接口实现类

5、创建thrift server

6、创建thrift client

7、创建Jmeter Java测试类

8、运行thrift server

9、运行thrift client测试

10、打包放到jmeter\lib\ext下面

11、运行Jmeter 创建java测试请求

详细操作步骤:

1、创建thrift接口文件(文件扩展名是“.thrift”)HelloWorld.thrift

namespace java com.joker.thrift   // 定义生成java类文件的包名字,以及语言种类 

//定义入参
struct Request {
1: optional string msg;
} //定义返回参数
struct Response {
1: optional string msg;
} //定义异常
exception OutOfLengthException {
1: optional string reason;
} // 定义接口名称
service HelloWorldService {
Response say(1: Request request) throws (1:OutOfLengthException outOfLengthException); // 可能抛出异常。
}

2、使用thrift.exe  将HelloWorld.thrift 编译生成 java 文件,执行完下面命令会产生java文件 (https://mirror.bit.edu.cn/apache/thrift/0.13.0/thrift-0.13.0.exe)

D:\thrift> .\thrift-0.13.0.exe -r -gen java .\HelloWorld.thrift   // 这是在win下面的命令行执行的执行的,不要复制粘贴就执行,看自己的 thrift-0.13.0.exe 和 HelloWorld.thrift 文件的位置  

3、创建maven项目,添加依赖包,添加步骤2生成的java文件 (不会的请百度)

最终的maven工程的目录结构:(lib包是要自己新建一个lib文件夹,然后将jmeter安装目录下面的lib/ext/拷贝到lib下面来)

<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
<modelVersion>4.0.0</modelVersion> <groupId>com.hhl</groupId>
<artifactId>demo</artifactId>
<version>0.0.1-SNAPSHOT</version>
<packaging>jar</packaging> <name>demo</name>
<url>http://maven.apache.org</url> <properties>
<project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
</properties> <dependencies>
<!-- junit测试依赖,可以不要,创建maven项目自带 -->
<dependency>
<groupId>junit</groupId>
<artifactId>junit</artifactId>
<version>3.8.1</version>
<scope>test</scope>
</dependency> <!-- thrift依赖 -->
<dependency>
<groupId>org.apache.thrift</groupId>
<artifactId>libthrift</artifactId>
<version>0.13.0</version>
</dependency> <!-- 将lib文件夹下面的包添加依赖 -->
<dependency>
<groupId>org.apache</groupId>
<artifactId>jmeter-java</artifactId>
<version>1.0</version>
<scope>system</scope>
<systemPath>${basedir}/lib/ApacheJMeter_java.jar</systemPath>
</dependency> <!-- 将lib文件夹下面的包添加依赖 -->
<dependency>
<groupId>org.apache</groupId>
<artifactId>jmeter-core</artifactId>
<version>1.0</version>
<scope>system</scope>
<systemPath>${basedir}/lib/ApacheJMeter_core.jar</systemPath>
</dependency> </dependencies> <build>
<plugins>
<!-- 指定jdk -->
<plugin>
<groupId>org.apache.maven.plugins</groupId>
<artifactId>maven-compiler-plugin</artifactId>
<configuration>
<source>1.8</source>
<target>1.8</target>
<encoding>UTF-8</encoding>
</configuration>
</plugin>
</plugins>
</build>
</project>

4、创建接口实现类HelloWorldServiceImpl.java

package com.joker.thrift.impl;

import org.apache.thrift.TException;

import com.joker.thrift.HelloWorldService;
import com.joker.thrift.OutOfLengthException;
import com.joker.thrift.Request;
import com.joker.thrift.Response; public class HelloWorldServiceImpl implements HelloWorldService.Iface{ @Override
public Response say(Request request) throws OutOfLengthException, TException { Response response = new Response();
response.setMsg(request.getMsg());
System.out.print("HelloWorldServiceImpl.say = "+request.getMsg());
if(request.getMsg().length() > 20) {
throw new OutOfLengthException();
} return response;
} }

5、创建thrift server

package com.joker.thrift.server;

import java.net.ServerSocket;

import org.apache.thrift.server.TServer;
import org.apache.thrift.server.TSimpleServer;
import org.apache.thrift.transport.TServerSocket; import com.joker.thrift.HelloWorldService;
import com.joker.thrift.HelloWorldService.Processor;
import com.joker.thrift.impl.HelloWorldServiceImpl; public class HelloWorldServer {
public static void main(String[] args) throws Exception{ // Transport
ServerSocket socket = new ServerSocket(8888);
TServerSocket serverTransport = new TServerSocket(socket); // Processor
HelloWorldService.Processor processor = new Processor(new HelloWorldServiceImpl()); TServer.Args tServerArgs = new TServer.Args(serverTransport);
tServerArgs.processor(processor); // Server
TServer server = new TSimpleServer(tServerArgs);
System.out.println("Starting the simple server...");
server.serve(); }
} 

6、创建thrift client

package com.joker.thrift.client;

import org.apache.thrift.protocol.TBinaryProtocol;
import org.apache.thrift.protocol.TProtocol;
import org.apache.thrift.transport.TSocket;
import org.apache.thrift.transport.TTransport; import com.joker.thrift.HelloWorldService;
import com.joker.thrift.Request;
import com.joker.thrift.Response; public class HelloWorldClient {
public static void main(String[] args) {
TTransport transport = null;
try {
// 创建TTransport
transport = new TSocket("localhost", 8888); // 创建TProtocol 协议要与服务端一致
TProtocol protocol = new TBinaryProtocol(transport); // 创建client
HelloWorldService.Client client = new HelloWorldService.Client(protocol); transport.open(); // 建立连接 Request request = new Request(); // 构造请求参数
request.setMsg("Hello world!!"); Response response = client.say(request); // client调用server端方法 System.out.println("result = "+response.msg); } catch (Exception e) {
e.printStackTrace();
} finally {
transport.close(); // 请求结束,断开连接 }
}
}

7、创建Jmeter Java测试类

package com.joker.thrift.jmeter;

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.TException;
import org.apache.thrift.protocol.TBinaryProtocol;
import org.apache.thrift.transport.TSocket;
import org.apache.thrift.transport.TTransport;
import org.apache.thrift.transport.TTransportException; import com.joker.thrift.HelloWorldService;
import com.joker.thrift.Request;
import com.joker.thrift.Response; public class HelloWorldServiceTest extends AbstractJavaSamplerClient{ private TTransport transport;
private HelloWorldService.Client helloWorldService;
private TBinaryProtocol binaryProtcol; @Override
public SampleResult runTest(JavaSamplerContext context) { // 需要关注的是这一块
System.out.println(" runTest start");
Request request = new Request();
request.setMsg(context.getParameter("msg")); SampleResult result = new SampleResult(); // 记录响应结果
result.sampleStart(); // 开始记录
try {
Response response = helloWorldService.say(request); // 调用thrift生成的 service类中的方法
result.setResponseData(response.getMsg().getBytes());
result.setSuccessful(true); // 设置成功
} catch (TException e) {
result.setSuccessful(false); // 设置失败
e.printStackTrace();
}
result.sampleEnd(); // 记录结束 用于统计耗时的
return result;
} @Override
public Arguments getDefaultParameters() { // 这个数获取默认参数的,应该是可以定制化的,在jmeter上面配置
System.out.println(" getDefaultParameters start");
Arguments args = new Arguments();
args.addArgument("msg","LoveHHL");
return args;
} @Override
public void setupTest(JavaSamplerContext context) { // 创建请求 super.setupTest(context);
System.out.println(" setupTest start");
transport = new TSocket("localhost",8888);
binaryProtcol = new TBinaryProtocol(transport);
helloWorldService = new HelloWorldService.Client(binaryProtcol); try {
transport.open();
} catch (TTransportException e) {
transport.close();
e.printStackTrace();
} } @Override
public void teardownTest(JavaSamplerContext context) { // 关闭资源
super.teardownTest(context);
System.out.println(" teardownTest start");
transport.close(); } }

8、运行thrift server

9、运行thrift client测试

查看下thrift server

10、打包放入到jmeter\lib\ext下面(libthrift-0.13.0.jar 和 新建的工程打的包)

mvn clean install  // 将创建的maven工程打成jar包,在项目的根目录下面执行

11、运行Jmeter 创建java测试请求 (thrift server 需要启动--IDE里面的main函数启动)

测试结果

使用Jmeter测试thrift接口的相关教程结束。

《使用Jmeter测试thrift接口.doc》

下载本文的Word格式文档,以方便收藏与打印。