使用Redis实现分布式可重入锁

本文主要是介绍使用Redis实现分布式可重入锁，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

在分布式应用中经常需要用到分布式锁。
redis使用set命令来实现分布式锁
SET key value [EX seconds] [PX milliseconds] [NX|XX]
直接使用该命令构建的redis分布式锁是不像jdk的reentrantlock具有可重入性的，使用线程的ThreadLocal变量存储当前持有锁的计数，可以实现redis分布式锁的可重入性。
另外Redis分布式锁有超时的问题，不要用于耗时较长的任务.如果真的有耗时较长的任务需要锁，建议使用数据库的乐观锁来解决。

import redis
import threadingHOST = '127.0.0.1'
PORT = '6379'
PASSWORD = ''pool = redis.ConnectionPool(host=HOST, port=PORT, password=PASSWORD, max_connections=1024)
conn = redis.Redis(connection_pool=pool)locks = threading.local()
locks.redis = {}def key_for(user_id):return "account_{}".format(user_id)def _lock(client, key):print("redis lock ")return bool(client.set(name=key, value="1", nx=True, ex=5))def _unlock(client, key):print("redis un lock")client.delete(key)def lock(client, user_id):key = key_for(user_id)if key in locks.redis:locks.redis[key] += 1return Trueok = _lock(client, key)if not ok:return Falselocks.redis[key] = 1return Truedef unlock(client, user_id):key = key_for(user_id)if key in locks.redis:locks.redis[key] -= 1if locks.redis[key] <= 0:del locks.redis[key]_unlock(client, key)return Truereturn Falseprint(lock(conn, "codehole"))
print(lock(conn, "codehole"))
print(unlock(conn, "codehole"))
print(unlock(conn, "codehole"))