单节点redis中的安全redis锁
safe_redis_lock的Python项目详细描述
项目描述
- 一个单机版的redis分布式锁( A single redis distributed lock)
- 项目使用set命令来加锁
- 使用lua脚本来释放锁以保证原子性
- 注意事项,传入的redis连接池需要必须参数一样,同一个redis数据库
- 当block=False的时候如果没有获得锁则返回一个AcquireFailException异常
- acquire方法返回True时候则说明获得锁成功
使用方法一
装饰器使用方法
新建一个RedisLock对象.使用RedisLock的lock装饰器来装饰需要保证原子性的函数
注意该函数不能是阻塞的
importthreadingimportredisfromsafe_redis_lockimportRedisLockpool=redis.ConnectionPool(host='127.0.0.1',port=6379,password='lyc',db=0)conn=redis.Redis(connection_pool=pool)conn.set('producer',10000)redis_lock_=RedisLock(lock_timeout=5,pool=pool,block=True)@redis_lock_.lockdefconsumer_pp(conn):num=int(conn.get('producer'))ifnum==0:returnTrueelse:print(f'we consumer one left --> {num}')conn.set('producer',num-1)defconsumer():pool=redis.ConnectionPool(host='127.0.0.1',port=6379,password='lyc',db=0)whileTrue:conn=redis.Redis(connection_pool=pool)ifconsumer_pp(conn):breakdefconsumer_thread():fromthreadingimportThreadforiinrange(20):Thread(target=consumer).start()consumer_thread()
使用方法二:
fromsafe_redis_lockimportRedisLock,AcquireFailExceptionimportredisimportthreadingimportospool=redis.ConnectionPool(host='127.0.0.1',port=6379,password='lyc',db=0)conn=redis.Redis(connection_pool=pool)conn.set('producer',10000)redis_lock_=RedisLock(lock_timeout=5,pool=pool,block=True)defconsumer_pp(conn):num=int(conn.get('producer'))ifnum==0:returnTrueelse:print(f'we consumer one left --> {num}')conn.set('producer',num-1)defconsumer():pool=redis.ConnectionPool(host='127.0.0.1',port=6379,password='lyc',db=0)whileTrue:conn=redis.Redis(connection_pool=pool)value=f'{threading.current_thread().ident}-{os.getpid()}'get_lock=Falsetry:get_lock=redis_lock_.acquire(value=value)ifget_lock:ifconsumer_pp(conn):breakexceptAcquireFailException:print('can not get the lock')finally:ifget_lock:redis_lock_.release(value)defconsumer_thread():fromthreadingimportThreadforiinrange(20):Thread(target=consumer).start()consumer_thread()
- 项目
标签: