fix: added retry case ConnectionError

encode · Feb 28, 2024 · d43d473 · d43d473
1 parent 973055d
commit d43d473
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/broadcaster/_backends/redis.py b/broadcaster/_backends/redis.py
@@ -6,6 +6,7 @@
 
 from .._base import Event
 from .base import BroadcastBackend
+import asyncio
 
 
 class RedisBackend(BroadcastBackend):
@@ -40,7 +41,12 @@ async def unsubscribe(self, channel: str) -> None:
         await self._sub_conn.unsubscribe(channel)
 
     async def publish(self, channel: str, message: Any) -> None:
-        await self._pub_conn.execute_command("PUBLISH", channel, message)
+        try:
+            await self._pub_conn.execute_command("PUBLISH", channel, message)
+        except redis.ConnectionError:
+            await asyncio.sleep(1)
+            self._pub_conn = redis.Redis(**self.kwargs).pubsub()
+            await self.publish(channel, message)
 
     async def next_published(self) -> Event:
         message = None