Python使用functools實現註解同步方法__Python

在 Python 中沒有類似 Java 中使用的 synchronized 關鍵字來同步方法，因此在 Python 中要實現同步方法，通常我們是使用 threading.Lock() 來實現。在進入函數的地方擷取鎖，出函數的時候釋放鎖，這樣實現代碼看起好非常不好看。另外網上也有人給出了其它幾種實現方式，但看起來都不美氣。

今天我在做項目的時候突然想到是不是可以通過 functools 來實現通過註解來標註方法為同步方法。

首先要求自己的類中有一個鎖對象並且在類初始化的時候初始化這個鎖對象，比如：

class MyWorker(object):    def __init__(self):        self.lock = threading.Lock()        ...    ...

然後建立一個 synchronized 函數，這個函數裝飾具體對象的具體方法，將方法放到擷取/釋放鎖之間來運行，如下

def synchronized(func):    @functools.wraps(func)    def wrapper(self, *args, **kwargs):        with self.lock:            return func(self, *args, **kwargs)    return wrapper

最後在需要使用同步的方法上使用 @synchronized 來標準方法是同步方法，比如：

@synchronizeddef test(self):    ...

下面是一個完整例子，僅供參考：

import threadingimport functoolsimport timedef synchronized(func):    @functools.wraps(func)    def wrapper(self, *args, **kwargs):        with self.lock:            return func(self, *args, **kwargs)    return wrapperclass MyWorker(object):    def __init__(self):        self.lock = threading.Lock()        self.idx = 0    @synchronized    def test1(self):        for i in range(1, 11):            self.idx = self.idx + 1            print "Test1: " + str(self.idx)            time.sleep(1)    @synchronized    def test2(self):        for i in range(1, 11):            self.idx = self.idx + 1            print "Test2: " + str(self.idx)            time.sleep(1)    @synchronized    def test3(self):        for i in range(1, 11):            self.idx = self.idx + 1            print "Test3: " + str(self.idx)            time.sleep(1)worker = MyWorker()threading.Thread(target=worker.test1).start()threading.Thread(target=worker.test2).start()threading.Thread(target=worker.test3).start()