python实现redis分布式锁

import time
import redis
from redis.exceptions import WatchError


class RedisUtil():
    def __init__(self):
        self.redis = redis.Redis(host="redis账户url",
                                 port=6379,
                                 password="你的redis密码",
                                 decode_responses=True)
        self.online_set_name = "palm_online_devices"

    # 获取锁
    def acquire_lock(self, key, val, acquire_time=60, time_out=60):
        start_time = time.time()
        # 阻塞等待，轮询去获取锁，获取不到，轮询等待，直到拿到锁或者超时
        while True:
            if time.time() - start_time > acquire_time:
                return False
            # 用setnx实现分布式锁，当key存在时候，setnx不做任何动作，如果key不存在，将value值设置为val
            # 如果 SETNX 返回1，说明该进程获得锁，SETNX将键 lock.foo 的值设置为锁的超时时间（当前时间 + 锁的有效时间）。
            # 如果 SETNX 返回0，说明其他进程已经获得了锁，进程不能进入临界区。进程可以在一个循环中不断地尝试 SETNX 操作，以获得锁
            if self.redis.setnx(name=key, value=val):
                # 设置key的时长为60秒，如果超过60秒，key销毁
                self.redis.expire(key, time_out)
                return True

    # 释放锁
    def release_lock(self, key, val):
        pip = self.redis.pipeline(True)
        while True:
            try:
                pip.watch(key)
                lock_value = self.redis.get(key)
                if not lock_value:
                    return True
                if lock_value == val:
                    pip.multi()
                    pip.delete(key)
                    pip.execute()
                    return True
                pip.unwatch()
                break
            except WatchError:
                pass
        return False