定義代碼如下
import redis
import contextlib
import pickle
import os, socket, threading
class RedisLock:
def __init__(self, lock_name, host='', port=6379, db=0):
self.lock_name = lock_name
self.redis = redis.Redis(connection_pool=redis.ConnectionPool(host=host, port=port, db=db))
def acquire_lock(self, lock_id, expire=None):
lock_id = lock_id if lock_id else self.get_lock_id()
return True if self.redis.set(self.lock_name, pickle.dumps(lock_id), nx=True, ex=expire) else False
# Above 1 line code can replace with follow codes to debug
# if self.redis.set(self.lock_name, pickle.dumps(lock_id), nx=True, ex=expire):
# print('Lock Succeed')
# return True
# else:
# print('Lock Failed')
# return False
def release_lock(self, lock_id=None):
lock_id = lock_id if lock_id else self.get_lock_id()
if lock_id == pickle.loads(self.redis.get(self.lock_name)):
self.redis.delete(self.lock_name)
# print('Unlock Succeed')
return True
else:
# print('Unlock Failed')
return False
@contextlib.contextmanager
def lock(self, lock_id=None, expire=None):
if not self.acquire_lock(lock_id, expire):
exit(0)
yield self
self.release_lock(lock_id)
def get_lock_id(self):
""" hostname+processID+threadName"""
return f'{socket.gethostname()}{os.getpid()}{threading.current_thread().name}'
調用代碼如下
redis_lock = RedisLock('lockname', host='Your IP') # 第一個匿名參數必傳,作為 redis的key
with redis_lock.lock() as lock:
print('You Can Do Something Here')
注意說明
1. 注釋部分是我寫的時候,調試用的代碼,最后寫完的時候都替換為簡潔的語法.
2. 因為鎖具有互斥特性, 所以選擇 set() 的 nx 參數來實現,
nx參數:我個人一直這樣記(讀作 not exist) ====> 不存在
理解:
不存在則添加,存在就不添加了。
舉一反三,沒有鎖就加個鎖。有鎖就不加鎖了。
3. set方法 的 ex 參數, 可代替expire方法的來設置過期時間
4. redis 有 很多指令變形。 比如 set(nx=, ex=) 可拆分為 setnx setex
但 "set() 這種指令更優", 能用就盡量用, 理由如下:
"set指令 好處是 set具有原子性", 避免了解決資源競爭的同時引發自身可能出現的資源競爭
5. 我使用裝飾器版 的 上下文管理器,對代碼做了封裝, 所以調用時,用"with語句"即可
6. with redis_lock.lock() as lock ,"lock() 這里可以自己指定2個參數":
lock_id=None # 這是區分不同線程的唯一標識符,默認為(主機名+進程ID+線程名),可自傳
expire=None # 過期時間,秒為單位
7. 需要注意一個點,與redis通信是以二進制形式。 所以我在代碼內部對 lock_id 做了"pickle序列化"
當然如果是字符串用 encode() 與 decode() 來實現也是可以的。
更多文章、技術交流、商務合作、聯系博主
微信掃碼或搜索:z360901061

微信掃一掃加我為好友
QQ號聯系: 360901061
您的支持是博主寫作最大的動力,如果您喜歡我的文章,感覺我的文章對您有幫助,請用微信掃描下面二維碼支持博主2元、5元、10元、20元等您想捐的金額吧,狠狠點擊下面給點支持吧,站長非常感激您!手機微信長按不能支付解決辦法:請將微信支付二維碼保存到相冊,切換到微信,然后點擊微信右上角掃一掃功能,選擇支付二維碼完成支付。
【本文對您有幫助就好】元
