feat: add db option to redis backend

leynier · Jan 25, 2025 · 6cb1db7 · 6cb1db7
1 parent db7411b
commit 6cb1db7
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/fastapi_cache_plus/backends/redis.py b/fastapi_cache_plus/backends/redis.py
@@ -19,10 +19,12 @@ class RedisCacheBackend(BaseCacheBackend[RedisKey, RedisValue]):
     def __init__(
         self,
         address: str,
+        db: Optional[int] = None,
         pool_minsize: Optional[int] = DEFAULT_POOL_MIN_SIZE,
         encoding: Optional[str] = DEFAULT_ENCODING,
     ) -> None:
         self._redis_address = address
+        self._redis_db = db
         self._redis_pool_minsize = pool_minsize
         self._encoding = encoding
         self._pool: Optional[Redis] = None
@@ -36,6 +38,7 @@ async def _client(self) -> Redis:
     async def _create_connection(self) -> Redis:
         return await aioredis.create_redis_pool(
             self._redis_address,
+            db=self._redis_db,
             minsize=self._redis_pool_minsize or 1,
         )