61 lines
1.8 KiB
Python
61 lines
1.8 KiB
Python
import redis
|
|
from .config_loader import global_config as config
|
|
|
|
class RedisManager:
|
|
"""
|
|
Redis 连接管理器
|
|
"""
|
|
_pool = None
|
|
_client = None
|
|
|
|
@classmethod
|
|
def initialize(cls):
|
|
"""
|
|
初始化 Redis 连接并进行健康检查
|
|
"""
|
|
if cls._pool is None:
|
|
try:
|
|
host = config.redis['host']
|
|
port = config.redis['port']
|
|
db = config.redis['db']
|
|
password = config.redis.get('password')
|
|
|
|
print(f" 正在尝试连接 Redis: {host}:{port}, DB: {db}")
|
|
|
|
cls._pool = redis.ConnectionPool(
|
|
host=host,
|
|
port=port,
|
|
db=db,
|
|
password=password,
|
|
decode_responses=True
|
|
)
|
|
cls._client = redis.Redis(connection_pool=cls._pool)
|
|
if cls._client.ping():
|
|
print(" Redis 连接成功！")
|
|
else:
|
|
print(" Redis 连接失败: PING 命令无响应")
|
|
except redis.exceptions.ConnectionError as e:
|
|
print(f" Redis 连接失败: {e}")
|
|
cls._pool = None
|
|
cls._client = None
|
|
except Exception as e:
|
|
print(f" Redis 初始化时发生未知错误: {e}")
|
|
cls._pool = None
|
|
cls._client = None
|
|
|
|
@classmethod
|
|
def get_redis(cls):
|
|
"""
|
|
获取 Redis 连接
|
|
|
|
:return: Redis 连接实例
|
|
"""
|
|
if cls._client is None:
|
|
# 理论上 initialize 应该在程序启动时被调用，这里作为备用
|
|
cls.initialize()
|
|
return cls._client
|
|
|
|
# 在模块加载时直接初始化
|
|
RedisManager.initialize()
|
|
redis_client = RedisManager.get_redis()
|