Fix backend to always return response for job forwarding - success/error...

Fix backend to always return response for job forwarding - success/error messages ensure proper job assignment feedback
parent 0f7b1478
...@@ -28,6 +28,7 @@ from .queue import queue_manager ...@@ -28,6 +28,7 @@ from .queue import queue_manager
worker_sockets = {} # type: dict worker_sockets = {} # type: dict
pending_results = {} # msg_id -> result message pending_results = {} # msg_id -> result message
job_progress = {} # job_id -> progress data
job_progress_times = {} # job_id -> last_progress_timestamp job_progress_times = {} # job_id -> last_progress_timestamp
...@@ -74,7 +75,7 @@ def handle_web_message(message: Message, client_sock=None) -> Message: ...@@ -74,7 +75,7 @@ def handle_web_message(message: Message, client_sock=None) -> Message:
worker_sockets[worker_key].sendall(msg_json.encode('utf-8')) worker_sockets[worker_key].sendall(msg_json.encode('utf-8'))
if get_debug(): if get_debug():
print(f"DEBUG: Sent to worker {worker_key}") print(f"DEBUG: Sent to worker {worker_key}")
return None # No immediate response return Message('success', message.msg_id, {'status': 'forwarded'})
else: else:
if get_debug(): if get_debug():
print(f"DEBUG: Worker {worker_key} not found in worker_sockets") print(f"DEBUG: Worker {worker_key} not found in worker_sockets")
...@@ -86,7 +87,7 @@ def handle_web_message(message: Message, client_sock=None) -> Message: ...@@ -86,7 +87,7 @@ def handle_web_message(message: Message, client_sock=None) -> Message:
worker_sockets[worker_key].sendall( worker_sockets[worker_key].sendall(
f'{{"msg_type": "{message.msg_type}", "msg_id": "{message.msg_id}", "data": {message.data}}}\n'.encode('utf-8') f'{{"msg_type": "{message.msg_type}", "msg_id": "{message.msg_id}", "data": {message.data}}}\n'.encode('utf-8')
) )
return None return Message('success', message.msg_id, {'status': 'forwarded'})
else: else:
return Message('error', message.msg_id, {'error': f'Worker {worker_key} not available'}) return Message('error', message.msg_id, {'error': f'Worker {worker_key} not available'})
elif message.msg_type == 'config_update': elif message.msg_type == 'config_update':
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment