For Redis add retry policy

Author: abpostelnicu

libmozevent/utils.py

@@ -12,7 +12,18 @@
 
 import hglib
 import structlog
+from redis.backoff import NoBackoff
 from redis import asyncio as aioredis
+from redis.asyncio.retry import Retry
+from redis.exceptions import (
+    ConnectionError,
+    ExecAbortError,
+    PubSubError,
+    RedisError,
+    ResponseError,
+    TimeoutError,
+    WatchError,
+)
 
 log = structlog.get_logger(__name__)
 
@@ -175,16 +186,31 @@ class AsyncRedis(object):
     @staticmethod
     async def connect():
         if AsyncRedis.redis.get(None) is None:
+            retry_on_error = [
+                ConnectionError,
+                ExecAbortError,
+                PubSubError,
+                RedisError,
+                ResponseError,
+                TimeoutError,
+                WatchError,
+            ]
+
+            retries = 5
             AsyncRedis.redis.set(
                 await aioredis.from_url(
                     os.environ["REDIS_TLS_URL"],
                     decode_responses=False,
                     ssl_cert_reqs=None,
+                    retry_on_error=retry_on_error,
+                    retry=Retry(NoBackoff(), retries),
                 )
                 if os.getenv("REDIS_TLS_URL", None) is not None
                 else await aioredis.from_url(
                     os.environ["REDIS_URL"],
                     decode_responses=False,
+                    retry_on_error=retry_on_error,
+                    retry=Retry(NoBackoff(), retries),
                 )
             )
         return AsyncRedis.redis.get()