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