aiohttp 如何在类中保存持久的 ClientSession？

Question

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

代码：

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

   # async defs methods from here 

做的时候

TestApi()

我收到错误：Unclosed client session。

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

Answer 1

TestApi()一行上的表达式本身会创建一个TestApi对象并立即将其丢弃。aiohttp抱怨会话从未关闭（通过离开async with块或显式调用close()），但即使没有警告，不将 API 对象分配给实际使用的变量也是没有意义的。

要重用会话，您的代码需要有权访问会话或持有它的对象：

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(). 更好的选择是创建一个 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()仍然可以作为函数存在，但它可以一致地使用持有会话的对象：

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语句用于确保只要离开范围就关闭会话。