Skip to content

Commit dacb7ab

Browse files
author
Alan Christie
committed
refactor: Better (leaner) message log
1 parent d529eb7 commit dacb7ab

File tree

1 file changed

+11
-8
lines changed

1 file changed

+11
-8
lines changed

app/app.py

Lines changed: 11 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -181,19 +181,22 @@ async def event_stream(websocket: WebSocket, uuid: str):
181181
_LOGGER.debug(
182182
"Reading messages for %s (message_reader=%s)...", es_id, message_reader
183183
)
184-
_running: bool = True
185184
_connected: bool = True
186-
while _running and _connected:
185+
while _connected:
187186
_LOGGER.debug("Calling anext() for %s...", es_id)
188187
reader = anext(message_reader)
189188
message_body = await reader
190-
_LOGGER.debug("Got message for %s (message_body=%s)", es_id, message_body)
191-
if message_body == b"POISON":
192-
_LOGGER.info("Taking POISON for %s (%s) (closing)...", es_id, uuid)
193-
_running = False
194-
else:
189+
if message_body:
190+
if message_body == b"POISON":
191+
_LOGGER.info("Taking POISON for %s (%s) (closing)...", es_id, uuid)
192+
break
193+
if _LOGGER.isEnabledFor(logging.DEBUG):
194+
message_class: str = message_body.decode("utf-8").split("|", 1)[0]
195+
_LOGGER.debug(
196+
"Got AS message for %s (message_class=%s)", es_id, message_class
197+
)
195198
try:
196-
await websocket.send_text(str(message_body))
199+
await websocket.send_text(message_body)
197200
except WebSocketDisconnect:
198201
_LOGGER.warning(
199202
"Got WebSocketDisconnect for %s (%s) (leaving)...", es_id, uuid

0 commit comments

Comments
 (0)