是否可以覆盖请求中的默认套接字选项？

我使用优秀的 python 请求库为 REST API 编写了一个非常简单的客户端。一切工作都很好，我通过负载平衡器运行客户端，负载平衡器可以正常检测空闲的 tcp 连接并终止它们。我希望我的客户端使用一些与我的平台（Linux）上的默认选项不同的 TCP 保持活动选项。但我没有看到任何简单的方法来告诉套接字库我想为新套接字选择一些默认选项。

当直接使用 socket.create_connection 时，使用装饰器就可以很容易地做到这一点，但我不知道当实际调用被埋在某些第三方库中时（就像请求的情况一样）如何使装饰调用可用。

提前致谢

较新版本的urllib3（自1.8.3起，2014-06-23发布）支持设置套接字选项。

您可以从以下位置设置这些选项requests（自 2.4.0 起，2014-08-29 发布）通过创建自定义适配器：

class HTTPAdapterWithSocketOptions(requests.adapters.HTTPAdapter):
    def __init__(self, *args, **kwargs):
        self.socket_options = kwargs.pop("socket_options", None)
        super(HTTPAdapterWithSocketOptions, self).__init__(*args, **kwargs)

    def init_poolmanager(self, *args, **kwargs):
        if self.socket_options is not None:
            kwargs["socket_options"] = self.socket_options
        super(HTTPAdapterWithSocketOptions, self).init_poolmanager(*args, **kwargs)

然后您可以将此适配器安装到需要自定义套接字选项的会话（例如设置SO_KEEPALIVE):

adapter = HTTPAdapterWithSocketOptions(socket_options=[(socket.SOL_SOCKET, socket.SO_KEEPALIVE, 1)])
s = requests.session()
s.mount("http://", adapter)
s.mount("https://", adapter)