1

在 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.

只要我不抓住 a ,它就很好用WatchError,然后我得到

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

我究竟做错了什么?

4

1 回答 1

5

我认为问题在于您多次屈服(重复任务时),但上下文管理器只输入一次(这yield只是该__enter__方法的语法糖)。所以一旦yield可以多次执行,你就有问题了。

我不确定如何以一种好的方式解决这个问题,我也无法测试它,所以我只是给出一些建议。

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

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

当然,这仍然不能解决多重收益,并且您不能更早地产生该对象,因为此时您也不会拥有原始值。因此,我们可以指定一个负责更新值的委托。

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

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

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

update(key, lambda value: value + 42)
于 2013-11-03T14:52:31.630 回答