Locust 压测介绍和使用

背景：随着公司对项目质量越来越看重，性能测试已经慢慢日常化，不同之前性能测试在高峰之前做，所以需要一个可以随时对某些场景接口进行压测的实现方法。
目前市场使用的压测工具，LoadRunner，Jmeter等，这些个人觉得不太轻量化，而且要专门写脚本。
我要介绍的是Locust压测，不仅快速可便捷的去实现压测，而且具有可编程性，可复用或使用接口自动化的case接口代码。
这篇文章先介绍Locust基本信息和入门使用，后面文章在详细深入介绍高级用法。

一、Locust介绍

Locust官网：https://www.locust.io/
Locust是纯运行在python语言下的测试工具，采用了gevent开发框架，底层利用greenlet微线程机制，单机并发可以支持非常高，直接HTTP等协议接口，特点就是协程，web可视化管理工具。
Locust有六个依赖的python模块：
1.gevent：在Python中实现协程的第三方库，协程又叫微线程Corouine，使用gevent可以获取极高的并发能力；
2.flask：Python的一个web开发框架；
3.requests：支持http/https访问的库；
4.msgpack-python：一种快速、紧凑的二进制序列化格式，使用与类似json的数据；
5.six：提供了一些简单的工具封装Python2和Python3 之间的差异；
6.pyzmq：安装这个第三方库，可以把Locust运行在多个进程或多个机器（分布式）。
由上面这几个库的功能，组成了整个可实现压测的工具。

二、Locust安装

1.在线（推荐）：pip install Locust

image.png

安装完后：locust -h 显示如下表示安装成功。

image.png

2.离线：https://github.com/locustio/locust

三、Locust使用

1、代码：

from locust import HttpUser, TaskSet, task
import json

class Testlocust(TaskSet):
    @task(1)
    def test_queryMessage(self):
        post_url = '/api/messageCenterController/queryMessage?pageNo=1&pageSize=10&titleName=&releaseDate='
        header = {
            'Accept-Encoding': 'gzip, deflate',
            'Content-Type': 'application/json;charset=UTF-8',
            "Cookie": ""
        }
        r = self.client.get(post_url, headers=header)
        print(r)
        assert r.succ == "ok"

    @task(3)
    def test_queryTotalFeedbackByPage(self):
        post_url = '/api/webFeedback/queryTotalFeedbackByPage'
        header = {
            'Accept-Encoding': 'gzip, deflate',
            'Content-Type': 'application/json;charset=UTF-8',
            "Cookie": ""
        }
        payload = {"type":"","status":"","moduleName":"","empInfo":"","contentKey":"","loginId":"","isReaded":""}
        s = json.dumps(payload)
        r = self.client.post(post_url, headers=header,data=s)
        print(r)
        assert r.totalNum == "8313"

class WebsiteUser(HttpUser):
    tasks = [Testlocust]
    min_wait = 500
    max_wait = 5000

##下面这些可以不用写
if __name__ == '__main__':
    import os
    os.system('locust -f locust_test1.py --web-host="http://*****.com"')

其中：1.@task(3)单次执行的次数，可以理解是循环次数，数值越大执行的次数越多。不写默认1次。
2.接口请求需要使用locust自带的接口请求：

self.client.get()

3.其他代码上特点和注意点，网上可以搜下，就不详细介绍了。

2、启动和执行
1）启动
可以在命令行启动，cmd命令框，进入此文件的目录输入：
locust -f locust_test.py --host="http://*****.com"
-f： 指定性能测试脚本文件的绝对路径。
–host： 指定被测试应用的URL的地址，就是测试项目的host地址。
如下图：

image.png

//分布式启动，下个版本介绍。
2）执行
在浏览器输入：http://localhost:8089/ 打开如下界面

image.png

Number of users：设置模拟用户数。（并发）
Spawn rate ：每秒产生（启动）的虚拟用户数。（递增数）
Host：host地址。//可在WebsiteUser类下直接写host = "www.*.com"，启动命令不加host参数，这里就不会显示，则默认走代码写死的host地址。
点击 “Start swarming” 按钮，立即开始运行性能测试。

image.png

不支持设置压测时间，所以自己看时间需要手动点击“STOP”红色按钮停止。

3.查看报告
1）Statistics：默认概括压测数据。
Type 访问类型
Name 任务名（python中定义的方法名）
Requests 请求的总次数
Fails 失败的次数
Median (ms) 中间数耗时
Average (ms) 平均耗时
Min (ms) 最低耗时
Max (ms) 最大耗时
Average size (bytes) 平均耗时
Current RPS 每秒钟处理的访问的次数
查看这个Current RPS就代表接口的TPS。
2）Charts：趋势图

image.png

主要就是查看这两类的数据。

上面这些就是locust最基本的介绍，安装和使用方法。后面在详细介绍locust的高级用法和扩展。