Add expireAt to idempotency lock to be able to configure TTL.

Co-authored-by: Alice Boberg <[email protected]>
BonnierNews · Jan 22, 2024 · d6bf9a2 · d6bf9a2
1 parent 8344cca
commit d6bf9a2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lib/job-storage/firestore-job-storage.js b/lib/job-storage/firestore-job-storage.js
@@ -1,4 +1,4 @@
-import Firestore from "@google-cloud/firestore";
+import { Firestore } from "@google-cloud/firestore";
 
 import { bucketHash, parentPayload, scanForInvalidKeys } from "./utils/job-storage-helper.js";
 import buildLogger from "../logger.js";
@@ -111,7 +111,7 @@ async function messageAlreadySeen(idempotencyKey, deliveryAttempt) {
     await db
       .collection("idempotencyLocks")
       .doc(`${idempotencyKey}:${deliveryAttempt}`)
-      .create({ idempotencyKey, deliveryAttempt });
+      .create({ idempotencyKey, deliveryAttempt, expireAt: new Date(Date.now() + 24 * 60 * 60 * 1000) });
     return false;
   } catch (err) {
     if (err.code.toLowerCase() === "already-exists") {