Speed up cluster shutdown

mrocklin · mrocklin · commit dc7acee9ca84 · 2022-05-22T21:51:48.000-05:00
This permits servers to allow "terminate" comms to persist when shutting down. Closes #6413
diff --git a/distributed/core.py b/distributed/core.py
@@ -680,10 +680,11 @@ def close(self):
         for i in range(20):
             # If there are still handlers running at this point, give them a
             # second to finish gracefully themselves, otherwise...
-            if any(self._comms.values()):
+            if any(comm for comm in self._comms.values() if comm != "terminate"):
                 yield asyncio.sleep(0.05)
             else:
                 break
+
         yield self.rpc.close()
         yield [comm.close() for comm in list(self._comms)]  # then forcefully close
         for cb in self._ongoing_coroutines:
