Merge branch 'master' of ssh://gitea.snsd0805.com:10805/snsd0805/gpu-provider

2024-05-31 03:47:27 +08:00 · 2024-05-31 03:47:27 +08:00 · a96c6d968e
commit a96c6d968e
parent 01024d2f0d dd67d88da0
1 changed files with 5 additions and 0 deletions
--- a/src/communication.py
+++ b/src/communication.py
@ -97,8 +97,13 @@ class ClusterCommunicationModule():
            self.client_sock.send(f'[STOP_CHECK]  {json.dumps(data)}'.encode())
            self.node_manager.docker_client.swarm.leave()
            self.node_manager.status = 'none'
+            self.node_manager.actions = [
+                {'explanation': 'Add another node into our cluster', 'function': 'add_node'},
+                {'explanation': 'Exit', 'function': 'exit'},
+            ]

            print("You have leaved the cluster.")
+            print("Please Enter to continued")

            return False