如何在蝗虫中为每个用户类分配固定数量的用户

Question

假设我有3个独立的用户类。我想为每个类分配固定数量的用户。我的代码如下。

class User_1(TaskSet):
    # I need 3 users to execute the tasks within this user class

class User_2(TaskSet):
    # I need only 1 user to execute the tasks within this user class

class User_3(TaskSet):
    # I need only 1 user to execute the tasks within this user class


class API_User_Test(HttpUser):
    #I already tried weighting the classes as below.
    tasks = {Site_User_1: 3, User_2: 1, User_3: 1}

我已经尝试过对这些类进行加权，如上面的代码所示。但它不起作用。有时，它会为类用户\u 2或类用户\u 3分配多个用户。有人能告诉我如何解决这个问题吗。

Answer 1

蝗虫的权重只是一个统计权重，不能保证。权重确定任务/用户被放入列表中的次数，以便从中进行选择。当新任务/用户产生时，蝗虫会从列表中随机选择一个任务。考虑到你的体重：

tasks = {Site_User_1: 3, User_2: 1, User_3: 1}

从统计学上讲，产生5个权重为3/1/1的用户会得到3/1/1，但可能不是每次都那么精确。虽然可能性较小，但你可能会得到4/0/1或3/2/0或5/0/0。

从蝗虫文件：

如果将“任务”属性指定为列表，则每次执行任务时，都将从“任务”属性中随机选择该任务。但是，如果任务是一个dict（可调用项作为键，整数作为值），则将随机选择要执行的任务，但整数作为比率。因此，对于这样一个任务：

{my_task： 3，another_task： 1}

my_task被处决的可能性是another_task的3倍。

在内部，上述dict实际上将扩展为一个列表（任务属性将更新），如下所示：

[my_taskmy_taskmy_taskanother_task]

然后使用Python的random.choice（）从列表中选择任务。

如果您必须完全控制用户正在运行什么，我可能会建议让一个蝗虫用户执行一个任务，该任务包含您自己关于运行什么的逻辑。创建您自己的函数列表，以便在每次创建新用户时调用和迭代它。可能必须是用户外部的全局或其他内容。但想法是你自己管理逻辑，而不是蝗虫。

编辑：如果在多个工作进程上运行，则使用单用户方法控制正在运行的内容将无法正常工作，因为这些工作进程彼此不通信。你可以考虑做一些更高级的事情，比如在主机和工人之间发送消息来协调，或者使用一个外部源，比如数据库或其他服务人员，以了解他们应该运行什么。