Python压力测试工具——Locust

最新推荐文章于 2025-02-24 10:56:47 发布

XerCis

最新推荐文章于 2025-02-24 10:56:47 发布

阅读量6.6k

点赞数 4

分类专栏： Python 文章标签： python 压力测试 locust

本文链接：https://blog.csdn.net/lly1122334/article/details/108220006

版权

Python 专栏收录该内容

529 篇文章

订阅专栏

本文介绍了一款名为Locust的开源性能测试工具，其特点包括代码定义用户行为、分布式和可扩展性，以及经过大规模应用验证的稳定性。文章详细讲解了Locust的安装过程、示例应用的创建步骤，以及在实际操作中可能遇到的问题和解决方案。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

简介

Locust一款开源性能测试工具，易于使用、脚本化、可扩展，对开发者非常友好。

特性：

代码定义用户行为
不用笨重的UI或膨胀的XML，只要简单的代码。
分布式和可扩展
Locust支持在多台机器上运行分布式的负载测试，可以用来模拟数百万个并发用户。
深受考验
Locust已被用来模拟数百万个并发用户。Battlelog是一个通过网页发送游戏连接请求的应用，它使用Locust进行了负载测试，所以说Locust是Battletested （久经沙场）。

在这里插入图片描述

安装

pip install locust

初试

app.py

import json
import tornado.web
import tornado.ioloop

LOGIN = False  # 是否登录


def fib(n):
    """计算斐波那契数列的第n项"""
    if n < 2:
        return n
    return fib(n - 1) + fib(n - 2)


class LoginHandler(tornado.web.RequestHandler):
    def post(self):
        body = self.request.body
        body = json.loads(body)
        username = body.get("username")
        password = body.get("password")
        if username == "foo" and password == "bar":
            global LOGIN
            LOGIN = True
            self.write("Welcome!")
        else:
            self.set_status(400)


class IndexHandler(tornado.web.RequestHandler):
    def get(self):
        if LOGIN:
            self.write("Hello World!")
        else:
            self.set_status(400)


class ItemHandler(tornado.web.RequestHandler):
    def get(self):
        id = self.get_argument("id")
        id = int(id)
        self.write(str(fib(id)))


if __name__ == "__main__":
    print("http://localhost:8888")
    app = tornado.web.Application([
        (r"/login", LoginHandler),
        (r"/hello", IndexHandler),
        (r"/world", IndexHandler),
        (r"/item", ItemHandler),
    ])
    app.listen(8888)
    tornado.ioloop.IOLoop.instance().start()

locustfile.py

import time
from locust import HttpUser, task, between


class QuickstartUser(HttpUser):
    host = "http://localhost:8888"  # 访问的站点，最后不要加斜杠
    wait_time = between(1, 2)

    @task
    def index_page(self):
        self.client.get("/hello")
        self.client.get("/world")

    @task(3)
    def view_item(self):
        for item_id in range(10):
            self.client.get(f"/item?id={item_id}", name="/item")
            time.sleep(1)

    def on_start(self):
        self.client.post("/login", json={"username": "foo", "password": "bar"})