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

fix: use 2 spaces as split token

Browse Source
This commit is contained in:
Ting-Jun Wang 2024-05-30 17:15:18 +08:00
parent 5b8b3e7c39
commit 0cae46b25b
Signed by: snsd0805
GPG Key ID: 48D331A3D6160354
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/communication.py
View File

@ -65,7 +65,7 @@ class ClusterCommunicationModule():
self.client_sock.send('[CHECK]'.encode()) self.client_sock.send('[CHECK]'.encode())
# join docker swarm cluster # join docker swarm cluster
token = self.client_sock.recv(1024).decode().split()[-1] token = self.client_sock.recv(1024).decode().split(' ')[-1]
print("Receive Docker Swarm Join_Token=", 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) status = self.node_manager.docker_client.swarm.join(remote_addrs=[f'{addr[0]}:2377'], join_token=token)
@ -84,7 +84,7 @@ class ClusterCommunicationModule():
return False return False
def handle_command(self, data): def handle_command(self, data):
command, args = data.split() command, args = data.split(' ')
if command == '[INFO]': if command == '[INFO]':
data = {'host': self.host, 'GPU': self.node_manager.GPU, 'GPU_num': self.node_manager.GPU_num} data = {'host': self.host, 'GPU': self.node_manager.GPU, 'GPU_num': self.node_manager.GPU_num}
self.client_sock.send(json.dumps(data).encode()) self.client_sock.send(json.dumps(data).encode())
@ -132,7 +132,7 @@ class ClusterCommunicationModule():
if self.node_manager.status == 'master': if self.node_manager.status == 'master':
for conn in self.worker_conns: for conn in self.worker_conns:
conn.send('[STOP] {}'.encode()) conn.send('[STOP] {}'.encode())
check, args = conn.recv(1024).decode().split() check, args = conn.recv(1024).decode().split(' ')
print(f'{args} has stopped.') print(f'{args} has stopped.')
self.node_manager.docker_client.swarm.leave(force=True) self.node_manager.docker_client.swarm.leave(force=True)