[本文出自天外歸雲的部落格園]
在前篇的基礎上,我們可以利用Locust性能測試架構編寫python腳本對指定網站或者接口進行壓測,對于Locust的使用我們可以歸納為三步:
1. 編寫python壓力測試腳本;
2. 在指令行中啟動Locust服務并指定壓測腳本與hosts等資訊;
對于以上三步,我通過編寫腳本來做一下進一步封裝,實作化繁為簡,“一步壓測”的目的。這裡有兩個python腳本,一個是壓測腳本,一個是啟動腳本。還有一個PowerShell腳本,用來啟動壓測。這三個腳本放在一個檔案夾下面:
壓測腳本格式如下,大家可以随意更改或添加配置,起名為“locust_test.py”并和啟動腳本放在同一目錄下:
啟動腳本如下,起名為“start_locust.py”:
PowerShell腳本如下(打開方式預設設為PowerShell啟動),起名為“start.ps1”:
使用方式很簡單,輕按兩下“start.ps1”腳本,出現如下界面:
之後浏覽器會自動打開我們的壓測頁面:
至此我們就可以對指定網站url接口施加壓力了。
如果不關心網站接口的性能資料,直接對網站進行壓力攻擊,我們可以對腳本進行無界面化精簡處理。建立python腳本“start_locust_implicit.py”,内容如下:
隻是在之前的腳本的基礎上對subprocess的調用腳本内容進行了修改,添加了需要傳入的參數。然後建立ps1腳本“start_locust_implicit.ps1”,内容如下:
使用方式仍然是輕按兩下以PowerShell運作ps1腳本,根據提示一步一步操作就可以對網站發動壓力攻擊了。