服务器web性能测试之Locust

歪小Y_02 · 发表于 2019-3-22 16:23:59

简介

Locust 是一个开源负载测试工具。使用 Python 代码定义用户行为，也可以仿真百万个用户。

Locust 是非常简单易用，分布式，用户负载测试工具。Locust主要为网站或者其他系统进行负载测试，能测试出一个系统可以并发处理多少用户。

Locust 是完全基于时间的，因此单个机器支持几千个并发用户。相比其他许多事件驱动的应用，Locust 不使用回调，而是使用对协程支持比较完善的gevent（基于IO切换）。

特点

完全使用纯Python代码编写用户测试场景，不需要任何配置文件。
分布式&可伸缩 - 支持成千上万的用户
基于webui（自带）
可以测试任意系统
安装

pip install locustio
或者
easy_install locustio

复制代码

检查是否安装成功；可以通过“locust --help”来检测。

简单示例

from locust import web, HttpLocust, TaskSet, events,task
import random, traceback
# 增加新的web界面
@web.app.route("/hi")
def my_added_page():
return "hello locust..."
# 钩子函数
def on_request_success(request_type, name, response_time, response_length):
print 'Type: %s, Name: %s, Time: %fms, Response Length: %d' % \
(request_type, name, response_time, response_length)
def on_request_failure(request_type, name, response_time, exception):
print 'Type: %s, Name: %s, Time: %fms, Reason: %r' % \
(request_type, name, response_time, exception)
events.request_success += on_request_success
events.request_failure += on_request_failure
class UserBehavior(TaskSet):
def on_start(self):
""" on_start is called when a Locust start before any task is scheduled """
self.login()
def login(self):
with self.client.post("/login", {"username":"ellen_key", "password":"education"}, catch_response = True) as r:
if random.choice([0, 1]):
r.success()
else:
r.failure('0')
@task(2)
def index(self):
self.client.get("/")
@task(1)
def profile(self):
self.client.get("/profile")
class WebsiteUser(HttpLocust):
# 指向一个TaskSet类，TaskSet类定义了每个用户的行为
task_set = UserBehavior
# 请求等待最小时间min_wait和请求等待最大时间max_wait
min_wait = 5000
max_wait = 9000
host = http://127.0.0.