ITPub博客

首页 > 数据库 > NoSQL > 利用Python+Redis实现分布式锁

利用Python+Redis实现分布式锁

原创 NoSQL 作者:ittangzeng 时间:2019-02-23 00:45:23 0 删除 编辑

利用Python+Redis实现分布式锁

欢迎一起交流技术:一个人可以走的很快,但是一群人却可以走的更远。

常见关系型数据库(Oracle+MySQL+PG+GP)+高可用企业解决方案+分布式锁

使用场景:

  彻底解决HA的脑裂问题

  多个服务器之间只允许一个进程执行的场景



class MyDLock(object):

    def __init__(self, lockID,timeout):

        self.connection = redis.Redis(host=cfg.REDIS_SERVER_IP

                                      , port=cfg.REDIS_SERVER_PORT

                                      , password=cfg.REDIS_SERVER_PWD

                                      , db=cfg.REDIS_SERVER_DB_NUM

                                      ,decode_responses=True)

        self.__lockID = lockID

        self.__timeout  = timeout


    def tryLock(self, appid):

        #To_Main

        try:

            val=self.connection.get(self.__lockID)

            if val is None:

                logging.info("The app(%s) try lock(%s) ok." % (appid, self.__lockID));

                self.connection.set(self.__lockID, appid, ex=self.__timeout)

                return True

            else:

                logging.info("The owner of lock(%s) is %s. you(%s) can not get it"%(self.__lockID, val, appid));

                return False

        except Exception as e:

            logging.error("Warning: Can't write log. (%s)" % e)

            return False




    def activeLock(self, appid):

        #心跳,定期激活

        val = self.connection.get(self.__lockID)

        if val is None:

            logging.info("There is no lock(%s)." % (self.__lockID));

            return False

        else:

            if appid == val:

                #只能激活自己的锁

                logging.info("Application(%s) active lock(%s) ok." % (appid, self.__lockID));

                self.connection.set(self.__lockID, appid, ex=self.__timeout)

                return True

            else:

                #不能激活非自己的锁

                logging.info("Application(%s) can not active lock(%s). The owner is (%s)" % (appid, self.__lockID, val));

                return False


    def releaseLock(self, key, appid):

        val = self.connection.get(self.__lockID)

        if val is None:

            return False

        else:

            if appid == val:

                # 只能删除自己的锁

                self.connection.delete(self.__lockID)

                return True

            else:

                # 不能删除非自己的锁

                return False


来自 “ ITPUB博客 ” ,链接:http://blog.itpub.net/29497382/viewspace-2636751/,如需转载,请注明出处,否则将追究法律责任。

请登录后发表评论 登录
全部评论
欢迎一起交流技术:一个人可以走的很快,但是一群人却可以走的更远。 常见关系型数据库(Oracle+MySQL+PG+GP)+高可用+分布式锁 即时沟通工具:19578602(注明交流内容)

注册时间:2014-03-29

  • 博文量
    32
  • 访问量
    62577