天天看點

Python Locust對指定網站“一鍵壓測”

[本文出自天外歸雲的部落格園]

在前篇的基礎上,我們可以利用Locust性能測試架構編寫python腳本對指定網站或者接口進行壓測,對于Locust的使用我們可以歸納為三步:

1. 編寫python壓力測試腳本;

2. 在指令行中啟動Locust服務并指定壓測腳本與hosts等資訊;

對于以上三步,我通過編寫腳本來做一下進一步封裝,實作化繁為簡,“一步壓測”的目的。這裡有兩個python腳本,一個是壓測腳本,一個是啟動腳本。還有一個PowerShell腳本,用來啟動壓測。這三個腳本放在一個檔案夾下面:

Python Locust對指定網站“一鍵壓測”

壓測腳本格式如下,大家可以随意更改或添加配置,起名為“locust_test.py”并和啟動腳本放在同一目錄下:

啟動腳本如下,起名為“start_locust.py”:

PowerShell腳本如下(打開方式預設設為PowerShell啟動),起名為“start.ps1”:

使用方式很簡單,輕按兩下“start.ps1”腳本,出現如下界面:

Python Locust對指定網站“一鍵壓測”

之後浏覽器會自動打開我們的壓測頁面:

Python Locust對指定網站“一鍵壓測”

至此我們就可以對指定網站url接口施加壓力了。

如果不關心網站接口的性能資料,直接對網站進行壓力攻擊,我們可以對腳本進行無界面化精簡處理。建立python腳本“start_locust_implicit.py”,内容如下:

隻是在之前的腳本的基礎上對subprocess的調用腳本内容進行了修改,添加了需要傳入的參數。然後建立ps1腳本“start_locust_implicit.ps1”,内容如下:

使用方式仍然是輕按兩下以PowerShell運作ps1腳本,根據提示一步一步操作就可以對網站發動壓力攻擊了。