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
worker_sockets = {} # type: dict
pending_results = {} # msg_id -> result message
job_progress = {} # job_id -> progress data
job_progress_times = {} # job_id -> last_progress_timestamp
......@@ -74,7 +75,7 @@ def handle_web_message(message: Message, client_sock=None) -> Message:
worker_sockets[worker_key].sendall(msg_json.encode('utf-8'))
if get_debug():
print(f"DEBUG: Sent to worker {worker_key}")
return None # No immediate response
return Message('success', message.msg_id, {'status': 'forwarded'})
else:
if get_debug():
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:
worker_sockets[worker_key].sendall(
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:
return Message('error', message.msg_id, {'error': f'Worker {worker_key} not available'})
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