服务器web性能测试之Locust

最近做系统性能调优，发现了一款比较好用的工具Locust。

简介

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

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

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

特点

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

安装

pip install locustio

或者

easy_install locustio

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

文档说明

官方文档：http://docs.locust.io/en/latest/installation.html

简单示例

test.py

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.1

单机

终端输入：

locust -f test.py

webui访问地址：localhost:8089

ui界面需要填写两个参数：

number of users to simulate：模拟用户的数量

Hatch rate (users spawned/second):表示产生模拟用户的速度，每秒启动多少个用户

分布式

master机终端输入：

locust -f test.py --master --master-port=8888(默认的是8080端口)

salve机终端输入：

locust -f test.py --slave --master-host=<master-IP> --master-bind-host=8888

博客已迁移，欢迎关注 最新博客