Locust 扩展

Locust扩展

蝗虫带有许多事件，这些事件提供了以不同方式扩展蝗虫的钩子。

事件监听器可以在模块级别上的 locust 文件中注册。

下面是一个例子：

from locust import events

def my_success_handler(request_type, name, response_time, response_length, **kw):
    print "Successfully fetched: %s" % (name)

events.request_success += my_success_handler

强烈建议您在监听器中添加通配符关键字参数（上面代码中的 **kw ），以防止在以后的版本中添加新参数时代码损坏。

要查看所有可用事件，请参阅 事件钩子。

添加网络路由

Locust 使用 Flask 来提供 Web UI ，因此很容易将Web端点添加到Web UI。

只需将 Flask应用程序导入您的 locustfile 并设置一条新路径：

from locust import web

@web.app.route("/added_page")
def my_added_page():
    return "Another page"

现在，您应该可以启动locust ，并且在 http://127.0.0.1:8089/added_pa​​ge 上看到内容。