如何锁定上下文管理器__enter__并进入后立即释放锁？异常处理方案

针对你需要先通过锁保护第三方上下文的__enter__（异步场景为__aenter__）方法，进入后立刻释放锁的需求，以下是同步和异步场景下的严谨实现方式，同时兼顾异常处理：

核心思路 #

由于两个上下文均为第三方提供无法修改，我们需要自定义上下文管理器封装逻辑：

• 在锁的临界区内完成目标上下文的进入操作
• 锁自动释放后，保持目标上下文活跃以执行业务逻辑
• 无论业务逻辑正常结束还是抛出异常，确保目标上下文的退出方法被正确调用

异步场景（你的实际场景：async lock + 异步botocore session） #

使用asynccontextmanager封装异步上下文逻辑，确保异常处理和资源清理的正确性：

from contextlib import asynccontextmanager
import sys

@asynccontextmanager
async def async_locked_context_entry(lock, async_ctx):
    entered_ctx = False
    exc_info = None
    try:
        # 在锁的临界区内执行目标上下文的进入操作
        async with lock:
            await async_ctx.__aenter__()
            entered_ctx = True
        # 锁已自动释放，进入业务逻辑块
        yield
    except:
        # 捕获异常信息，用于后续传给目标上下文的退出方法
        exc_info = sys.exc_info()
        raise
    finally:
        # 确保目标上下文的退出方法被调用，传入对应异常信息
        if entered_ctx:
            await async_ctx.__aexit__(*exc_info)

使用方式 #

# 假设my_lock是你的async lock实例，my_async_session是异步botocore session
async with async_locked_context_entry(my_lock, my_async_session):
    # 此处已进入session上下文，且锁已释放
    # 执行你的业务逻辑，比如调用botocore接口
    await my_async_session.do_something()

同步场景（通用参考） #

如果是同步锁和同步上下文，用contextmanager封装类似逻辑：

from contextlib import contextmanager
import sys

@contextmanager
def locked_context_entry(lock, ctx):
    entered_ctx = False
    exc_info = None
    try:
        with lock:
            ctx.__enter__()
            entered_ctx = True
        yield
    except:
        exc_info = sys.exc_info()
        raise
    finally:
        if entered_ctx:
            ctx.__exit__(*exc_info)

使用方式 #

with locked_context_entry(my_lock, my_context):
    # 业务逻辑
    pass

为什么这比手动实现更优 #

1. 代码复用：避免重复编写try-finally和标记变量的样板代码
2. 异常安全：
   • 锁的释放由自身上下文管理器保证，即使目标上下文进入失败也不会死锁
   • 目标上下文的退出方法会接收完整的异常信息，确保第三方上下文能正确处理资源清理
3. 可读性：符合Python上下文管理器的惯用写法，逻辑清晰

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