Replace job calls in queue with self calls (#158)

tobymao · Sep 16, 2024 · 89de1e2 · 89de1e2
1 parent 0ed1042
commit 89de1e2
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/saq/queue/postgres.py b/saq/queue/postgres.py
@@ -323,7 +323,7 @@ async def sweep(self, lock: int = 60, abort: float = 5.0) -> list[str]:
                 if job.retryable:
                     await self.retry(job, error="swept")
                 else:
-                    await job.finish(Status.ABORTED, error="swept")
+                    await self.finish(job, Status.ABORTED, error="swept")
         return swept
 
     async def listen(

diff --git a/saq/queue/redis.py b/saq/queue/redis.py
@@ -261,7 +261,7 @@ async def sweep(self, lock: int = 60, abort: float = 5.0) -> list[str]:
                     if job.retryable:
                         await self.retry(job, error="swept")
                     else:
-                        await job.finish(Status.ABORTED, error="swept")
+                        await self.finish(job, Status.ABORTED, error="swept")
             else:
                 swept.append(job_id)
 
@@ -332,7 +332,7 @@ async def abort(self, job: Job, error: str, ttl: float = 5) -> None:
                 )
 
             if dequeued:
-                await job.finish(Status.ABORTED, error=error)
+                await self.finish(job, Status.ABORTED, error=error)
                 await self.redis.delete(job.abort_id)
             else:
                 await self.redis.lrem(self._active, 0, job.id)