58 lines
1.7 KiB
Python
58 lines
1.7 KiB
Python
import uuid
|
|
import logging
|
|
from django.conf import settings
|
|
from django.core.cache import cache
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
# Redis客户端（延迟初始化）
|
|
_redis_client = None
|
|
|
|
def _get_redis_client():
|
|
global _redis_client
|
|
if _redis_client is None:
|
|
import redis
|
|
_redis_client = redis.Redis(
|
|
host=settings.REDIS_HOST,
|
|
port=settings.REDIS_PORT,
|
|
password=settings.REDIS_PASSWORD,
|
|
db=settings.REDIS_DB_CELERY,
|
|
decode_responses=True,
|
|
)
|
|
return _redis_client
|
|
|
|
|
|
def acquire_lock(lock_key, timeout=10):
|
|
"""获取分布式锁，返回锁标识符，失败返回None"""
|
|
client = _get_redis_client()
|
|
identifier = str(uuid.uuid4())
|
|
result = client.set(lock_key, identifier, nx=True, ex=timeout)
|
|
if result:
|
|
logger.debug(f"获取锁成功: {lock_key}, 标识: {identifier}")
|
|
return identifier
|
|
return None
|
|
|
|
|
|
def release_lock(lock_key, identifier):
|
|
"""释放分布式锁（仅当标识符匹配时才释放）"""
|
|
if identifier is None:
|
|
return False
|
|
client = _get_redis_client()
|
|
lua_script = """
|
|
if redis.call('get', KEYS[1]) == ARGV[1] then
|
|
return redis.call('del', KEYS[1])
|
|
else
|
|
return 0
|
|
end
|
|
"""
|
|
try:
|
|
result = client.eval(lua_script, 1, lock_key, identifier)
|
|
if result:
|
|
logger.debug(f"释放锁成功: {lock_key}")
|
|
else:
|
|
logger.warning(f"释放锁失败（标识不匹配或锁已过期）: {lock_key}")
|
|
return bool(result)
|
|
except Exception as e:
|
|
logger.error(f"释放锁异常: {lock_key}, 错误: {e}")
|
|
return False
|