Shutdown FastStream app when STOMP broker becomes unavailable (but was available when app started) (#117)

Author: vrslev

packages/faststream-stomp/faststream_stomp/broker.py

@@ -1,3 +1,4 @@
+import asyncio
 import logging
 import types
 from collections.abc import Callable, Iterable, Mapping, Sequence
@@ -30,6 +31,15 @@ def get_schema(self) -> dict[str, dict[str, str]]:  # noqa: PLR6301
         return {"user-password": {"type": "userPassword"}}
 
 
+def _handle_listen_task_done(listen_task: asyncio.Task[None]) -> None:
+    # Not sure how to test this. See https://github.com/community-of-python/stompman/pull/117#issuecomment-2983584449.
+    task_exception = listen_task.exception()
+    if isinstance(task_exception, ExceptionGroup) and isinstance(
+        task_exception.exceptions[0], stompman.FailedAllConnectAttemptsError
+    ):
+        raise SystemExit(1)
+
+
 class StompBroker(StompRegistrator, BrokerUsecase[stompman.MessageFrame, stompman.Client]):
     _subscribers: Mapping[int, StompSubscriber]
     _publishers: Mapping[int, StompPublisher]
@@ -96,7 +106,9 @@ async def _connect(self, client: stompman.Client) -> stompman.Client:  # type: i
             return client
         self._attempted_to_connect = True
         self._producer = StompProducer(client)
-        return await client.__aenter__()
+        await client.__aenter__()
+        client._listen_task.add_done_callback(_handle_listen_task_done)  # noqa: SLF001
+        return client
 
     async def _close(
         self,