如何在python locust中关闭TCP连接（requests.session）

Question

我正在使用 python Locust 包对 Kubernetes 上运行的服务 API 进行负载测试。

我在源代码中看到它HttpUser用于requests.session.request()发送请求。默认情况下requests.session保持连接处于活动状态（这会导致所有请求都发送到一个 pod，而不是分布到 Kubernetes 上的所有 pod）。我知道在每个请求后关闭连接的唯一方法是connection: close在初始化 时设置requests.session，这是在 Locust 中从我那里抽象出来的。

我尝试添加headers={'connection':'close'}请求调用，但这并没有成功。所有请求仍然发送到同一个 Pod。有人知道如何在运行时更改此设置吗？

Answer 1

你可以试试

request = requests.session.request()
request.close()