python/aiogram
1
0
Fork 0
mirror of https://github.com/aiogram/aiogram.git synced 2026-04-08 16:37:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

chore: refactor _get_redis for adapter

Browse source
This commit is contained in:
Oleg A 2021-08-01 15:27:59 +03:00
parent ba08aab63f
commit 7d45f59b98
1 changed files with 7 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

14
aiogram/contrib/fsm_storage/redis.py
View file

@ -239,7 +239,7 @@ class AioRedisAdapterBase(ABC):
self._connection_lock = asyncio.Lock(loop=self._loop)
@abstractmethod
async def redis(self) -> aioredis.Redis:
async def _get_redis(self) -> aioredis.Redis:
"""Get Redis connection."""
pass
@ -266,7 +266,7 @@ class AioRedisAdapterBase(ABC):
class AioRedisAdapterV1(AioRedisAdapterBase):
"""Redis adapter for aioredis v1."""
async def redis(self) -> aioredis.Redis:
async def _get_redis(self) -> aioredis.Redis:
"""Get Redis connection."""
async with self._connection_lock: # to prevent race
if self._redis is None or self._redis.closed:
@ -294,11 +294,11 @@ class AioRedisAdapterV1(AioRedisAdapterBase):
return True
async def get(self, name):
redis = await self.redis()
redis = await self._get_redis()
return await redis.get(name, encoding="utf8")
async def set(self, name, value, expire=None):
redis = await self.redis()
redis = await self._get_redis()
return await redis.set(name, value, expire=expire)
async def delete(self, name):
@ -309,7 +309,7 @@ class AioRedisAdapterV1(AioRedisAdapterBase):
class AioRedisAdapterV2(AioRedisAdapterBase):
"""Redis adapter for aioredis v2."""
async def redis(self) -> aioredis.Redis:
async def _get_redis(self) -> aioredis.Redis:
"""Get Redis connection."""
async with self._connection_lock: # to prevent race
if self._redis is None:
@ -325,11 +325,11 @@ class AioRedisAdapterV2(AioRedisAdapterBase):
return self._redis
async def get(self, name):
redis = await self.redis()
redis = await self._get_redis()
return await redis.get(name)
async def set(self, name, value, expire=None):
redis = await self.redis()
redis = await self._get_redis()
return await redis.set(name, value, ex=expire)
async def delete(self, name):