Locust 向其他用户界面报告 /stats/requests 数据

Question

我发现 Locust UI 会从 http://my-locust-server/stats/requests 端点获取数据，并绘制图表。例如Total request per Second、Response Time 和 Number of Users。

我想将/stats/requests 数据发送到另一台服务器以进行绘图。如何在locustfile.py 中做到这一点？

以下代码不起作用，仅用于演示目的。

from locust import HttpUser, task, events
import requests

class User(HttpUser):

    @task
    def scenario_01(self):
        self.client.get('/some_endpoint')

# ==========
# following codes won't work, will block the whole locust.
# ==========
import time
starttime = time.time()
while True:
    r = requests.get('http://my-locust-server/stats/requests')
    payload = r.json()
    requests.post('http://another-server-for-plotting', json=payload)
    time.sleep(5.0 - ((time.time() - starttime) % 5.0))

现在我能想到的是运行另一个程序来获取和报告数据。但是如何在一个locustfile.py 中做到这一点。

谢谢！

Answer 1

导入此模块后，while True 代码将被评估并永远运行。如果您希望该循环在后台运行，您可以使用 python threading 库。将该 while 循环放在一个可调用函数中，然后将该函数传递给一个新的 Thread 对象并在导入时启动该线程。

from locust import HttpUser, task, events
import requests

class User(HttpUser):

    @task
    def scenario_01(self):
        self.client.get('/some_endpoint')

def backgroundloop():
    while True:
        import time
        starttime = time.time()
        r = requests.get('http://my-locust-server/stats/requests')
        payload = r.json()
        requests.post('http://another-server-for-plotting', json=payload)
        time.sleep(5.0 - ((time.time() - starttime) % 5.0))

import threading
t = threading.Thread(target=backgroundloop)
t.start()

Answer 2

或者，您可以使用 Locust 的 Event Hooks 直接发送数据，而无需从网络路由中获取数据。如果您在分布式模式下运行，worker_report 可能是最好的。不完整的例子：

from locust import events
import requests


@events.worker_report.add_listener
def send_data_off_somewhere(client_id, data):
    requests.post('http://another-server-for-plotting', json=data)