httpx.Client连接池关联逻辑及外部客户端复用管理咨询

连接池与客户端绑定关系的疑问 #

在httpx文档的clients章节提及了连接池，且官方普遍推荐使用httpx.Client，但存在一个疑问：连接池是仅与httpx.Client实例绑定，还是调用open/close方法、进出客户端上下文管理器会对连接池产生影响？

举个例子，如果连接池仅与客户端实例绑定，那么以下代码应该能受益于连接池的复用：

shared_client = httpx.Client()

with shared_client as client:
    # 执行操作

with shared_client as client:
    # 执行更多操作

但如果连接池会受close方法或退出上下文的影响，那么上述代码就无法利用连接池的复用机制。

补充说明与实际需求 #

后来发现上述示例违反了httpx客户端的基本限制：客户端无法重新开启，在此先致歉，应该先做测试再提问。

我的实际需求是：让内部使用httpx.Client/httpx.AsyncClient的类的用户，能够自行提供客户端实例——用户可以复用该客户端，但同时需要自行负责客户端的关闭操作。

实现思路与代码 #

以下是满足需求的初步实现方案，通过包装类来实现客户端的复用管理：

import warnings

import httpx


warnings.simplefilter("always")


class _ClientWrapper:
    def __init__(self, client: httpx.Client) -> None:
        self.client = client

    def __getattr__(self, value):
        return getattr(self.client, value)

    def __enter__(self):
        return self.client.__enter__()

    def __exit__(self, exc_type, exc_value, traceback):
        if not self.client.is_closed:
            warnings.warn(f"httpx.Client instance '{self.client}' is still open. ")


class Getter:
    def __init__(self, client: httpx.Client | None = None) -> None:
        self.client = httpx.Client() if client is None else _ClientWrapper(client)

    def get(self, url: str) -> httpx.Response:
        with self.client:
            response = self.client.get(url)
            response.raise_for_status()

        return response


client = httpx.Client()
getter = Getter(client=client)

response = getter.get("https://www.example.com")
print(response)  # 200

print(getter.client.is_closed)  # False
client.close()
print(getter.client.is_closed)  # True

这个包装类的设计逻辑是：将所有属性访问委托给原始客户端实例，但重写__exit__方法——仅在客户端未关闭时发出警告，而不执行关闭操作。这样一来，若用户自行提供客户端实例，就需要自行负责客户端的生命周期管理（包括关闭）。

另外一种思路是直接子类化httpx.Client，但这种方式的局限性在于：用户必须使用我子类化后的客户端，无法使用任意的httpx.Client实例，灵活性不足。

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