python 中的互斥装饰器

Question

我看到了这个： http://yukelzon.blogspot.com/2005/07/python -locks.html 在寻找解决方案以向现有代码添加一些锁定以写入文件时。

满足我需要的简化版本如下所示：

  def mlock(orig):
    def inner(*args, **kwargs):
      Coloring.lock.acquire()
      try:
        ret = orig(*args, **kwargs)
        return ret
      finally:
        Coloring.lock.release()
    return inner

锁是一个类变量。任何人都可以想到改进或更好的方法吗？

Answer 1

如果您使用的是 Python 2.6+（我认为），互斥对象是上下文管理器，因此：

def mlock(f):
    def inner(*args, **kwargs):
        with Coloring.lock:
            return f(*args, **kwargs)
    return inner

Answer 2

我不喜欢自由变量。我可能会让锁成为装饰器的显式变量，如下所示：

def mlock(Coloring):
    def mlock_decorator(orig):
        def inner(*args, **kwargs):
          Coloring.lock.acquire()
          try:
            ret = orig(*args, **kwargs)
            return ret
          finally:
            Coloring.lock.release()
        return inner
    return mlock_decorator

好吧，因为 Coloring 是一个类变量，如果您希望它非常特定于该类，请显式访问它：

def mlock(orig):
    def inner(self, *args, **kwargs):
      self.Coloring.lock.acquire()
      try:
        ret = orig(self, *args, **kwargs)
        return ret
      finally:
        self.Coloring.lock.release()
    return inner

或者，至少声明该锁是全局的

def mlock(orig):
    global Coloring
    def inner(*args, **kwargs):
      Coloring.lock.acquire()
      try:
        ret = orig(*args, **kwargs)
        return ret
      finally:
        Coloring.lock.release()
    return inner