11

app/redis_client.py

@@ -0,0 +1,36 @@
+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()