37 lines
745 B
Python
37 lines
745 B
Python
from collections.abc import AsyncGenerator
|
|
|
|
import redis.asyncio as redis
|
|
|
|
from app.config import get_settings
|
|
|
|
_settings = get_settings()
|
|
_redis: redis.Redis | None = None
|
|
|
|
|
|
async def init_redis() -> redis.Redis:
|
|
global _redis
|
|
_redis = redis.from_url(
|
|
_settings.redis_url,
|
|
encoding="utf-8",
|
|
decode_responses=True,
|
|
)
|
|
await _redis.ping()
|
|
return _redis
|
|
|
|
|
|
async def close_redis() -> None:
|
|
global _redis
|
|
if _redis is not None:
|
|
await _redis.aclose()
|
|
_redis = None
|
|
|
|
|
|
def get_redis() -> redis.Redis:
|
|
if _redis is None:
|
|
raise RuntimeError("Redis 尚未初始化")
|
|
return _redis
|
|
|
|
|
|
async def redis_dependency() -> AsyncGenerator[redis.Redis, None]:
|
|
yield get_redis()
|