Flask接口压测：Locust高并发测试教程

本文深入剖析了使用Locust对Python Flask接口进行高并发压力测试时的四大典型陷阱：从环境配置（Python版本、虚拟环境激活、错误安装locustio）到Flask服务绑定（必须显式监听0.0.0.0而非仅127.0.0.1）、JSON请求发送（正确使用json=参数而非data=）、再到性能瓶颈根源——Werkzeug开发服务器天生不支持并发，必须替换为gunicorn等生产级WSGI服务器。每一步都直击实操痛点，揭示“QPS上不去”“404报错”“ModuleNotFoundError”等表象背后的本质原因，帮你绕过踩坑弯路，真正跑出可信、高效的压测结果。

Locust安装后运行报错 ModuleNotFoundError: No module named 'locust'

这是 Python 环境没装对，或者用了虚拟环境但没激活。Locust 从 2.0 开始已不再兼容 Python 2，且必须用 pip 安装到当前 Python 解释器环境里。

实操建议：

• 确认 Python 版本 ≥ 3.8：python --version
• 用 pip install locust（不是 pip install locustio，后者是旧版）
• 如果用虚拟环境，先 source venv/bin/activate（Linux/macOS）或 venv\Scripts\activate（Windows），再装
• 验证是否装进当前环境：python -c "import locust; print(locust.__version__)"

Flask 接口在 Locust 里调不通，返回 404 或连接拒绝

常见原因是 Flask 默认只监听 127.0.0.1:5000，而 Locust 脚本默认也跑在本地，但可能因 host 配置、调试模式或端口冲突导致请求发不到。

实操建议：

• 启动 Flask 时显式绑定地址：app.run(host='0.0.0.0', port=5000)（仅开发用，别在生产环境这么干）
• 检查 Flask 是否启用了调试模式（debug=True），某些情况下会干扰长连接，压测前设为 False
• Locust 脚本中 host 参数要写全，比如 host="http://127.0.0.1:5000"，不能只写 "localhost"（DNS 解析可能失败）
• 用 curl http://127.0.0.1:5000/your-endpoint 先手动验证接口可达

Locust 脚本里怎么发带 JSON 的 POST 请求

Flask 接口常接收 application/json，但 Locust 默认不设 Content-Type，也不自动序列化 body，直接传 dict 会 400。

实操建议：

• 用 self.client.post("/api/login", json={"username": "test", "password": "123"}) —— json= 参数会自动序列化 + 设 header
• 如果要用 data= 手动传，得自己 import json，然后 data=json.dumps({...}) + 显式加 header：headers={"Content-Type": "application/json"}
• 注意 Flask 端用 request.get_json() 取值，别用 request.form
• 如果接口需要 token 或 cookie，记得在 on_start 里登录一次并保存 session

压测时 QPS 上不去，CPU 占用低但响应延迟飙升

这不是 Locust 不行，而是 Flask 默认的 Werkzeug 开发服务器单线程 + 单进程，根本扛不住并发。它只适合调试，不是 Web 服务器。

实操建议：

• 压测前必须换 WSGI 服务器：推荐 gunicorn（简单）或 uWSGI（灵活）
• 用 gunicorn 启动：gunicorn -w 4 -b 127.0.0.1:5000 app:app（4 个工作进程）
• Locust 脚本里的 host 改成 "http://127.0.0.1:5000"，指向 gunicorn 端口
• 观察 gunicorn 日志是否有 worker timeout 或重启，适当调大 --timeout 和 --keep-alive

真正卡住的地方往往不在代码逻辑，而在 Flask 默认服务器的线程模型和你没意识到它根本不该出现在压测链路里。

文中关于的知识介绍，希望对你的学习有所帮助！若是受益匪浅，那就动动鼠标收藏这篇《Flask接口压测：Locust高并发测试教程》文章吧，也可关注golang学习网公众号了解相关技术文章。