python - 我如何将其写为上下文管理器？

Question

在redis中更新变量的无竞争条件的方式是:

r = redis.Redis()
with r.pipeline() as p:
    while 1:
        try:
            p.watch(KEY)
            val = p.get(KEY)
            newval = int(val) + 42
            p.multi()
            p.set(KEY, newval)
            p.execute()  # raises WatchError if anyone else changed KEY
            break
        except redis.WatchError:
            continue  # retry

这比直接版本(包含竞争条件)复杂得多:

r = redis.Redis()
val = r.get(KEY)
newval = int(val) + 42
r.set(KEY, newval) 

所以我认为上下文管理器会使它更容易使用，但是，我遇到了问题...

我最初的想法是

with update(KEY) as val:
    newval = val + 42
    somehow return newval to the contextmanager...?

没有明显的方法来完成最后一行，所以我尝试了::

@contextmanager
def update(key, cn=None):
    """Usage::

            with update(KEY) as (p, val):
                newval = int(val) + 42
                p.set(KEY, newval)

    """
    r = cn or redis.Redis()
    with r.pipeline() as p:
        while 1:
            try:
                p.watch(key)  # --> immediate mode
                val = p.get(key)
                p.multi()  # --> back to buffered mode
                yield (p, val)
                p.execute()  # raises WatchError if anyone has changed `key`
                break  # success, break out of while loop
            except redis.WatchError:
                pass  # someone else got there before us, retry.

只要我没有捕捉到 WatchError 就很好用, 然后我得到

  File "c:\python27\Lib\contextlib.py", line 28, in __exit__
    raise RuntimeError("generator didn't stop")
RuntimeError: generator didn't stop

我做错了什么？

Answer 1

我认为问题是你多次让出(当任务重复时)但上下文管理器只输入一次(yield 只是 __enter__< 的语法糖 方法)。所以一旦 yield 可以执行多次，你就有问题了。

我不是很清楚如何解决这个问题，我也无法测试，所以我只是给出一些建议。

首先，我会避免产生相当内部的p；您应该产生一些专门为更新过程制作的对象。例如这样的事情:

with update(KEY) as updater:
    updater.value = int(updater.original) + 42

当然，这仍然不能解决多次 yield 的问题，而且您不能更早地 yield 该对象，因为此时您也不会有原始值。因此，我们可以指定一个负责值更新的委托(delegate)。

with update(KEY) as updater:
    updater.process = lambda value: value + 42

这会在生成的对象中存储一个函数，然后您可以在上下文管理器中使用它来继续尝试更新值，直到成功为止。在进入 while 循环之前，您可以尽早从上下文管理器中产生该更新程序。

当然，如果您已经做到了这一点，那么实际上就不需要上下文管理器了。相反，您可以只创建一个函数:

update(key, lambda value: value + 42)