Connection to connection pool

Author: chandr-andr

poetry.lock

@@ -9,7 +9,6 @@ packages = [{ include = "taskiq", from = "taskiq_redis" }]
 python = "^3.7"
 taskiq = { git = "https://github.com/taskiq-python/taskiq", branch = "master" }
 redis = "4.3.4"
-types-redis = "^4.3.12"
 flake8 = "^4.0.1"
 
 [tool.poetry.dev-dependencies]

pyproject.toml

@@ -1,7 +1,7 @@
 from typing import Mapping, Optional, Union
 
 from errors import RedisConnectionError
-from redis.asyncio import Redis
+from redis.asyncio import ConnectionPool, Redis
 
 
 class RedisClient:
@@ -13,11 +13,11 @@ def __init__(self, redis_url: str) -> None:
 
         :param redis_url: Redis URL to connect.
         """
-        self.redis_client = Redis.from_url(redis_url)
+        self.redis_pool = ConnectionPool.from_url(redis_url)
 
     async def close(self) -> None:
         """Closes redis connection."""
-        await self.redis_client.close()
+        await self.redis_pool.disconnect()
 
     async def hset(
         self,
@@ -33,7 +33,8 @@ async def hset(
         :raises RedisConnectionError: if redis is not available.
         """
         try:
-            await self.redis_client.hset(name, mapping=mapping)
+            async with Redis(connection_pool=self.redis_pool) as redis:
+                await redis.hset(name, mapping=mapping)
         except ConnectionError as exc:
             raise RedisConnectionError("Redis is unavailable") from exc
 
@@ -49,7 +50,8 @@ async def hget(self, name: str, key: str) -> Optional[bytes]:
         :returns: bytes.
         """
         try:
-            return await self.redis_client.hget(name=name, key=key)
+            async with Redis(connection_pool=self.redis_pool) as redis:
+                return await redis.hget(name=name, key=key)
         except ConnectionError as exc:
             raise RedisConnectionError("Redis is unavailable") from exc
 
@@ -64,6 +66,7 @@ async def exists(self, name: str) -> bool:
         :returns: True if name exists else False
         """
         try:
-            return bool(await self.redis_client.exists(name))
+            async with Redis(connection_pool=self.redis_pool) as redis:
+                return bool(await redis.exists(name))
         except ConnectionError as exc:
             raise RedisConnectionError("Redis is unavailable") from exc