feat: add locking expires check

includable · Sep 7, 2024 · 5a703da · 5a703da
1 parent b28e09c
commit 5a703da
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/packages/api/src/lib/locks.js b/packages/api/src/lib/locks.js
@@ -3,6 +3,7 @@ import { deleteItem, put } from "./database";
 export const acquireLock = async (key, ttl = 900) => {
   const lockKey = `lock#${key}`;
   const expires = Math.floor(Date.now() / 1000) + ttl;
+
   try {
     await put(
       {
@@ -12,7 +13,12 @@ export const acquireLock = async (key, ttl = 900) => {
       },
       true,
       {
-        ConditionExpression: "attribute_not_exists(pk)",
+        ConditionExpression: "attribute_not_exists(pk) OR _expires < :now",
+        AttributeValues: {
+          ":now": {
+            N: `${Math.floor(Date.now() / 1000)}`,
+          },
+        },
       },
     );
     return true;