使用依赖关系注入关闭资源-解网

问：

我有以下伪代码：

class Resource:
    """E.g. Session, connection, file, etc."""
    async def open(self):
        pass
    async def close(self):
        pass

class ResourceUser:
    def __init__(self, resource: Resource):
        self.resource = resource

async def main():
    r = Resource(...)  # Plenty of those
    await r.open(...)
    # More resource initialization
    try:
        ResourceUser(r, r2, r3)
    finally:
        await r.close(...) # In practice done with AsyncExitStack
        await r.close(...)

Main 是一个大函数，我想提取资源和 ResourceUser 的创建：

async def create_user():
    r, r2, r3, ... = Resource(), ...
    await r.open()
    return ResourceUser(r)

这样一来，我就失去了正确关闭资源的选项。

解决此问题的可选方法是在 ResourceUser 中创建一个 close（）函数：

async def close(self):
    await self.resource.close()

不幸的是，这假定 ResourceUser“拥有”给定的资源，并且有许多潜在的缺点，例如阻止在多个实例之间共享资源（如连接池）。

指望（类似于 RAII）不是一种选择，特别是考虑到关闭方法的异步性质。__del__

在不同的函数中初始化资源，然后在 main（）中创建 User，将导致 return 语句中出现大量不同的资源，这是相当丑陋的。猴子包装也很丑陋。ResourceUser.close()

有没有一种标准化的方法既不会过于复杂，又能达到预期的效果？

Python 资源管理

from contextlib import asynccontextmanager


class Resource:
    """E.g. Session, connection, file, etc."""

    async def open(self):
        pass

    async def close(self):
        pass


class ResourceUser:
    def __init__(self, resource: Resource, ...):
        self.resource = resource


@asynccontextmanager
async def get_resource_user():
    r1 = Resource(...)  # Plenty of those
    r2 = Resource(...)  # Plenty of those
    try:
        await r1.open(...)  # better use asynccontextmanager for this too, to not call close() explicitly
        await r2.open(...)
        yield ResourceUser(r1, r2)

    finally:
        await r1.close(...)
        await r2.close(...)


async def main():

    async with get_resource_user() as user:
        # do stuff with `user`

    # resources are closed automatically at this point

上一个：Visual Studio 2022 XAML 设计器不显示用作静态资源的图像

下一个：SLURM 报告的内存消耗

使用依赖关系注入关闭资源

Closing resources with dependency injection

评论