在Locust子类中创建用户凭据并传递给TaskSet

Question

我希望在locust运行时提供用户的凭据，然后将这些凭据传递给任务。

其思想是在蝗虫产生每个用户(客户端)时提供一个用户，然后在蝗虫停止时取消对用户的供应。

用户可以在后端系统中被供应/取消供应、从文件或数据库表中读取等。

有没有办法做到这一点？

Answer 1

我找到了一个方法来做到这一点。

1. 在TaskSet子类中，实现on_start和on_stop。将实例变量(self.whatever)设置为为“用户”提供的值(例如，用户名和密码)。
2. 使用装饰器方法定义任务。这使得这些方法成为TaskSet子类的一部分，因此您可以使用预置的值引用实例变量。

例如,

class MyTaskSet(TaskSet):
    def on_start(self):
        self.username = <however you want to get it>

    @task(1)
    def MyTask(self):
        <wherever you need it> = self.username