Add Job last modified

hummingbird-project · Mar 6, 2024 · 3c94c17 · 3c94c17
1 parent 7c0470f
commit 3c94c17
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/Sources/HummingbirdJobsPostgres/CreateJobQueue.swift b/Sources/HummingbirdJobsPostgres/CreateJobQueue.swift
@@ -23,7 +23,8 @@ struct CreateJobQueue: HBPostgresMigration {
             CREATE TABLE IF NOT EXISTS _hb_jobs (
                 id uuid PRIMARY KEY,
                 job json,
-                status smallint
+                status smallint,
+                lastModified TIMESTAMPTZ DEFAULT CURRENT_TIMESTAMP
             )     
             """,
             logger: logger

diff --git a/Sources/HummingbirdJobsPostgres/PostgresJobsQueue.swift b/Sources/HummingbirdJobsPostgres/PostgresJobsQueue.swift
@@ -216,7 +216,7 @@ public final class HBPostgresQueue: HBJobQueueDriver {
 
     func setStatus(jobId: JobID, status: Status, connection: PostgresConnection) async throws {
         try await connection.query(
-            "UPDATE _hb_jobs SET status = \(status) WHERE id = \(jobId)",
+            "UPDATE _hb_jobs SET status = \(status), lastModified = \(Date.now) WHERE id = \(jobId)",
             logger: self.logger
         )
     }

diff --git a/Tests/HummingbirdPostgresTests/JobsTests.swift b/Tests/HummingbirdPostgresTests/JobsTests.swift
@@ -231,8 +231,8 @@ final class JobsTests: XCTestCase {
             try await jobQueue.push(id: jobIdentifer, parameters: 0)
             await self.wait(for: [expectation], timeout: 5)
 
-            let failedJobs = try await jobQueue.queue.getJobs(withStatus: .processing)
-            XCTAssertEqual(failedJobs.count, 1)
+            let processingJobs = try await jobQueue.queue.getJobs(withStatus: .processing)
+            XCTAssertEqual(processingJobs.count, 1)
             let pendingJobs = try await jobQueue.queue.getJobs(withStatus: .pending)
             XCTAssertEqual(pendingJobs.count, 0)
             return jobQueue