Python 包 - aiohttp 有一个警告消息"Unclosed client session"

Question

我的代码如下：

import asyncio
import aiohttp
urls = [
    'http://www.163.com/',
    'http://www.sina.com.cn/',
    'https://www.hupu.com/',
    'http://www.csdn.net/'
]
async def get_url_data(u):
    """
    read url data
    :param u:
    :return:
    """
    print('running ', u)
    resp = await aiohttp.ClientSession().get(url=u)
    headers = resp.headers
    print(u, headers)
    return headers

async def request_url(u):
    """
    main func
    :param u:
    :return:
    """
    res = await get_url_data(u)
    return res
loop = asyncio.get_event_loop()
task_lists = asyncio.wait([request_url(u) for u in urls])
loop.run_until_complete(task_lists)
loop.close()

运行代码时，它是一个警告消息：未封闭的客户端会话

有人可以给我一些解决方案吗？

非常感谢

Answer 1

您应该在最后关闭连接。您有2个选项：

您可以手动关闭连接：

import aiohttp
session = aiohttp.ClientSession()
# use the session here
session.close()

，也可以与contex管理器一起使用：

import aiohttp
import asyncio
async def fetch(client):
    async with client.get('http://python.org') as resp:
        assert resp.status == 200
        return await resp.text()
async def main(loop):
    async with aiohttp.ClientSession(loop=loop) as client:
        html = await fetch(client)
        print(html)
loop = asyncio.get_event_loop()
loop.run_until_complete(main(loop))

客户端会话支持自我关闭的上下文管理协议。

Answer 2

如果您不使用上下文管理器，则关闭它的正确方法也需要await。互联网上的许多答案都错过了那部分，很少有人真正注意到它，大概是因为大多数人使用更方便的上下文经理。但是，当/如果您在执行tearDownClass()内关闭CC_3内的课程范围时，手册await session.close()是必不可少的。

import aiohttp
session = aiohttp.ClientSession()
# use the session here
await session.close()

Answer 3

您应该使用async上下文管理器使用ClientSession进行适当的阻止/释放资源：

async def get_url_data(u):
    """
    read url data
    :param u:
    :return:
    """
    print('running ', u)
    async with aiohttp.ClientSession() as session:
        resp = await session.get(url=u)
        headers = resp.headers
        print(u, headers)
        return headers