133 lines
3.9 KiB
Python
133 lines
3.9 KiB
Python
import logging
|
|
from typing import Optional
|
|
from redis.asyncio import Redis, from_url as async_from_url
|
|
from app.config.settings import settings
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
class RedisClient:
|
|
def __init__(self):
|
|
self._redis: Optional[Redis] = None
|
|
|
|
async def connect(self):
|
|
"""
|
|
Initializes the Redis connection pool (Async).
|
|
"""
|
|
if settings.REDIS_ENABLED:
|
|
try:
|
|
self._redis = async_from_url(
|
|
settings.REDIS_URL,
|
|
encoding="utf-8",
|
|
decode_responses=True,
|
|
max_connections=settings.REDIS_MAX_CONNECTIONS,
|
|
socket_timeout=settings.REDIS_TIMEOUT
|
|
)
|
|
ping = await self._redis.ping()
|
|
if ping:
|
|
logger.info("Connected to Redis (Async)")
|
|
except Exception as e:
|
|
logger.error(f"Failed to connect to Redis: {e}")
|
|
self._redis = None
|
|
else:
|
|
logger.info("Redis is disabled in settings")
|
|
|
|
async def close(self):
|
|
"""
|
|
Closes the Redis connection.
|
|
"""
|
|
if self._redis:
|
|
await self._redis.close()
|
|
logger.info("Redis connection closed")
|
|
|
|
async def get(self, key: str) -> Optional[str]:
|
|
"""
|
|
Get a value by key.
|
|
"""
|
|
if not self._redis:
|
|
return None
|
|
try:
|
|
return await self._redis.get(key)
|
|
except Exception as e:
|
|
logger.error(f"Redis GET error for key {key}: {e}")
|
|
return None
|
|
|
|
async def set(self, key: str, value: str, expire: int = None) -> bool:
|
|
"""
|
|
Set a value by key with optional expiration time in seconds.
|
|
"""
|
|
if not self._redis:
|
|
return False
|
|
try:
|
|
return await self._redis.set(key, value, ex=expire)
|
|
except Exception as e:
|
|
logger.error(f"Redis SET error for key {key}: {e}")
|
|
return False
|
|
|
|
async def delete(self, key: str) -> bool:
|
|
"""
|
|
Delete a value by key.
|
|
"""
|
|
if not self._redis:
|
|
return False
|
|
try:
|
|
return await self._redis.delete(key) > 0
|
|
except Exception as e:
|
|
logger.error(f"Redis DELETE error for key {key}: {e}")
|
|
return False
|
|
|
|
@property
|
|
def client(self) -> Optional[Redis]:
|
|
"""
|
|
Expose the raw Redis client if needed for advanced operations.
|
|
"""
|
|
return self._redis
|
|
|
|
class SyncRedisClient:
|
|
def __init__(self):
|
|
self._redis = None
|
|
|
|
def connect(self):
|
|
"""
|
|
Initializes the Redis connection pool (Sync).
|
|
"""
|
|
if settings.REDIS_ENABLED:
|
|
try:
|
|
import redis
|
|
self._redis = redis.from_url(
|
|
settings.REDIS_URL,
|
|
encoding="utf-8",
|
|
decode_responses=True,
|
|
max_connections=settings.REDIS_MAX_CONNECTIONS,
|
|
socket_timeout=settings.REDIS_TIMEOUT
|
|
)
|
|
if self._redis.ping():
|
|
logger.info("Connected to Redis (Sync)")
|
|
except Exception as e:
|
|
logger.error(f"Failed to connect to Redis (Sync): {e}")
|
|
self._redis = None
|
|
|
|
def close(self):
|
|
if self._redis:
|
|
self._redis.close()
|
|
logger.info("Redis connection closed (Sync)")
|
|
|
|
def rpush(self, key: str, *values) -> int:
|
|
if not self._redis:
|
|
try:
|
|
self.connect()
|
|
except:
|
|
pass
|
|
if not self._redis:
|
|
return 0
|
|
try:
|
|
return self._redis.rpush(key, *values)
|
|
except Exception as e:
|
|
logger.error(f"Redis RPUSH error: {e}")
|
|
return 0
|
|
|
|
@property
|
|
def client(self):
|
|
return self._redis
|
|
|
|
redis_client = RedisClient()
|
|
sync_redis_client = SyncRedisClient() |