fix: remove stale runs from redis jobs

Signed-off-by: Keshav Priyadarshi <git@keshav.space>

Author: keshav-space

vulnerabilities/models.py

@@ -2219,8 +2219,9 @@ def stop_run(self):
             job_id=str(self.run_id),
         )
         self.set_run_stopped()
+        self.delete_run_job()
 
-    def delete_run(self):
+    def delete_run_job(self):
         if job := self.job:
             job.delete()
 
@@ -2231,7 +2232,7 @@ def delete(self, *args, **kwargs):
         with suppress(redis.exceptions.ConnectionError, AttributeError):
             self.stop_run()
 
-        self.delete_run()
+        self.delete_run_job()
 
         return super().delete(*args, **kwargs)
 

vulnerabilities/schedules.py

@@ -131,7 +131,7 @@ def mark_stale_runs():
     stale_jobs_count = stale_jobs.count()
 
     for job in stale_jobs.iterator(chunk_size=1000):
-        job.set_run_staled()
+        job.stop_run()
 
     log.info(f"Marked {stale_jobs_count} unfinished jobs as stale.")
 
@@ -151,7 +151,7 @@ def requeue_missing_jobs():
                 enqueue_run(run=job)
                 missing_jobs_count += 1
             else:
-                job.set_run_staled()
+                job.stop_run()
 
     log.info(f"Requeued {missing_jobs_count} missing jobs.")