前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >Python使用functools实现注解同步方法

Python使用functools实现注解同步方法

作者头像
kongxx
发布2018-05-14 14:02:33
1.3K0
发布2018-05-14 14:02:33
举报

在 Python 中没有类似 Java 中使用的 synchronized 关键字来同步方法,因此在 Python 中要实现同步方法,通常我们是使用 threading.Lock() 来实现。在进入函数的地方获取锁,出函数的时候释放锁,这样实现代码看起好非常不好看。另外网上也有人给出了其它几种实现方式,但看起来都不美气。

今天我在做项目的时候突然想到是不是可以通过 functools 来实现通过注解来标注方法为同步方法。

首先要求自己的类中有一个锁对象并且在类初始化的时候初始化这个锁对象,比如:

代码语言:javascript
复制
class MyWorker(object):

    def __init__(self):
        self.lock = threading.Lock()
        ...

    ...

然后创建一个 synchronized 函数,这个函数装饰具体对象的具体方法,将方法放到获取/释放锁之间来运行,如下

代码语言:javascript
复制
def synchronized(func):
    @functools.wraps(func)
    def wrapper(self, *args, **kwargs):
        with self.lock:
            return func(self, *args, **kwargs)
    return wrapper

最后在需要使用同步的方法上使用 @synchronized 来标准方法是同步方法,比如:

代码语言:javascript
复制
@synchronized
def test(self):
    ...

下面是一个完整例子,仅供参考:

代码语言:javascript
复制
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()
本文参与 腾讯云自媒体分享计划,分享自作者个人站点/博客。
原始发表:2018.02.06 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档