如何将蝗虫的运行限制为最大数量？

Question

我正在运行一些蝗虫测试，例如想精确地击中API端点1000次。除了看到UI中的测试达到1000时要尽早停止测试，我看不到其他方法。

我看到一个选项，它指定了运行测试的最长时间，但没有最大请求数。

from locust import HttpLocust, TaskSet, task
import json

class UserBehavior(TaskSet):
    MAX_REQUESTS_PER_USER = 10
    COUNT = 0
    @task(1)
    def some_route(self):
      self.client.get("/some-route", headers=headers, name="some route")
      if (self.COUNT >= self.MAX_REQUESTS_PER_USER):
        sys.exit()

class WebsiteUser(HttpLocust):
    task_set = UserBehavior

Answer 1

由于您尚未发布任何代码，所以我只能给您一个通用的答案：在脚本中添加一个计数器，并在每次调用break达到1000时将其递增。

如果您正在寻找内置函数，请阅读软件包的文档，如果不存在此函数，请使用上述方法。据我了解，没有限制请求数量的功能。