分布式使用
如果要使用多進程/分布式的形式來進行壓測，啟動時需要增加以下參數：
主：locust -f test.py --master --host=http://example.com
從：locust -f test.py --slave --host=http://example.com
如果想在多臺機器上運行Locust惠呼，在啟動slaves時砸捏，我們應該指定master地址(當運行Locust分布式在同一臺機器時找御，master的默認地址是127.0.0.1):
locust -f test.py --slave --master-host=ip --host=http://example.
locust 測試程序的編寫
以前一篇中的示例程序 test.py 為例
from locust import HttpLocust, TaskSet, task, between
class UserBehavior(TaskSet):
def on_start(self):
""" on_start is called when a Locust start before any task is scheduled """
self.login()
def on_stop(self):
""" on_stop is called when the TaskSet is stopping """
self.logout()
def login(self):
self.client.post("/login", {"username":"ellen_key", "password":"education"})
def logout(self):
self.client.post("/logout", {"username":"ellen_key", "password":"education"})
@task(2)
def index(self):
self.client.get("/")
@task(1)
def profile(self):
self.client.get("/profile")
class WebsiteUser(HttpLocust):
task_set = UserBehavior
wait_time = between(5, 9)
主要是兩個類，HttpLocust 和 TaskSet
Locust 表示每個虛擬的用戶涤伐。
HttpLocust 是繼承自 Locust 類, 且 client 綁定了自己的實現，如果不是http/https協議，我們需要繼承Locust后實現自己的client亚茬。TaskSet 描述了每個locust 實例要運行的任務集，并提供了調度浓恳，包括執(zhí)行順序刹缝、權重等等碗暗。
@task 裝飾器表示權重屬性，即index()的執(zhí)行頻率是profile()的兩倍
自定義客戶端測試非http/https的系統(tǒng)
只需要編寫一個觸發(fā)request_success和request_failure事件的自定義客戶端即可梢夯。
以一個xml-rpc協議為例
import time
import xmlrpclib
from locust import Locust, TaskSet, events, task, between
class XmlRpcClient(xmlrpclib.ServerProxy):
"""
Simple, sample XML RPC client implementation that wraps xmlrpclib.ServerProxy and
fires locust events on request_success and request_failure, so that all requests
gets tracked in locust's statistics.
"""
def __getattr__(self, name):
func = xmlrpclib.ServerProxy.__getattr__(self, name)
def wrapper(*args, **kwargs):
start_time = time.time()
try:
result = func(*args, **kwargs)
except xmlrpclib.Fault as e:
total_time = int((time.time() - start_time) * 1000)
###########################
## 1. 實現 request_failure 事件
###########################
events.request_failure.fire(request_type="xmlrpc", name=name, response_time=total_time, exception=e)
else:
total_time = int((time.time() - start_time) * 1000)
###########################
## 2. 實現 request_success 事件
###########################
events.request_success.fire(request_type="xmlrpc", name=name, response_time=total_time, response_length=0)
# In this example, I've hardcoded response_length=0. If we would want the response length to be
# reported correctly in the statistics, we would probably need to hook in at a lower level
return wrapper
class XmlRpcLocust(Locust):
"""
This is the abstract Locust class which should be subclassed. It provides an XML-RPC client
that can be used to make XML-RPC requests that will be tracked in Locust's statistics.
"""
def __init__(self, *args, **kwargs):
super(XmlRpcLocust, self).__init__(*args, **kwargs)
self.client = XmlRpcClient(self.host)
class ApiUser(XmlRpcLocust):
host = "http://127.0.0.1:8877/"
wait_time = between(0.1, 1)
class task_set(TaskSet):
@task(10)
def get_time(self):
self.client.get_time()
@task(5)
def get_random_number(self):
self.client.get_random_number(0, 100)
在上面的例子中言疗，關鍵之處，即自定義客戶端的實現中颂砸，當調用完成時噪奄，通過 events.request_failure 和 events.request_success 事件進行數據上報，從何locust可以完成數據統(tǒng)計人乓。
如果需要運行起來以上例子勤篮，可以查看官方文檔獲得svr端代碼。
