File tree Expand file tree Collapse file tree 1 file changed +11
-8
lines changed
Expand file tree Collapse file tree 1 file changed +11
-8
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments