Skip to content

Commit b7254fd

Browse files
author
Alan Christie
committed
style: Now logs exit
1 parent f6045e5 commit b7254fd

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

app/app.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -111,14 +111,19 @@ async def event_stream(websocket: WebSocket, uuid: str):
111111
_LOGGER.debug("Creating reader for %s...", es_id)
112112
message_reader = _get_from_queue(routing_key)
113113

114-
_LOGGER.debug("Reading %s (message_reader=%s)...", es_id, message_reader)
115-
while True:
114+
_LOGGER.debug(
115+
"Reading messages for %s (message_reader=%s)...", es_id, message_reader
116+
)
117+
_running: bool = True
118+
while _running:
116119
_LOGGER.debug("Calling anext() for %s...", es_id)
117120
reader = anext(message_reader)
118121
message_body = await reader
119122
_LOGGER.debug("Got message for %s (message_body=%s)", es_id, message_body)
120123
await websocket.send_text(str(message_body))
121124

125+
_LOGGER.debug("Leaving %s (uid=%s)...", es_id, uuid)
126+
122127

123128
async def _get_from_queue(routing_key: str):
124129
"""An asynchronous generator yielding message bodies from the queue

0 commit comments

Comments
 (0)