Python locust工具使用詳解

	
pip install locust

from locust import HttpLocust
from locust import HttpUser
from locust import task
from locust import TaskSet
#指定一個任務集
class My_task_set(TaskSet):
 
 
    #這是某個任務,30是比例，比如這裡是30/50
    @task(30)
    def getindex1(self):
        # client就是個requests對象
        # catch_response，告訴locust如何判斷請求失敗還是成功
        res  = self.client.get("/bainianminguo/p/10952586.html")
 
    @task(20)
    def getindex2(self):
        # client就是個requests對象
        res = self.client.get("/bainianminguo/p/7253930.html")
 
 
class WebSite(HttpUser):
    # 指定要執行哪個任務集
    tasks = [My_task_set,]
    # 請求和請求之間最小的間隔時間
    min_wait = 1000
    # 請求和請求之間最大的間隔時間
    max_waif = 2000

class My_task_set(TaskSet):
 
    #添加初始化方法
    def on_start(self):
        print("類似類中的構造方法，每個用戶在任務開始前，隻執行一次,在這裡可以定義一個對象的屬性，這樣其它測試集就可以使用這個屬性")

def on_stop(self):
    print("類似類中的後置方法，每個用戶在任務開始後，隻執行一次,在這裡可以定義一個對象的屬性，這樣其它測試集就可以使用這個屬性")

from locust import HttpLocust
from locust import HttpUser
from locust import task
from locust import TaskSet
 
# locust中的client會自動幫我們處理cookies。類似request.session(),所以如果我們登陸的時候，隻需要在on_start中登陸一次就可以瞭
 
# 如果在locust中，如果url是不需要統計，則我們不要用clent去訪問api，應該用request去訪問，這樣就locust就不會統計request庫發起的請請求
#指定一個任務集
class My_task_set(TaskSet):
 
    #添加初始化方法
    def on_start(self):
        print("類似類中的構造方法，每個用戶在任務開始前，隻執行一次,在這裡可以定義一個對象的屬性，這樣其它測試集就可以使用這個屬性")
 
    def on_stop(self):
        print("類似類中的後置方法，每個用戶在任務開始後，隻執行一次,在這裡可以定義一個對象的屬性，這樣其它測試集就可以使用這個屬性")
 
    #這是某個任務,30是比例，比如這裡是30/50
    @task(30)
    def getindex1(self):
        # client就是個requests對象
        # catch_response，告訴locust如何判斷請求失敗還是成功
        res  = self.client.get("/bainianminguo/p/10952586.html",catch_response=True)
        if res.code == 200:
            res.success()
        else:
            res.failure("ff")
        print(res)
    @task(20)
    def getindex2(self):
        # client就是個requests對象
        res = self.client.get("/bainianminguo/p/7253930.html")
        print(res)
 
class WebSite(HttpUser):
    # 指定要執行哪個任務集
    # task_set = My_task_set
 
    tasks = [My_task_set,]
    # 請求和請求之間最小的間隔時間
    min_wait = 1000
    # 請求和請求之間最大的間隔時間
    max_waif = 2000
 
 
# Number of total users to simulate   模擬的用戶數
# Spawn rate                          每秒鐘產生的用戶數

一旦單臺機器不夠模擬足夠多的用戶時，Locust支持運行在多臺機器中進行壓力測試。
 
為瞭實現這個，你應該在 master 模式中使用--master標記來啟用一個 Locust 實例。這個實例將會運行你啟動測試的 Locust 交互網站並查看實時統計數據。<br>master 節點的機器自身不會模擬任何用戶。相反，你必須使用 --slave 標記啟動一臺到多臺 Locustslave 機器節點，與標記 --master-host 一起使用(指出master機器的IP/hostname)。
 
常用的做法是在一臺獨立的機器中運行master，在slave機器中每個處理器內核運行一個slave實例。
 
註意：master 和每一臺 slave 機器，在運行分佈式測試時都必須要有 locust 的測試文件。

在 master 模式下啟動 Locust:
 
locust -f my_loucstfile.py --master
 
在每個 slave 中執行(192.168.0.14 替換為你 msater 的IP):
 
locust -f my_locustfile.py --slave --master-host=192.168.0.14