前言

在一个网站中，很多数据需要即时更新，比如期货交易类的用户资产。在、以前，这种功能的实现一般使用http轮询，即客户端用定时任务每隔一段时间向服务器发送查询请求来获取最新值。这种方式的弊端显而易见：

• 有可能造成数据更新不及时，如果前端轮询频率为5s，也许数据在这5s内已经更新多次了。
• 有可能对数据库造成额外压力，例如一个用户资产长时间不变化，但客户端还是要定时去查询，这种无意义查询占比相当高，对服务器造成不必要的压力。
• 要经过请求和响应两次交互，增加了耗时，而且http请求可能携带大量的header信息，增加网络带宽占用

HTML5 开始提供的一种在单个 TCP 连接上进行全双工通讯的协议-WebSocket，很好地解决了http轮询的弊端。

在 WebSocket API 中，浏览器和服务器只需要做一个握手的动作，然后，浏览器和服务器之间就形成了一条快速通道。两者之间就直接可以数据互相传送。

也就是说，http轮询机制，主动权完全在客户端，而WebSocket机制中，主动权可以交给服务端，数据推送可以更精确，包括何时推送（定时推送还是更新即推送），推送什么数据。

准备工作

JMeter可以非常便利地进行WebSocket接口测试，但需要引入下列依赖：

• jetty-http
• jetty-io
• jetty-util
• websocket-api
• websocket-client
• websweocket-common

相关依赖下载：

https://pan.baidu.com/s/1PTOyTBzmOwLPNhxB-TxR7g ，提取码：uq25

下面内容基于JMeter5.1.1

将相关jar包放入JMeter安装目录的/lib/etc中，重启JMeter。在取样器中，可以看到比之前多了websweocket相关的取样器。

脚本编写

在编写脚本之前，先要搞清楚推送服务的逻辑，它的逻辑是这样的：

• 首先，客户端向服务端发送请求，建立连接
• 建立连接后，客户端需要定时向服务端发送ping-pong消息，维持心跳
• 客户端发送主动断开连接的请求，服务端断开该连接

建立连接，使用【WebSocket request-response Sampler】，顾名思义，这个取样器既能发送请求也能接收响应。

说明一下各项内容：

• Connection：有use existing connection和setup new connection两种模式，前者是使用已有连接，即上一个websocket请求所建立的连接通道，选择后Server URL全置灰只读不可操作。后者指新建连接通道。
• Server URL：ws协议和wss（加密的websocket）可选，sever name or IP（服务器地址）、Port（端口号）、Path（路径）、Connection timeout（连接超时时间）这些含义也很明了。
• Data：发送数据，可以选择Text（文本，包括JSON）和Binary（二进制）形式，也可以通过勾选Read request data from file来从文件中获取data。

至此，与服务器的连接已建立。下一步就是实现心跳检测，仍然是【WebSocket request-response Sampler】，但Request Data不同（data具体是什么要看服务端的实现）：

这个请求要与连接请求是同一个线程，并且要定时运行，因此设计脚本结构如下：

线程组设置3000并发，但每个线程只运行一次，这样就实现了3000个线程与服务端建立连接。在每个迭代中，依赖循环控制器和固定定时器实现间隔30s发送一次ping-pong请求