diff options
Diffstat (limited to 'redis/asyncio/sentinel.py')
-rw-r--r-- | redis/asyncio/sentinel.py | 13 |
1 files changed, 10 insertions, 3 deletions
diff --git a/redis/asyncio/sentinel.py b/redis/asyncio/sentinel.py index 99c5074..ec17886 100644 --- a/redis/asyncio/sentinel.py +++ b/redis/asyncio/sentinel.py @@ -1,7 +1,7 @@ import asyncio import random import weakref -from typing import AsyncIterator, Iterable, Mapping, Sequence, Tuple, Type +from typing import AsyncIterator, Iterable, Mapping, Optional, Sequence, Tuple, Type from redis.asyncio.client import Redis from redis.asyncio.connection import ( @@ -63,9 +63,16 @@ class SentinelManagedConnection(Connection): lambda error: asyncio.sleep(0), ) - async def read_response(self, disable_decoding: bool = False): + async def read_response( + self, + disable_decoding: bool = False, + timeout: Optional[float] = None, + ): try: - return await super().read_response(disable_decoding=disable_decoding) + return await super().read_response( + disable_decoding=disable_decoding, + timeout=timeout, + ) except ReadOnlyError: if self.connection_pool.is_master: # When talking to a master, a ReadOnlyError when likely |