Python压力测试：Locust框架实战指南

本文深入解析了Python自动化压力测试中Locust框架的实战要点，涵盖从环境搭建（强调Python 3.8+、conda下避坑安装、配置文件显式加载）到核心编码（推荐HttpUser类、手动管理登录态、合理设置timeout与User-Agent）、再到分布式压测排障（master绑定0.0.0.0、开放5557/5558端口、ulimit调优、多进程/多机扩展策略）的完整链路，并揭示了响应时间突增背后的常见陷阱——如系统文件描述符耗尽、DNS缓存滥用、单核CPU瓶颈等非服务端因素，帮助读者避开“看似代码没问题却压不出真实性能”的典型误区，真正用好Locust实现高效、稳定、可扩展的高并发压测。

Locust安装与基础配置要注意什么

Locust不是pip install locust就能直接跑通的。Python 3.8+是硬性要求，低于3.7会报ModuleNotFoundError: No module named 'asyncio'；Windows用户如果用conda环境，得先conda install -c conda-forge locust，否则可能因pyzmq编译失败卡在启动阶段。

• 确保locust命令能执行：运行locust --version看到输出才算装对
• 配置文件默认不加载，想用locust.conf需显式加--config=locust.conf
• LOCUSTFILE环境变量可指定脚本路径，避免每次敲-f xxx.py

如何写一个能发真实HTTP请求的User类

HttpUser才是发请求的主力，别用过时的Locust基类。它自带self.client（本质是requests.Session），但默认不带User-Agent，某些网站会直接返回403。

• 登录态必须手动维护：调用self.client.post("/login", json={"u": "a", "p": "b"})后，后续请求自动携带Set-Cookie头
• 请求超时要设：不加timeout=10参数，慢接口会让整个worker卡住，压测数据失真
• 避免在on_start里做耗时操作（比如读大文件），它会在每个虚拟用户启动时执行，拖慢整体并发初始化

示例片段：

from locust import HttpUser, task, between
class ApiUser(HttpUser):
wait_time = between(1, 3)
def on_start(self):
    self.client.post("/auth/login", json={"username": "test", "password": "123"})

@task
def get_list(self):
    self.client.get("/api/items", timeout=5)

分布式模式下常见连接失败怎么排查

locust -f script.py --master和--worker分开跑时，worker连不上master是最高频问题。根本原因通常是网络隔离或端口未开放，而非代码错误。

• master默认只监听127.0.0.1:5557，worker在另一台机器上就必然连不上——必须加--master-bind-host=0.0.0.0
• 防火墙常拦截5557（master通信）和5558（web UI），Linux用sudo ufw allow 5557，云服务器还要配安全组
• worker启动后没出现在master页面？检查日志里有没有Failed to connect to master，有就说明网络层不通，不用查Python逻辑

压测结果里响应时间突增但错误率很低，可能是什么原因

这通常不是服务端崩了，而是Locust自身资源瓶颈。尤其当--users设到5000+、单机跑worker时，Python GIL和系统文件描述符限制会先于目标服务拖垮测试。

• Linux默认ulimit -n是1024，压测中每个TCP连接占一个fd，很快打满，现象是大量ConnectionRefusedError或请求卡在pending状态
• 检查top里locust进程CPU是否长期100%，是的话说明单核跑满，得加--processes启多进程（注意：非多线程）
• 更隐蔽的是DNS缓存：频繁请求不同子域时，没配resolve_hostnames=False会导致每秒上百次DNS查询，把worker拖慢

真实场景里，宁愿多起几台低配worker，也不要在一台机器上硬扛5000+ users。Locust的分布式设计，本来就是为横向扩展准备的。

到这里，我们也就讲完了《Python压力测试：Locust框架实战指南》的内容了。个人认为，基础知识的学习和巩固，是为了更好的将其运用到项目中，欢迎关注golang学习网公众号，带你了解更多关于的知识点！