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: