Python locust工具使用詳解

	
pip install locust

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

class My_task_set(TaskSet):
 
    #添加初始化方法
    def on_start(self):
        print("類似類中的構(gòu)造方法，每個(gè)用戶在任務(wù)開始前，只執(zhí)行一次,在這里可以定義一個(gè)對(duì)象的屬性，這樣其它測(cè)試集就可以使用這個(gè)屬性")

def on_stop(self):
    print("類似類中的后置方法，每個(gè)用戶在任務(wù)開始后，只執(zhí)行一次,在這里可以定義一個(gè)對(duì)象的屬性，這樣其它測(cè)試集就可以使用這個(gè)屬性")

from locust import HttpLocust
from locust import HttpUser
from locust import task
from locust import TaskSet
 
# locust中的client會(huì)自動(dòng)幫我們處理cookies。類似request.session(),所以如果我們登陸的時(shí)候，只需要在on_start中登陸一次就可以了
 
# 如果在locust中，如果url是不需要統(tǒng)計(jì)，則我們不要用clent去訪問(wèn)api，應(yīng)該用request去訪問(wèn)，這樣就locust就不會(huì)統(tǒng)計(jì)request庫(kù)發(fā)起的請(qǐng)請(qǐng)求
#指定一個(gè)任務(wù)集
class My_task_set(TaskSet):
 
    #添加初始化方法
    def on_start(self):
        print("類似類中的構(gòu)造方法，每個(gè)用戶在任務(wù)開始前，只執(zhí)行一次,在這里可以定義一個(gè)對(duì)象的屬性，這樣其它測(cè)試集就可以使用這個(gè)屬性")
 
    def on_stop(self):
        print("類似類中的后置方法，每個(gè)用戶在任務(wù)開始后，只執(zhí)行一次,在這里可以定義一個(gè)對(duì)象的屬性，這樣其它測(cè)試集就可以使用這個(gè)屬性")
 
    #這是某個(gè)任務(wù),30是比例，比如這里是30/50
    @task(30)
    def getindex1(self):
        # client就是個(gè)requests對(duì)象
        # catch_response，告訴locust如何判斷請(qǐng)求失敗還是成功
        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就是個(gè)requests對(duì)象
        res = self.client.get("/bainianminguo/p/7253930.html")
        print(res)
 
class WebSite(HttpUser):
    # 指定要執(zhí)行哪個(gè)任務(wù)集
    # task_set = My_task_set
 
    tasks = [My_task_set,]
    # 請(qǐng)求和請(qǐng)求之間最小的間隔時(shí)間
    min_wait = 1000
    # 請(qǐng)求和請(qǐng)求之間最大的間隔時(shí)間
    max_waif = 2000
 
 
# Number of total users to simulate   模擬的用戶數(shù)
# Spawn rate                          每秒鐘產(chǎn)生的用戶數(shù)

一旦單臺(tái)機(jī)器不夠模擬足夠多的用戶時(shí)，Locust支持運(yùn)行在多臺(tái)機(jī)器中進(jìn)行壓力測(cè)試。
 
為了實(shí)現(xiàn)這個(gè)，你應(yīng)該在 master 模式中使用--master標(biāo)記來(lái)啟用一個(gè) Locust 實(shí)例。這個(gè)實(shí)例將會(huì)運(yùn)行你啟動(dòng)測(cè)試的 Locust 交互網(wǎng)站并查看實(shí)時(shí)統(tǒng)計(jì)數(shù)據(jù)。br>master 節(jié)點(diǎn)的機(jī)器自身不會(huì)模擬任何用戶。相反，你必須使用 --slave 標(biāo)記啟動(dòng)一臺(tái)到多臺(tái) Locustslave 機(jī)器節(jié)點(diǎn)，與標(biāo)記 --master-host 一起使用(指出master機(jī)器的IP/hostname)。
 
常用的做法是在一臺(tái)獨(dú)立的機(jī)器中運(yùn)行master，在slave機(jī)器中每個(gè)處理器內(nèi)核運(yùn)行一個(gè)slave實(shí)例。
 
注意：master 和每一臺(tái) slave 機(jī)器，在運(yùn)行分布式測(cè)試時(shí)都必須要有 locust 的測(cè)試文件。

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