Rpc interface stress test

前言

现今有比较多的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请求.有问题的小伙伴,欢迎留言区交流~~