python - aiohttp如何在一个类中保存一个持久化的ClientSession？

Question

我正在编写一个将使用 aiohttp 执行 http 请求的类。根据文档，我不应该为每个请求创建一个 ClientSession，所以我想重用同一个 session 。

代码:

class TestApi:
   def __init__(self):
      self.session = aiohttp.ClientSession()

   # async defs methods from here 

做的时候

TestApi()

我收到错误:Unclosed client session。

持久化 ClientSession 对象的解决方案是什么？

Answer 1

一行中的表达式 TestApi() 本身创建了一个 TestApi 对象并立即将其丢弃。 aiohttp 提示 session 从未关闭(通过保留 async with block 或显式调用 close())，但即使在没有警告的情况下，不将 API 对象分配给将实际使用它的变量是没有意义的。

要重用 session ，您的代码需要有权访问 session 或持有它的对象:

async def fetch(url):
    async with aiohttp.request('GET', url) as resp:
        resp.raise_for_status()
        return await resp.read()

async def main():
    url1_data, url2_data = asyncio.gather(
        fetch('http://url1'), fetch('http://url2'))
    url3_data, url4_data = asyncio.gather(
        fetch('http://url3'), fetch('http://url4'))

一个选项是将 session 参数添加到 fetch(和其他函数)，并使用在 main() 中创建的 session 一致地调用它>。更好的选择是创建一个 API 类并将全局函数(如 fetch)转换为方法:

class Http:
    async def __aenter__(self):
        self._session = aiohttp.ClientSession()
        return self

    async def __aexit__(self, *err):
        await self._session.close()
        self._session = None

    async def fetch(self, url):
        async with self._session.get(url) as resp:
            resp.raise_for_status()
            return await resp.read()

main() 仍然可以作为函数存在，但它可以始终使用持有 session 的对象:

async def main():
    async with Http() as http:
        url1_data, url2_data = await asyncio.gather(
            http.fetch('http://url1'), http.fetch('http://url2'))
        url3_data, url4_data = await asyncio.gather(
            http.fetch('http://url3'), http.fetch('http://url4'))

在上面的代码中，async with 语句用于确保在离开作用域时关闭 session 。