python - 如何为动态 URL 列表创建 Locust 任务？

Question

我正在尝试获取基于 OpenAPI JSON 的所有我的 API 端点的计时信息，因此我想在 setup 方法中执行类似的操作:

for api_path in self.client.get('/api/doc/?format=openapi').json()['paths'].keys():
    self.get_[api_path] = [@task decorated method which retrieves the API path]

或者有其他方法在运行时生成任务吗？

Answer 1

在 API docs 中找到它:

def setup(self) -> None:
    for api_path in self.client.get('/api/doc/?format=openapi').json()['paths'].keys():
        self.schedule_task(self.api_task, args=[api_path])

def api_task(self, path: str) -> None:
    response = self.client.get("/api{}?limit=1000".format(path))
    assert response.status_code == HTTPStatus.OK, response.content.decode()