Locust-Web压测神器(一)
Locust-Web压测神器
Locust 完全基本 Python 编程语言,采用 Pure Python 描述测试脚本,并且 HTTP 请求完全基于 Requests 库。除了 HTTP/HTTPS 协议,Locust 也可以测试其它协议的系统,只需要采用Python调用对应的库进行请求描述即可。
LoadRunner 和 Jmeter 这类采用进程和线程的测试工具,都很难在单机上模拟出较高的并发压力。Locust 的并发机制摒弃了进程和线程,采用协程(gevent)的机制。协程避免了系统级资源调度,由此可以大幅提高单机的并发能力
windows安装
pip install locustio
不逼逼上代码
# coding=utf-8
from locust import HttpLocust, TaskSet, task
import json
class OffcalWebSite(TaskSet):
# 访问官网
@task(3)
def login(self):
# 定义请求头
header = {
"Content-Type": "application/json; charset=UTF-8",
"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/74.0.3729.169 Safari/537.36",
"X-Requested-With":"XMLHttpRequest"
}
# 嘿嘿自己换成想要测试的数据
datas = {"mobile": "xx", "verify_code": "111111", "login_type": 2, "device_code": 2577347094}
req = self.client.post("/v2/auth/login", data=json.dumps(datas), headers=header)
if req.status_code == 200:
print("SUCCESS")
else:
print("Fail")
class WebsiteUser(HttpLocust):
task_set = OffcalWebSite
min_wait = 3000 # 单位为毫秒
max_wait = 6000 # 单位为毫秒
# 也可以自定义哈,这样写
wait_function = lambda self: random.expovariate(1) * 1000 #random.expovariate(lambda) 指数分布
if __name__ == "__main__":
import os
os.system("locust -f locustfile.py --host=http://xxx.com")
脚本简述:新建一个类OffcalWebSite(TaskSet),继承TaskSet,该类下面写需要请求的接口以及相关信息;
self.client调用get和post方法,和requests一样;
@task装饰该方法表示为用户行为,括号里面参数表示该行为的执行权重:数值越大,执行频率越高,不设置默认是1;
属性 | 说明 |
---|---|
task_set | 指向定义了用户行为的类 |
min_wait | 模拟负载的任务之间执行时的最小等待时间,单位为毫秒 |
max_wait | 模拟负载的任务之间执行时的最大等待时间,单位为毫秒 |
PS:默认情况下,时间是在min_wait和max_wait之间随机选择,但是可以通过将wait_function设置为任意函数来使用任何用户定义的时间分布。
启动
1、如果启动的locust文件名为locustfile.py并位于当前工作目录中,可以在编译器中直接运行该文件,或者通过cmd,执行如下命令:
locust --host=http://offical.test.langpedu.com
2、如果Locust文件位于子目录下且名称不是locustfile.py,可以使用-f命令启动上面的示例locust文件:
locust -f E:\Locust\locustfile.py --host=http://offical.test.langpedu.com
3、如果要运行分布在多个进程中的Locust,通过指定-master以下内容来启动主进程 :
locust -f E:\Locust\locustfile.py --master --host=http://offical.test.langpedu.com
4、如果要启动任意数量的从属进程,可以通过-salve命令来启动locust文件:
locust -f E:\Locust\locustfile.py --salve --host=http://offical.test.langpedu.com
5、如果要运行分布式Locust,必须在启动从机时指定主机(运行分布在单台机器上的Locust时不需要这样做,因为主机默认为127.0.0.1):
locust -f E:\Locust\locustfile.py --slave --master-host=192.168.20.145 --host=http://offical.test.langpedu.com
6、启动locust文件成功后,编译器控制台会显示如下信息:
[2019-08-08 21:18:55,819] jarstick/INFO/locust.main: Starting web monitor at *:8089
[2019-08-08 21:18:55,819] jarstick/INFO/locust.main: Starting Locust 0.11.0
PS:8089是该服务启动的端口号,如果是本地启动,可以直接在浏览器输入http://localhost:8089打开UI界面,如果是其他机器搭建locust服务,则输入该机器的IP+端口即可;
Locust的UI页面
说一下这2个输入框的意思:
Number of users to simulate:设置模拟的用户总数
Hatch rate (users spawned/second):每秒启动的虚拟用户数
Start swarming:执行locust脚本
后续会加上influxdb时序性数据库和grafana可视化仪表监控压测平台