[ADD] infinity loop for websocket

Author: shurshilov

const.py

@@ -1 +1 @@
-VERSION = "1.0.7"
+VERSION = "1.0.8"

main.py

@@ -145,16 +145,24 @@ async def catch_exception_internal(req: Request, exc: Exception):
 
 async def producer_webhook(config: Config, timeout: int = 30) -> None:
     """Producer send events to cusomer webhook from config file"""
-    websocket_client = WebsocketEvents(
-        ari_config=config.ari_config,
-        api_key=config.api_key,
-        api_key_base64=config.api_key_base64,
-        webhook_url=f"{config.webhook_url}",
-        timeout=timeout,
-    )
-    app.state.websocket_client = websocket_client
-
-    await websocket_client.start_consumer()
+    while True:
+        try:
+            websocket_client = WebsocketEvents(
+                ari_config=config.ari_config,
+                api_key=config.api_key,
+                api_key_base64=config.api_key_base64,
+                webhook_url=f"{config.webhook_url}",
+                timeout=timeout,
+            )
+            app.state.websocket_client = websocket_client
+
+            await websocket_client.start_consumer()
+        except asyncio.CancelledError:
+            break
+        except Exception as e:
+            log.exception("Unknown producer_webhook error: %s", e)
+        finally:
+            await asyncio.sleep(timeout)
 
 
 @app.on_event("startup")

services/websocket.py

@@ -224,5 +224,5 @@ async def start_consumer(self):
             self.disconnect_count += 1
             self.disconnected_reason = str(exc)
             self.disconnected_time = str(datetime.datetime.now())
-            await asyncio.sleep(self.timeout)
-            await self.start_consumer()
+            # await asyncio.sleep(self.timeout)
+            # await self.start_consumer()