54 lines
1.6 KiB
Python
54 lines
1.6 KiB
Python
|
import redis
|
|||
|
|
from loguru import logger
|
|||
|
|
from config.settings import REDIS_CONFIG
|
|||
|
|
|
|||
|
|
class RedisClient:
|
|||
|
|
"""Redis客户端管理器"""
|
|||
|
|
|
|||
|
|
_instance = None
|
|||
|
|
|
|||
|
|
def __new__(cls):
|
|||
|
|
if cls._instance is None:
|
|||
|
|
cls._instance = super().__new__(cls)
|
|||
|
|
cls._instance._initialized = False
|
|||
|
|
return cls._instance
|
|||
|
|
|
|||
|
|
def __init__(self):
|
|||
|
|
if not self._initialized:
|
|||
|
|
self._pool = None
|
|||
|
|
self._client = None
|
|||
|
|
self._initialized = True
|
|||
|
|
|
|||
|
|
@property
|
|||
|
|
def client(self):
|
|||
|
|
"""获取Redis客户端（懒加载）"""
|
|||
|
|
if self._client is None:
|
|||
|
|
self._init_connection_pool()
|
|||
|
|
return self._client
|
|||
|
|
|
|||
|
|
def _init_connection_pool(self):
|
|||
|
|
"""初始化连接池"""
|
|||
|
|
try:
|
|||
|
|
self._pool = redis.ConnectionPool(
|
|||
|
|
host=REDIS_CONFIG['host'],
|
|||
|
|
port=REDIS_CONFIG['port'],
|
|||
|
|
db=REDIS_CONFIG['db'],
|
|||
|
|
decode_responses=REDIS_CONFIG['decode_responses'],
|
|||
|
|
max_connections=REDIS_CONFIG['max_connections'],
|
|||
|
|
socket_keepalive=True
|
|||
|
|
)
|
|||
|
|
self._client = redis.Redis(connection_pool=self._pool)
|
|||
|
|
|
|||
|
|
# 测试连接
|
|||
|
|
self._client.ping()
|
|||
|
|
logger.info("Redis连接池初始化成功")
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"Redis连接失败: {e}")
|
|||
|
|
raise
|
|||
|
|
|
|||
|
|
def close(self):
|
|||
|
|
"""关闭连接池"""
|
|||
|
|
if self._pool:
|
|||
|
|
self._pool.disconnect()
|
|||
|
|
logger.info("Redis连接池已关闭")
|