Python 基于Redis实现一个简单的分布式锁

2023-02-17 21:07:13
# redis_lock.py
import redis
import time
import threading

# 连接池方式
pool = redis.ConnectionPool(host='127.0.0.1',port=6379)
redis_con = redis.Redis(connection_pool=pool)


class RedisLock(object):
    def __init__(self):
        self.redis_con = redis_con

    def get_lock(self, val):
        while True:
            res = self.redis_con.set('lock', val,nx=True, ex=3 * 60)
            if res:
                break
            # print('继续等待锁...')
            time.sleep(0.1)

    def del_lock(self, val):
        old_val = redis_con.get('lock')
        if old_val == val.encode():
            self.redis_con.delete('lock')
            print('锁释放成功')
SUMS = 0

def test_lock(lock, name, num, val):
    try:
        print('%s 开始工作' % name)
        print('%s 准备获取锁并加锁' % name)
        lock.get_lock(val)
        print('%s 得到锁,继续工作' % name)
        global SUMS
        SUMS += 15
        time.sleep(num)
        print(SUMS)
    except Exception as e:
        print('发生异常:%s' % str(e))
    finally:
        print('%s 操作完成,准备释放锁'%name)
        lock.del_lock(val)


if __name__ == '__main__':
    start_time = time.time()
    r_lock = RedisLock()
    tasks = []
    for num in range(1,4):
        t = threading.Thread(target=test_lock, args=(r_lock, '任务%d'%num,num,'lock%d'%num))
        tasks.append(t)
        t.start()
    [item.join() for item in tasks]
    print('总耗时:', time.time() - start_time)

运行效果:

$ python redis_lock.py
任务1 开始工作
任务2 开始工作
任务2 准备获取锁并加锁
任务3 开始工作
任务1 准备获取锁并加锁
任务3 准备获取锁并加锁
任务2 得到锁,继续工作
15
任务2 操作完成,准备释放锁
锁释放成功
任务3 得到锁,继续工作
30
任务3 操作完成,准备释放锁
锁释放成功
任务1 得到锁,继续工作
45
任务1 操作完成,准备释放锁
锁释放成功
总耗时: 6.061553478240967
  • 作者:haeasringnar
  • 原文链接:https://blog.csdn.net/haeasringnar/article/details/106693255
    更新时间:2023-02-17 21:07:13