From 459b5ff8531420369894105ff2bcb7f983e97e87 Mon Sep 17 00:00:00 2001
From: prandla <prandla@r9.pm>
Date: Sun, 27 Jul 2025 08:12:28 +0300
Subject: [PATCH] Fix "RPCError: write failed" when killing service from AWS

---
 cms/service/ResourceService.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/cms/service/ResourceService.py b/cms/service/ResourceService.py
index bf41a477a1..6ec43d4b75 100644
--- a/cms/service/ResourceService.py
+++ b/cms/service/ResourceService.py
@@ -440,6 +440,7 @@ def kill_service(self, service: str):
             logger.error("Unable to decode service shard.")
 
         remote_service = self.connect_to(ServiceCoord(name, shard))
+        remote_service.wait_for_connection(timeout=5)
         result = remote_service.quit(reason="Asked by ResourceService")
         return result.get()