如果非阻塞或超时，则获取多处理。在 with 语句中锁定

Question

使用普通的 multiprocessing.Lock （或 threading.Lock），您可以将以下代码简化为：

lock = multiprocessing.Lock()

lock.acquire()

try:
    ...
finally:
    lock.release()

然而

with lock:
    ...

，当我想传递一些内容时，我仍然可以使用上下文管理器吗？ lock.acquire(...) 的参数，例如 block= 或 timeout=？例如，我有这样的代码：

lock_success = lock.acquire(block=False)

if not lock_success:
    return

try:
    ...
finally:
    lock.release()

我没有找到将此参数传递给上下文管理器的方法（因为它位于 acquire 调用中，而不是构造函数中）。

（这个想法是，如果无法获取锁，with 块将被跳过。）

类似地，threading.Lock 也提供了类似的 API。

Answer 1

太长了；您不能使用内置的锁定上下文管理器，但它仍然可以相当干净地完成。

它几乎可以工作，因为 Lock.__enter__() 确实返回了从调用 acquire() 中返回的值，这应该是获取锁成功或失败的布尔值。

l = Lock()
with l as success:
    if success:
        print("run some code")
    else:
        print("skip the code")

然而令人沮丧的是，无法将参数传递给内部调用acquire（硬编码参数此处）。我建议您编写自己的上下文管理器来解决这个问题，因为它非常简单：

from multiprocessing import Lock
from contextlib import contextmanager

@contextmanager
def get_lock(lock, block=True, timeout=None):
    held = lock.acquire(block=block, timeout=timeout)
    try:
        yield held
    finally:
        if held:
            lock.release()

#### example usage ####

l = Lock()
#lock should be acquired so success == True
with get_lock(l) as success:
    if success:
        print("run some code")
    else:
        print("skip the code")

l.acquire()
#lock won't be acquired and block will proceed after timeout
#use the value of success to determine what to do
with get_lock(l, True, 3) as success:
    if success:
        print("run some code")
    else:
        print("skip the code")

l.release()

Answer 2

我怀疑这是不可能的，因为 Python 中上下文管理器的设计方式不同。上下文管理器 (CM) 发生的情况是：

with CM:
    ...

• CM 的 __enter__ 方法被调用
• with 内部的块被运行
• CM 的 __exit__ 方法是因此

，没有办法“跳过”with 块的内部，所以我想这解释了 Lock 类的设计。尽管有一些黑暗魔法方法解决这个问题（应该避免）。