快捷導(dǎo)航

基于python locust庫實(shí)現(xiàn)性能測試

更新時(shí)間：2020年05月11日 09:14:49 作者：wanglai

這篇文章主要介紹了基于python locust庫實(shí)現(xiàn)性能測試,文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下

Locust（俗稱蝗蟲）, 一個(gè)輕量級(jí)的開源壓測工具，用Python編寫。

安裝

pip3 install locust

Python編寫性能測試腳本

import json

from locust import HttpLocust, TaskSet, task

"""
創(chuàng)建后臺(tái)管理站點(diǎn)壓測類，需要繼承TaskSet
可以添加多個(gè)測試任務(wù)
"""


class AdminLoadTest(TaskSet):

  # 用戶執(zhí)行task前調(diào)用
  def on_start(self):
    pass

  # 用戶執(zhí)行task后調(diào)用
  def on_stop(self):
    pass

  @task(1)
  def download(self):
    header = {"Content-Type": "application/json"}
    data = {
      "staff_id": 94
    }
    data = json.dumps(data)
    self.client.post('/get_know_list', data=data, headers=header)

  @task(2)
  def body(self):
    header = {"Content-Type": "application/json"}
    data = {
      "staff_id": 94
    }
    data = json.dumps(data)
    self.client.post('/get_answer_list', data=data, headers=header)


class RunLoadTests(HttpLocust):
  """
  創(chuàng)建運(yùn)行壓測類
  """
  task_set = AdminLoadTest
  min_wait = 1000
  max_wait = 50000
  weight = 1
if __name__ == "__main__":
  import os

  os.system("locust -f test_download.py --host=http://127.0.0.1:5000")

AdminLoadTest類繼承TaskSet類，用以描述用戶行為：

使用@task裝飾的方法為一個(gè)事務(wù)，方法的參數(shù)用于指定該行為的執(zhí)行權(quán)重，參數(shù)越大每次被用戶執(zhí)行的概率越高，默認(rèn)為1（事務(wù)blogs()被執(zhí)行的概率是bky_index()的2倍）；

on_start()：每個(gè)locust用戶執(zhí)行測試事務(wù)之前執(zhí)行一次，用于做初始化的工作，如登錄；
on_stop(): 每次執(zhí)行完測試事務(wù)之后執(zhí)行一次。
RunLoadTests類用于設(shè)置性能測試屬性：
task_set：指向定義的一個(gè)用戶行為類；
min_wait：模擬用戶在執(zhí)行每個(gè)任務(wù)之間等待的最小時(shí)間，單位為毫秒；
max_wait：模擬用戶在執(zhí)行每個(gè)任務(wù)之間等待的最大時(shí)間，單位為毫秒（min_wait和max_wait默認(rèn)值為1000，因此，如果沒有聲明min_wait和max_wait，則locust將在每個(gè)任務(wù)之間始終等待1秒。）；
weight：一個(gè)文件中有多個(gè)locust用戶類時(shí)，指定用戶類的權(quán)重（默認(rèn)新增locust用戶時(shí)會(huì)隨機(jī)選擇一個(gè)用戶類）；

啟動(dòng)

終端進(jìn)入代碼目錄，輸入 locust -f start.py --host=http://localhost:5000

使用os.system在代碼里面寫入啟動(dòng)命令

各個(gè)參數(shù)的含義

-f : 指定要運(yùn)行的測試腳本文件
--host : 要加載主機(jī)的URL前綴,不指定時(shí)，讀取WebsiteUser類中定義的host；
--master : 單臺(tái)機(jī)器不能模擬更多用戶時(shí)，分布式模式，通過--master指定主進(jìn)程；打開Locust的web界面

使用上面的命令行啟動(dòng)Locust之后，應(yīng)該打開瀏覽器并將其指向http://localhost:8089/(如果您在本地運(yùn)行Locust)。

Number of total users to simulate : 設(shè)置模擬用戶數(shù)；Hatch rate（users spawned/second） : 每秒產(chǎn)生（啟動(dòng)）的虛擬用戶數(shù)；