Skip to content

V
vidai
Commit 6c0c6b0b authored by Stefy Lanza (nextime / spora )'s avatar Stefy Lanza (nextime / spora )
Browse files
Options

Update cluster_master to store progress in database for both local and remote workers

parent 14d4ce69
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 0 deletions
vidai/cluster_master.py
View file @ 6c0c6b0b
...@@ -287,6 +287,11 @@ class ClusterMaster: ...@@ -287,6 +287,11 @@ class ClusterMaster:
# Update the job's last activity time to reset timeout # Update the job's last activity time to reset timeout
self.active_jobs[job_id]['last_progress'] = time.time() self.active_jobs[job_id]['last_progress'] = time.time()
print(f"Progress update for job {job_id}: {message.data.get('progress', 0)}%") print(f"Progress update for job {job_id}: {message.data.get('progress', 0)}%")
# Update database for consistency (both local and remote workers)
from .database import update_job_progress
progress = message.data.get('progress', 0)
progress_message = message.data.get('message', '')
update_job_progress(job_id, progress, progress_message)
return {'type': 'progress_ack'} return {'type': 'progress_ack'}
elif msg_type == 'ping': elif msg_type == 'ping':
......
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