From bfdabd8d625c7ba53f938073455951f7da272d86 Mon Sep 17 00:00:00 2001 From: KKlochko Date: Tue, 12 Dec 2023 20:30:40 +0200 Subject: [PATCH] Update the producer's handle to close the connection if empty data. --- notification_server/server.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/notification_server/server.py b/notification_server/server.py index f71d1be..ed2c865 100644 --- a/notification_server/server.py +++ b/notification_server/server.py @@ -76,8 +76,14 @@ class Server: try: while True: + title = await self.receive_message(reader) message = await self.receive_message(reader) - print(f"{message=}") + + if title == '' and message == '': + self.console.print(f"The producer ({addr[0]}:{addr[1]}) finished.") + break + + await self.broadcast_message(title) await self.broadcast_message(message) except asyncio.CancelledError: