Retry initial Redis connection (Fixes #1534)

Author: miguelgrinberg

src/socketio/async_redis_manager.py

@@ -134,11 +134,12 @@ async def _publish(self, data):  # pragma: no cover
                     break
 
     async def _redis_listen_with_retries(self):  # pragma: no cover
-        retry_sleep = 1
         _, error = self._get_redis_module_and_error()
+        retry_sleep = 1
+        subscribed = False
         while True:
             try:
-                if not self.connected:
+                if not subscribed:
                     self._redis_connect()
                     await self.pubsub.subscribe(self.channel)
                     retry_sleep = 1
@@ -149,7 +150,7 @@ async def _redis_listen_with_retries(self):  # pragma: no cover
                                          'retrying in '
                                          f'{retry_sleep} secs',
                                          extra={"redis_exception": str(exc)})
-                self.connected = False
+                subscribed = False
                 await asyncio.sleep(retry_sleep)
                 retry_sleep *= 2
                 if retry_sleep > 60:

src/socketio/redis_manager.py

@@ -169,11 +169,12 @@ def _publish(self, data):  # pragma: no cover
                     break
 
     def _redis_listen_with_retries(self):  # pragma: no cover
-        retry_sleep = 1
         _, error = self._get_redis_module_and_error()
+        retry_sleep = 1
+        subscribed = False
         while True:
             try:
-                if not self.connected:
+                if not subscribed:
                     self._redis_connect()
                     self.pubsub.subscribe(self.channel)
                     retry_sleep = 1
@@ -182,7 +183,7 @@ def _redis_listen_with_retries(self):  # pragma: no cover
                 logger.error('Cannot receive from redis... '
                              f'retrying in {retry_sleep} secs',
                              extra={"redis_exception": str(exc)})
-                self.connected = False
+                subscribed = False
                 time.sleep(retry_sleep)
                 retry_sleep *= 2
                 if retry_sleep > 60: