Locust迁移HttpUser至FastHttpUser及并发连接错误排查求助

环境信息 #

• 操作系统：Windows Server 2016
• Python版本：3.8.10
• Locust版本：2.15.1

问题1：高并发下出现连接中断错误 #

针对你遇到的ConnectionError(ProtocolError('Connection aborted.', RemoteDisconnected('Remote end closed connection without response')))，结合Locust官方推荐的优化方向，给出以下排查和解决建议：

• 优先切换到FastHttpUser：HttpUser基于requests/urllib3，高并发下连接管理效率较低；FastHttpUser采用geventhttpclient原生实现高并发连接，能从根源减少这类连接问题，这是官方首推的优化手段。
• 配置FastHttpUser连接参数：
  在Locust类中设置max_connections，控制单用户的最大并发连接数，避免连接过载：

  class MyLoadTestUser(FastHttpUser):
      max_connections = 8  # 根据实际场景调整
  

• 检查目标服务状态：该错误多数情况下是目标服务器因压力过大主动断开连接，需排查服务端日志，确认是否存在限流、连接数超限或资源（CPU/内存）耗尽的情况。
• 优化Windows网络限制：Windows Server默认的TCP连接参数对高并发场景不够友好，可调整以下注册表项：
  • HKEY_LOCAL_MACHINE\SYSTEM\CurrentControlSet\Services\Tcpip\Parameters下设置TcpTimedWaitDelay为30（缩短TIME_WAIT连接回收时间）
  • 增大MaxUserPort的值（如设置为65534），提升可用端口范围
• 采用分布式压测：单Locust实例在Windows上的并发能力有限，启动多个worker节点分散压力，避免单节点连接资源耗尽。
• 开启调试日志：通过--log-level DEBUG启动Locust，查看连接建立、请求发送的详细日志，定位错误发生的具体阶段。

问题2：HttpUser转FastHttpUser后401认证失败的最小化修复 #

HttpUser的self.client是requests.Session，会自动维护cookie和认证状态；而FastHttpUser的self.client基于geventhttpclient，状态管理逻辑不同，仅替换类名会导致认证状态丢失。以下是最小改动的修复方案：

1. 手动同步登录Cookie #

原HttpUser的登录逻辑会自动保存cookie，迁移后需手动将登录返回的cookie添加到FastHttpUser的client中：

def login(self):
    # 执行登录请求
    response = self.client.post("/login", data={"username": "your_user", "password": "your_pwd"})
    # 提取并设置cookie
    for cookie in response.cookies:
        self.client.cookies.set(
            name=cookie.name,
            value=cookie.value,
            domain=cookie.domain,
            path=cookie.path
        )

2. 对齐默认请求头 #

FastHttpUser的默认请求头与HttpUser存在差异（如User-Agent），可在on_start方法中对齐：

class MyLoadTestUser(FastHttpUser):
    def on_start(self):
        # 复制HttpUser的默认请求头
        self.client.headers.update({
            "User-Agent": "python-requests/2.28.1",
            "Accept-Encoding": "gzip, deflate",
            "Accept": "*/*",
            "Connection": "keep-alive"
        })
        self.login()

3. Token认证场景适配 #

如果业务用Token认证，只需保持原有的Token设置逻辑即可：

def login(self):
    response = self.client.post("/login", data={"username": "your_user", "password": "your_pwd"})
    token = response.json().get("token")
    self.client.headers["Authorization"] = f"Bearer {token}"

4. 校验请求一致性 #

开启DEBUG日志对比两种User发送的请求，确保请求方法、参数、headers完全一致，排除因细微差异导致的认证失败。

通过以上步骤，无需修改核心业务逻辑，即可完成迁移并解决401问题。

内容的提问来源于stack exchange，提问作者Saurabh Gupta