高并发下redis 分布式锁 实现示例

当我们需要在高并发环境下实现分布式锁时，可以考虑使用Redis实现，以下是一个基于Redis的分布式锁实现示例：

import redis
import time

class RedisLock(object):
    def __init__(self, redis_conn, lock_name, timeout=10):
        self.conn = redis_conn
        self.lock_name = lock_name
        self.timeout = timeout
        self.lock_val = None

    def acquire(self):
        while True:
            self.lock_val = str(time.time()) + str(int(time.clock() * 1000000))
            if self.conn.setnx(self.lock_name, self.lock_val):
                self.conn.expire(self.lock_name, self.timeout)
                return True
            elif self.conn.ttl(self.lock_name) < 0:
                self.conn.expire(self.lock_name, self.timeout)
            time.sleep(0.1)

    def release(self):
        if self.conn.get(self.lock_name) == self.lock_val:
            self.conn.delete(self.lock_name)

if __name__ == '__main__':
    redis_conn = redis.Redis(host='localhost', port=6379, db=0)
    lock = RedisLock(redis_conn, 'test_lock')
    if lock.acquire():
        try:
            # 这里写需要加锁的代码
            pass
        finally:
            lock.release()

在上面的代码中，我们定义了一个`RedisLock`类，其中`acquire()`方法用于获取分布式锁，`release()`方法用于释放分布式锁。当获取锁时，使用`setnx()`方法在Redis中创建一个名为`lock_name`的键值对，并设置键的过期时间为`timeout`。如果获取不到锁，就等待一段时间后再次尝试。当释放锁时，判断之前设置的值是否与当前的值一致，如果一致则删除键值对。

使用时只需要创建一个`RedisLock`对象，调用`acquire()`方法获取锁，然后执行需要加锁的代码，最后调用`release()`方法释放锁即可。

需要注意的是，由于Redis是内存数据库，如果在获取锁和释放锁之间服务器宕机，那么就会出现死锁情况。因此，需要结合心跳机制和超时机制来保证锁的可靠性。