在 Rainbond 上使用 Locust 进行压力测试

2022 年 1 月 18 日
本文字数：1839 字
阅读完需：约 6 分钟

Locust 简介

Locust 是一种易于使用、可编写脚本且可扩展的性能测试工具。并且有一个用户友好的 Web 界面，可以实时显示测试进度。甚至可以在测试运行时更改负载。它也可以在没有 UI 的情况下运行，使其易于用于 CI/CD 测试。

Locust 使运行分布在多台机器上的负载测试变得容易。Locust 基于事件（gevent），因此可以在一台计算机上支持数千个并发用户。与许多其他基于事件的应用程序相比，它不使用回调。相反，它通过 gevent 使用轻量级进程。并发访问站点的每个 Locust（蝗虫）实际上都在其自己的进程中运行（Greenlet）。这使用户可以在 Python 中编写非常有表现力的场景，而不必使用回调或其他机制。

快速部署 Locust

Locust 应用已发布到 开源应用商店，搜索 locust 安装最新2.5.1版本即可。

安装完成后，您将会得到一个 Locust 主从集群，其中，master 组件负责提供 UI 界面，和并发任务的调度；slave 组件负责执行并发任务，slave 组件同时支持横向伸缩，当产生的测试并发达到一定的限额时，只需扩展 slave 组件实例即可，例如：

如何使用

Locust_Master 提供了一个基于 WEB-UI 的图形化管理界面，首次登陆，会提示输入一些信息：

❝
默认用户密码：locust locust，可配置 Locust_Master 组件的环境变量 LOCUST_WEB_AUTH进行修改。
❞

「Number of users」 填写模拟的并发用户数量，经过测试，单个 slave 实例可以轻松提供上千个用户并发的压力。

「Spawn rate」 填写蝗虫的孵化率，即每秒产生多少用户。

「Host」 填写想要压测的站点地址。

当 Host 以及用户，并发量定义完毕以后，还需要去定义一下测试用例，即用户访问 Host 之后的行为，Locust 是通过一个名为/locustfile.py 的 Python 脚本来定义用户行为，在 Rainbond 平台的 Locust_Master 组件内 环境配置 -> 配置文件设置 进行编辑修改。

代码示例如下：

from locust import HttpUser, task, between
class MyUser(HttpUser):    wait_time = between(5, 15)
    @task(2)    def index(self):        self.client.get("/")
    @task(1)    def about(self):        self.client.get("/docs/")

复制代码

这个脚本将按照顺序模仿以下行为：

请求 Host 的 / 路径两次
请求 Host 的 /docs/ 路径一次
每次执行任务之间，间隔 5-15 秒

之所以要这么设计的原因，是 Locust 的设计者们认为，真正的用户行为，不会像脚本一样接连不断的执行完所有的请求然后退出。更多的情况是，用户做完一件事后，会停顿一会，比如读读说明，思考下一步要干嘛。所以会在每个步骤之间留下一个随机时长的空白期。这种假设实际上更符合用户实际行为。

这个文件，将会以配置文件的方式挂载到 locust_master 组件上，并且共享挂载给所有的locust_slave组件。这意味着，如果你想要更改这个文件的内容，只需要去编辑 locust_master 组件中，环境配置下所挂载的配置文件即可。然后更新整个 Locust 集群即可生效。

结果分析

借助 Locust 提供的 WEB-UI 界面，我们可以非常方便的分析压力测试结果。

Statistics 页面，将向我们展示所有被压测接口的汇总报告。结果包括：

「Type」 请求类型；「Name」 请求路径；「Requests」 请求总数；「Fails」 失败次数；「Median」 中位数响应时间；「90%ile」 90%请求响应时间；「Average」 平均响应时间；「Min」 最小响应时间；「Max」 最大响应时间；「Average size」 请求的平均大小；「Current PRS」 当前吞吐率；「Current Failures」 当前错误率；