蝗虫作为库运行时 grpc 客户端不输出统计信息

Question

我们正在尝试添加一个 locust grpc 客户端并从示例目录 here 获取代码。

我们的用例是 locust 将用作库而不是通过 UI。

我们在 csv 文件或代码中看到的统计信息输出的问题。我在 locust doc here 中看到的代码和我们的代码之间的唯一区别是服务是独立启动的。

此问题已在此 GitHub 存储库中重现：https://github.com/saurabhsharma721/grpc_locust_stats_repro_prj

此问题仅在库模式下观察到。如果我们通过 ui 运行它，则生成的统计信息是正确的。请帮忙。

Answer 1

我不确定，但你试过添加这个吗？

# start a greenlet that save current stats to history
gevent.spawn(stats_history, env.runner)

（来自文档https://docs.locust.io/en/stable/use-as-lib.html）

Answer 2

我可以使用以下代码解决这个问题：

添加了两个静态方法在初始化和拆除时调用

 @staticmethod
    def on_init(environment, **kwargs) -> None:  # type: ignore[no-untyped-def]
        print("Perform global setup to create a global state in LocustRunner")

 @staticmethod
    def on_quit(environment, **kwargs) -> None:  # type: ignore[no-untyped-def]
        print('Perform global teardown to clear the global state in LocustRunner')

然后在驱动中添加这段代码

events.init.add_listener(LocustRunner.on_init)
env = Environment(user_classes=[TestClass], events=events)
# Using local runner in start
runner = env.create_local_runner()
# init event
env.events.init.fire(environment=env, runner=runner)
Driver code 
# quit event
env.events.quitting.fire(environment=env, reverse=True)

已经有一个堆栈溢出答案，但目前无法搜索