Qainsights Locust MCP Server

🚀 Locust-MCP-Server

Locust-MCP-Server是一個支持模型上下文協議（MCP）的Locust負載測試服務器實現，它能夠讓Locust負載測試功能與AI驅動開發環境實現無縫集成。

🚀 快速開始

創建Locust測試腳本

首先，你需要創建一個Locust測試腳本，比如hello.py，示例代碼如下：

from locust import HttpUser, task, between

class QuickstartUser(HttpUser):
    wait_time = between(1, 5)

    @task
    def hello_world(self):
        self.client.get("/hello")
        self.client.get("/world")

    @task(3)
    def view_items(self):
        for item_id in range(10):
            self.client.get(f"/item?id={item_id}", name="/item")
            time.sleep(1)

    def on_start(self):
        self.client.post("/login", json={"username":"foo", "password":"bar"})

配置MCP服務器

使用你喜歡的MCP客戶端（Claude Desktop、Cursor、Windsurf等）來配置MCP服務器，示例配置如下：

{
  "mcpServers": {
    "locust": {
      "command": "/Users/naveenkumar/.local/bin/uv",
      "args": [
        "--directory",
        "/Users/naveenkumar/Gits/locust-mcp-server",
        "run",
        "locust_server.py"
      ]
    }
  }
}

啟動測試

現在，你可以讓LLM運行測試，例如輸入運行Locust測試以執行hello.py。Locust MCP服務器將使用以下工具啟動測試：

• run_locust：帶有配置選項的測試運行（無頭模式、主機、持續時間、用戶數和啟動率）

✨ 主要特性

• 可簡單集成到模型上下文協議框架。
• 支持無頭模式和UI模式。
• 可配置測試參數，如用戶數、啟動率、運行時間。
• 提供易用的API用於運行Locust負載測試。
• 能即時輸出測試執行情況。
• 原生支持HTTP/HTTPS協議。
• 支持自定義任務場景。

🔧 先決條件

開始之前，請確保安裝以下內容：

• Python 3.13或更高版本。
• uv包管理器（安裝指南）。

📦 安裝指南

克隆倉庫

git clone https://github.com/yourusername/locust-mcp-server.git

安裝依賴項

uv pip install -r requirements.txt

設置環境變量（可選）

在項目根目錄創建一個.env文件：

LOCUST_HOST=http://localhost:8089  # 測試默認主機
LOCUST_USERS=3                     # 默認用戶數
LOCUST_SPAWN_RATE=1               # 用戶啟動率默認值
LOCUST_RUN_TIME=10s               # 默認測試持續時間

📚 詳細文檔

運行Locust測試API

run_locust(
    test_file: str,
    headless: bool = True,
    host: str = "http://localhost:8089",
    runtime: str = "10s",
    users: int = 3,
    spawn_rate: int = 1
)

參數詳情

🤝 貢獻

歡迎貢獻！請隨意提交拉取請求。

📄 許可證

本項目在MIT許可證下，有關詳細信息，請參閱LICENSE文件。