Rpc接口压测

前言

现今有比较多的rpc框架应用于实际的生产中，像比较流行的Dubbo、Motan、 Thrift、Grpc等。今天作者将以最近项目中用到的grpc为例，结合jmeter来介绍下rpc压测实施步骤。学习本文前需对rpc框架、jmeter有个大致的了解，知道rpc如何用工具生成各种语言的代码。Grpc本身是支持很多种语言的，而jmeter本身只支持java语言，因此我们的脚本也需要选用java。

步骤一：rpc脚本准备

先来看看我本地的项目目录，对结果有个大致的了解，我的工程里包含多个微服务(gnid、hdr等)的代码，每个微服务我建了一个包。这里只以gnid为例来说明：

1.IDE新建maven项目，修改pom.xml，加入grpc、jmeter的依赖包，以及grpc的代码生成插件

848×741 93.2 KB

1. 在src/main/java下新建目录（如命名proto），将项目的.proto文件copy进去，如图1中的gnid.proto

2. 执行”maven install”,会生成相应的通信及服务代码

1059×716 78.1 KB

4.将图3中生成的代码copy到图1中的建的gnid包中去

1. 编写rpc的client端的代码，如下图。在构造方法中初始化与server端的连接（这里的写法都是固定的）。需要说明的是，接口多个参数赋值，可按照下图中2的方法进行（对check和sign两个参数赋值）

1080×711 113 KB

步骤二：java sample 脚本准备

1. 编写jmeter的java sample代码。

关于jmeter的java sample脚本这里需要说明下：

创建一个类并实现JavaSamplerClient接口或继承bstractJavaSamplerClient，并重写如下几个方法：

public Arguments getDefaultParameters()：//设置可用参数的默认值；
public void setupTest(JavaSamplerContext arg0)：//每个线程测试前执行一次，做一些初始化工作；
public SampleResult runTest(JavaSamplerContext arg0)：//开始测试，从arg0参数可以获得参数值；
public void teardownTest(JavaSamplerContext arg0)：//测试结束时调用；

1018×764 104 KB

953×790 101 KB

至此，脚本就准备好了。关于脚本还有如下几个点需要注意的：

a. rpc建立的是长连接，初始化服务端连接的代码应放在“setupTest”方法中，这样单个用户在执行期间都是复用该连接（实际生产中也是这么用的，建立连接的过程还是挺耗时的）。同样， shutdown请求，放在teardownTest方法中即可

b. 通常一个rpc服务会包含多个接口，为了避免每个接口都写一个java sample请求，这里有个小技巧，可以在参数中增加一个字段，用于区分不同的接口

813×712 94.8 KB

步骤三：将脚本打成可执行包，放到jmeter的\lib\ext\目录下

步骤四：启动jmeter，新建“线程组”，在线程组下新建java请求

1080×594 67.4 KB

选择测试类

1080×522 37 KB

填写在代码中设定的参数：

1080×413 66.4 KB

剩下的增加相应的断言、监听器、参数化（如需要），就可以像玩http一样开始压测了。

总结

本文重点介绍了两个方面的知识点。其一，rpc压测脚本如何写；其二，jmeter如何执行java sample请求。有问题的小伙伴，欢迎留言区交流~~