Add debug prints to backend forwarding to trace job delivery to workers

parent 7167f743
......@@ -58,9 +58,7 @@ def handle_web_message(message: Message, client_sock=None) -> Message:
from .config import get_analysis_backend
backend = get_analysis_backend()
worker_key = f'analysis_{backend}'
from .config import get_debug
if get_debug():
print(f"DEBUG: Backend forwarding analyze_request {message.msg_id} to worker {worker_key}")
print(f"Backend forwarding analyze_request {message.msg_id} to worker {worker_key}")
if worker_key in worker_sockets:
# Forward to local worker
import json
......@@ -70,15 +68,12 @@ def handle_web_message(message: Message, client_sock=None) -> Message:
'data': message.data
}
msg_json = json.dumps(msg_dict) + '\n'
if get_debug():
print(f"DEBUG: Sending to worker: {msg_json.strip()}")
print(f"Backend sending to worker: {msg_json.strip()}")
worker_sockets[worker_key].sendall(msg_json.encode('utf-8'))
if get_debug():
print(f"DEBUG: Sent to worker {worker_key}")
print(f"Backend sent to worker {worker_key}")
return Message('success', message.msg_id, {'status': 'forwarded'})
else:
if get_debug():
print(f"DEBUG: Worker {worker_key} not found in worker_sockets")
print(f"Backend: Worker {worker_key} not found in worker_sockets")
return Message('error', message.msg_id, {'error': f'Worker {worker_key} not available'})
elif message.msg_type == 'train_request':
backend = get_training_backend()
......
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