Wrong use of timeout. Now proper grace period time

Author: Ematerasu

scripts_of_tribute/server.py

@@ -1,4 +1,3 @@
-import time
 from typing import Tuple
 import grpc
 from concurrent import futures
@@ -47,13 +46,8 @@ def CloseServer(self, request, context):
         print(f"Received CloseServer request from {self.ai.bot_name}. Shutting down server...")
         context.set_code(grpc.StatusCode.OK)
         context.set_details(f"Bot {self.ai.bot_name}'s connection closed.")
-        def delayed_shutdown():
-            time.sleep(0.1)
-            self.server_instance.bot_disconnected()
-            print(f"Bot {self.ai.bot_name}'s connection closed.")
-
-        import threading
-        threading.Thread(target=delayed_shutdown, daemon=True).start()
+        self.server_instance.bot_disconnected()
+        print(f"Bot {self.ai.bot_name}'s connection closed.")
 
         return main_pb2.Empty()
 
@@ -74,7 +68,7 @@ def bot_disconnected(self):
 
     def shutdown_server(self):
         if self.server:
-            self.server.stop(0)
+            self.server.stop(2)
         else:
             print("Server is already stopped.")
 
@@ -108,12 +102,6 @@ def run_grpc_server(
         server2.server.start()
 
     if bot1 is not None:
-        try:
-            server1.server.wait_for_termination(timeout=2)
-        except grpc.FutureTimeoutError:
-            print("Server didn't terminate cleanly in time.")
+        server1.server.wait_for_termination()
     if bot2 is not None:
-        try:
-            server2.server.wait_for_termination(timeout=2)
-        except grpc.FutureTimeoutError:
-            print("Server didn't terminate cleanly in time.")
+        server2.server.wait_for_termination()