在每个任务之前进行一些设置的蝗虫方法

Question

Locust 中是否有某种方法（理想情况下适用于 User 类）或事件或其他方式在每个任务之前执行一些设置代码？

---

我所调查的：

• 有一些有用的事件，不幸的是没有像on_task_start
• 有一些有用的方法，例如test_start，但不幸的是没有像task_start 这样的方法
• 我可以在 User 类上使用 wait_time() 方法，但这是一个丑陋的解决方法，并且在每个 User 实例的第一个任务之前不会调用该方法。

Answer 1

您可以向 locust 添加自定义事件，请参阅： https://docs.locust.io/en/stable/api.html#EventHook%20class

更新：

class DbTaskSet(TaskSet):
    def __init__(self, parent):
        super().__init__(parent)

    def execute_next_task(self):
        myevent.fire()
        super().execute_next_task()