Locust 是一个用于负载测试的 Python 库,它允许您模拟许多用户并发执行各种任务
这是一个示例代码,将只对 your_task
函数发起一次请求:
from locust import HttpUser, task, TaskSet
class MyTaskSet(TaskSet):
@task(1)
def your_task(self):
# 在这里编写您的请求代码
self.client.get("/path/to/your/endpoint")
class MyUser(HttpUser):
tasks = [MyTaskSet]
min_wait = 0
max_wait = 0
这个代码示例创建了一个名为 MyUser
的 Locust 用户,该用户执行名为 MyTaskSet
的任务集。your_task
函数使用 @task(1)
装饰器,并且在 MyTaskSet
里面只执行一次。
为了运行此测试,请在命令行中执行以下命令:
locust -f your_locustfile.py --headless -u 1 -r 1 -t 1
your_locustfile.py
是包含上述代码的 .py
文件的名称。-f
标志指定了要使用的 locustfile。--headless
表示不使用 Web 界面,而是以无头模式运行。-u 1
表示模拟的用户数为 1。-r 1
表示每秒启动一个新用户。-t 1
表示测试持续时间为 1 秒。您可以根据需要调整此值。注意:虽然这个方法可以在 Locust 中定义单次请求,但请记住 Locust 主要用于负载和性能测试,因此建议尽可能模拟实际用户行为以获得更准确的结果。
领取专属 10元无门槛券
手把手带您无忧上云