Python使用functools實現(xiàn)注解同步方法

更新時間：2018年02月06日 09:42:45 作者：kongxx

這篇文章主要介紹了Python使用functools實現(xiàn)注解同步方法,非常不錯，具有參考借鑒價值，需要的朋友可以參考下

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

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

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

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

然后創(chuàng)建一個 synchronized 函數(shù)，這個函數(shù)裝飾具體對象的具體方法，將方法放到獲取/釋放鎖之間來運行，如下

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

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

@synchronized
def test(self):
  ...

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

import threading
import functools
import time
def synchronized(func):
  @functools.wraps(func)
  def wrapper(self, *args, **kwargs):
    with self.lock:
      return func(self, *args, **kwargs)
  return wrapper
class 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()

總結

以上所述是小編給大家介紹的Python使用functools實現(xiàn)注解同步方法，希望對大家有所幫助，如果大家有任何疑問請給我留言，小編會及時回復大家的。在此也非常感謝大家對腳本之家網(wǎng)站的支持！

您可能感興趣的文章: