如何从库视角实现程序结束前自动关闭aiohttp.ClientSession？

问题本质 #

你遇到的SSL致命错误，是因为ClientSession未被优雅关闭时，底层的SSL/TCP连接没有正常断开，程序退出时触发资源清理警告。之前尝试的__del__和atexit方案失效，核心原因是程序退出阶段asyncio事件循环已被销毁，无法执行异步的session.close()操作。

解决方案 #

结合两种实现，既支持规范的上下文管理器用法（最可靠），又能自动处理用户直接实例化的场景：

1. 实现异步上下文管理器（推荐用法） #

这是aiohttp官方推荐的资源管理方式，能严格控制会话的生命周期：

import asyncio
from aiohttp import ClientSession

class Client:
    def __init__(self):
        self._session = ClientSession()

    async def get_foo(self, url):
        async with self._session.get(url) as resp:
            return await resp.json()

    async def patch_bar(self, url, data):
        async with self._session.patch(url, json=data) as resp:
            return await resp.json()

    # 异步上下文管理器入口
    async def __aenter__(self):
        return self

    # 上下文退出时自动关闭会话
    async def __aexit__(self, exc_type, exc_val, exc_tb):
        if not self._session.closed:
            await self._session.close()

用户使用示例：

async def main():
    async with Client() as client:
        await client.get_foo("https://api.example.com/foo")
        await client.patch_bar("https://api.example.com/bar", {"key": "value"})

asyncio.run(main())

2. 自动清理机制（兼容非上下文场景） #

针对用户直接实例化Client的场景，用weakref.finalize注册同步清理函数，适配事件循环的两种状态：

import asyncio
import weakref
from aiohttp import ClientSession

class Client:
    def __init__(self):
        self._session = ClientSession()
        # 注册对象销毁时的清理回调，不受循环引用影响
        weakref.finalize(self, self._auto_cleanup, self._session)

    async def get_foo(self, url):
        async with self._session.get(url) as resp:
            return await resp.json()

    async def patch_bar(self, url, data):
        async with self._session.patch(url, json=data) as resp:
            return await resp.json()

    @staticmethod
    def _auto_cleanup(session):
        if session.closed:
            return
        try:
            # 优先尝试用当前运行的事件循环关闭会话
            loop = asyncio.get_running_loop()
            if loop.is_running():
                loop.create_task(session.close())
            else:
                # 事件循环已关闭，创建临时循环执行关闭
                asyncio.run(session.close())
        except RuntimeError:
            # 无运行中的事件循环，用临时循环处理
            asyncio.run(session.close())

    # 保留上下文管理器支持
    async def __aenter__(self):
        return self

    async def __aexit__(self, exc_type, exc_val, exc_tb):
        if not self._session.closed:
            await self._session.close()

这个实现中，weakref.finalize会在Client对象被垃圾回收时触发清理：

• 若事件循环仍在运行，提交异步任务优雅关闭会话；
• 若事件循环已销毁，创建临时事件循环完成会话清理，避免资源泄漏。

为什么原有方案失效？ #

• __del__：垃圾回收时机不可控，且执行时事件循环大概率已销毁，无法执行异步操作；
• atexit：程序退出时asyncio.run()已关闭事件循环，此时无法在原有循环中执行异步关闭，延迟清理容易引发资源警告；
• weakref.finalize比__del__更可靠，不受循环引用影响，能在对象销毁时及时触发清理逻辑。

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