Method _user_gen

locust/dispatch.py:370–388

(self)

        return users_on_workers, user_gen, worker_gen, active_users

    def _user_gen(self) -> Iterator[str | None]:
        weighted_users_gen = _kl_generator((u.__name__, u.weight) for u in self._user_classes if not u.fixed_count)

        while True:
            if self._try_dispatch_fixed:  # Fixed_count users are spawned before weight users.
                # Some peoples treat this implementation detail as a feature.
                self._try_dispatch_fixed = False
                fixed_users_missing = [
                    (u.__name__, miss)
                    for u in self._user_classes
                    if u.fixed_count and (miss := u.fixed_count - self._get_user_current_count(u.__name__)) > 0
                ]
                total_miss = sum(miss for _, miss in fixed_users_missing)
                fixed_users_gen = _kl_generator(fixed_users_missing)  # type: ignore[arg-type]
                # https://mypy.readthedocs.io/en/stable/common_issues.html#variance
                for _ in range(total_miss):
                    yield next(fixed_users_gen)
            else:
                yield next(weighted_users_gen)

    @staticmethod
    def _fast_users_on_workers_copy(users_on_workers: dict[str, dict[str, int]]) -> dict[str, dict[str, int]]: