Add exception handling to the receiver in the debug_logger

una-auxme · Nov 12, 2024 · 09342c2 · 09342c2
1 parent 4db8bda
commit 09342c2
Showing 1 changed file with 15 additions and 10 deletions.
diff --git a/code/debugging/src/debug_logger.py b/code/debugging/src/debug_logger.py
@@ -79,16 +79,21 @@ def run_listener(listener: Listener):
     """
     running = True
     while running:
-        conn = listener.accept()
-        print(f"[debug_logger]: connection accepted from {listener.last_accepted}")
-        msg = None
-        if conn.poll(timeout=2.0):
-            msg = conn.recv()
-            if isinstance(msg, str):
-                if msg.lower() == CLOSE_MSG:
-                    running = False
-                    msg = None
-        conn.close()
+        try:
+            conn = listener.accept()
+            print(f"[debug_logger]: connection accepted from {listener.last_accepted}")
+            msg = None
+            if conn.poll(timeout=2.0):
+                msg = conn.recv()
+                if isinstance(msg, str):
+                    if msg.lower() == CLOSE_MSG:
+                        running = False
+                        msg = None
+            conn.close()
+        except Exception as error:
+            eprint(f"Failed to receive message: {error}")
+            continue
+
         if msg is not None:
             with MESSAGES_MUTEX:
                 MESSAGES.append(msg)