82 lines
2.4 KiB
Python
82 lines
2.4 KiB
Python
|
# -*- coding: utf-8 -*-
|
|||
|
|
"""
|
|||
|
|
Redis连接管理器
|
|||
|
|
统一管理所有Redis连接，避免重复连接
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
import redis
|
|||
|
|
import logging
|
|||
|
|
import threading
|
|||
|
|
from typing import Optional
|
|||
|
|
|
|||
|
|
logger = logging.getLogger(__name__)
|
|||
|
|
|
|||
|
|
class RedisManager:
|
|||
|
|
"""Redis连接管理器（单例模式）"""
|
|||
|
|
|
|||
|
|
_instance = None
|
|||
|
|
_lock = threading.Lock()
|
|||
|
|
|
|||
|
|
def __new__(cls):
|
|||
|
|
if cls._instance is None:
|
|||
|
|
with cls._lock:
|
|||
|
|
if cls._instance is None:
|
|||
|
|
cls._instance = super().__new__(cls)
|
|||
|
|
cls._instance._initialized = False
|
|||
|
|
return cls._instance
|
|||
|
|
|
|||
|
|
def __init__(self):
|
|||
|
|
if self._initialized:
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
self.redis_client = None
|
|||
|
|
self.connected = False
|
|||
|
|
self.connection_lock = threading.Lock()
|
|||
|
|
self._initialized = True
|
|||
|
|
|
|||
|
|
# Redis配置
|
|||
|
|
self.config = {
|
|||
|
|
'host': '43.134.68.207',
|
|||
|
|
'port': 6379,
|
|||
|
|
'password': '123456',
|
|||
|
|
'decode_responses': True,
|
|||
|
|
'socket_connect_timeout': 2,
|
|||
|
|
'socket_timeout': 2,
|
|||
|
|
'retry_on_timeout': True
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
def get_connection(self) -> Optional[redis.Redis]:
|
|||
|
|
"""获取Redis连接（懒加载）"""
|
|||
|
|
with self.connection_lock:
|
|||
|
|
if not self.connected:
|
|||
|
|
try:
|
|||
|
|
self.redis_client = redis.Redis(**self.config)
|
|||
|
|
self.redis_client.ping()
|
|||
|
|
self.connected = True
|
|||
|
|
logger.info("Redis连接成功")
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.debug(f"Redis连接失败: {e}")
|
|||
|
|
self.redis_client = None
|
|||
|
|
self.connected = False
|
|||
|
|
|
|||
|
|
return self.redis_client
|
|||
|
|
|
|||
|
|
def is_connected(self) -> bool:
|
|||
|
|
"""检查Redis是否已连接"""
|
|||
|
|
return self.connected and self.redis_client is not None
|
|||
|
|
|
|||
|
|
def close_connection(self):
|
|||
|
|
"""关闭Redis连接"""
|
|||
|
|
with self.connection_lock:
|
|||
|
|
if self.redis_client:
|
|||
|
|
try:
|
|||
|
|
self.redis_client.close()
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.debug(f"关闭Redis连接失败: {e}")
|
|||
|
|
finally:
|
|||
|
|
self.redis_client = None
|
|||
|
|
self.connected = False
|
|||
|
|
|
|||
|
|
# 全局Redis管理器实例
|
|||
|
|
redis_manager = RedisManager()
|