snsd0805/gpu-provider
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: multi client, master only need to build cluster once

Browse Source
This commit is contained in:
Ting-Jun Wang 2024-05-30 16:15:31 +08:00
parent e72b9021be
commit 755a929aeb
Signed by: snsd0805
GPG Key ID: 48D331A3D6160354
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/communication.py
View File

@ -42,12 +42,15 @@ class ClusterCommunicationModule():
break
elif data == '[CHECK]': # master side
# build docker swarm
self.node_manager.docker_client.swarm.init(advertise_addr=self.host, listen_addr=f"{self.host}:2377", force_new_cluster=True)
if self.node_manager.docker_client.swarm.attrs == {}:
print("Build new docker swarm...")
self.node_manager.docker_client.swarm.init(advertise_addr=self.host, listen_addr=f"{self.host}:2377", force_new_cluster=True)
# send docker swarm token to the worker
token = self.node_manager.docker_client.swarm.attrs['JoinTokens']['Worker']
conn.send(f'[DOCKER_TOKEN] {token}'.encode())
print(f"Send token: {token} to the worker.")
print("Please Enter to continue...")
self.worker_conns.append(conn)
continue