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

feat: docker swarm join (worker)

Browse Source
This commit is contained in:
snsd0805 2024-05-30 15:43:13 +08:00
parent 4cbf9957a2
commit e72b9021be
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/communication.py
View File

@ -48,7 +48,6 @@ class ClusterCommunicationModule():
token = self.node_manager.docker_client.swarm.attrs['JoinTokens']['Worker'] token = self.node_manager.docker_client.swarm.attrs['JoinTokens']['Worker']
conn.send(f'[DOCKER_TOKEN] {token}'.encode()) conn.send(f'[DOCKER_TOKEN] {token}'.encode())
print(f"Send token: {token} to the worker.") print(f"Send token: {token} to the worker.")
time.sleep(3)
self.worker_conns.append(conn) self.worker_conns.append(conn)
continue continue
@ -63,8 +62,12 @@ class ClusterCommunicationModule():
self.client_sock.send('[CHECK]'.encode()) self.client_sock.send('[CHECK]'.encode())
# join docker swarm cluster # join docker swarm cluster
docker_token = self.client_sock.recv(1024).decode().split()[-1] token = self.client_sock.recv(1024).decode().split()[-1]
print("Receive Docker Swarm Join_Token=", docker_token) print("Receive Docker Swarm Join_Token=", token)
status = self.node_manager.docker_client.swarm.join(remote_addrs=[f'{addr[0]}:2377'], join_token=token)
if not status:
print("Some Errors!")
# remove 'add node' # remove 'add node'
self.node_manager.actions = self.node_manager.actions[1:] self.node_manager.actions = self.node_manager.actions[1:]