Don't crash when failing to read ZMQ reply in some circumstances

lkhq · Mar 24, 2024 · 5d106b8 · 5d106b8
1 parent 4b68604
commit 5d106b8
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/spark/connection.py b/spark/connection.py
@@ -122,7 +122,10 @@ def send_job_status(self, job_id, status):
             self._send_attempt_failed()
             log.error('ZMQ error while waiting for reply: %s', str(e))
         if sockev.get(self._sock) == zmq.POLLIN:
-            self._sock.recv_multipart()  # discard reply
+            try:
+                self._sock.recv_multipart()  # discard reply
+            except zmq.error.ZMQError as e:
+                log.error('ZMQ error, unable to receive reply: %s', str(e))
         else:
             log.error('Unable to send job status: No reply from master')