🚀 ⚡️ 蝗虫mcp服务器
用于运行Locust负载测试的模型上下文协议（MCP）服务器实现。该服务器实现了Locust负载测试功能与AI驱动的开发环境的无缝集成。
✨ 特性
- 与模型上下文协议框架简单集成
- 支持无头模式和UI模式
- 可配置的测试参数（用户、生成率、运行时间）
- 易于使用的API用于运行Locust负载测试
- 实时测试执行输出
- 开箱即用的HTTP/HTTPS协议支持
- 自定义任务场景支持
🔧 先决条件
在开始之前，请确保已安装以下内容：
- Python 3.13或更高版本
- uv包管理器(安装指南)
📦 安装
- 克隆存储库：
git clone https://github.com/qainsights/locust-mcp-server.git- 安装所需的依赖项：
uv pip install -r requirements.txt- 设置环境变量（可选）：
创建一个 .env 项目根目录中的文件：
LOCUST_HOST=http://localhost:8089 # Default host for your tests
LOCUST_USERS=3 # Default number of users
LOCUST_SPAWN_RATE=1 # Default user spawn rate
LOCUST_RUN_TIME=10s # Default test duration🚀 入门指南
- 创建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客户端（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运行测试，例如。
run locust test for hello.pyLocust MCP服务器将使用以下工具启动测试：
run_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
)参数：
test_file：Locust测试脚本的路径headless：以无头模式运行（True）或使用UI运行（False）host：要进行负载测试的目标主机runtime：测试持续时间（例如，“30秒”、“1米”、“5米”）users：要模拟的并发用户数spawn_rate：用户生成的速率
✨ 用例
- LLM驱动的结果分析
- 借助LLM进行有效调试
🤝 贡献
欢迎投稿！请随时提交拉取请求。
📄 许可证
此项目根据MIT许可证获得许可-有关详细信息，请参阅许可证文件。
