70 lines
1.8 KiB
Python
70 lines
1.8 KiB
Python
"""
|
||
Redis客户端
|
||
"""
|
||
import redis
|
||
from app.core.config import settings
|
||
import logging
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
_redis_client = None
|
||
|
||
def get_redis_client():
|
||
"""
|
||
获取Redis客户端（单例模式）
|
||
|
||
Returns:
|
||
redis.Redis: Redis客户端实例，如果连接失败则返回None
|
||
"""
|
||
global _redis_client
|
||
|
||
if _redis_client is not None:
|
||
try:
|
||
# 测试连接
|
||
_redis_client.ping()
|
||
return _redis_client
|
||
except:
|
||
# 连接已断开，重新创建
|
||
_redis_client = None
|
||
|
||
try:
|
||
redis_url = getattr(settings, 'REDIS_URL', None)
|
||
if not redis_url:
|
||
logger.warning("REDIS_URL未配置，无法使用Redis缓存")
|
||
return None
|
||
|
||
# 解析Redis URL: redis://host:port/db
|
||
if redis_url.startswith('redis://'):
|
||
redis_url = redis_url.replace('redis://', '')
|
||
|
||
# 分离host:port和db
|
||
parts = redis_url.split('/')
|
||
host_port = parts[0]
|
||
db = int(parts[1]) if len(parts) > 1 and parts[1].isdigit() else 0
|
||
|
||
# 分离host和port
|
||
if ':' in host_port:
|
||
host, port = host_port.split(':')
|
||
port = int(port)
|
||
else:
|
||
host = host_port
|
||
port = 6379
|
||
|
||
_redis_client = redis.Redis(
|
||
host=host,
|
||
port=port,
|
||
db=db,
|
||
decode_responses=True, # 自动解码为字符串
|
||
socket_connect_timeout=2,
|
||
socket_timeout=2
|
||
)
|
||
|
||
# 测试连接
|
||
_redis_client.ping()
|
||
logger.info(f"Redis连接成功: {host}:{port}/{db}")
|
||
return _redis_client
|
||
except Exception as e:
|
||
logger.warning(f"Redis连接失败: {str(e)}，将使用内存缓存")
|
||
_redis_client = None
|
||
return None
|