python locust學習筆記

from locust import HttpLocust,TaskSet,task

class Discuz_Login(TaskSet):
#task（1）task（10）代表權重，eg：如下表示login方法執行10次時，index方法才執行1次
@task(1)
def index(self):
url=“https://www.baidu.com/”
headers={“User-Agent”:“Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/70.0.3538.77 Safari/537.36”}
self.client.get(url,headers=headers,name=“打開首頁”)
@task(10)
def login(self):
url=""
headers=""
data=""
with self.client.post(url,headers=headers,data=data,name=‘登錄’)as response:
print(response.text)
return response

class Discus_User(HttpLocust):
task_set =Discuz_Login
host = “https://www.baidu.com/”
#思考時間在1秒到3秒內隨機使用一個時間,不設置的話默認停頓1秒
min_wait = 1000
max_wait = 3000
執行：
打開cmd，到腳本所在目錄：locust -f xx.py
瀏覽器打開網址：http://localhost:8089/
參數化&關聯：



分佈式執行locust腳本
電腦cpu是幾個核，就可以在本機開幾個cmd進程；這樣可以最大限度的利用cpu，產生更大的壓力
當然slave也可以是不同的電腦

當主機和slave在不同的機器上：