fix: allow non expiring claims

storacha · Jul 31, 2023 · ffce3b7 · ffce3b7
1 parent b2bce88
commit ffce3b7
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/packages/lambda/src/lib/store/index.js b/packages/lambda/src/lib/store/index.js
@@ -17,6 +17,7 @@ export { BlockIndexClaimFetcher } from './block-index.js'
 export class ClaimStorage extends DynamoTable {
   /** @param {import('@web3-storage/content-claims/server/api').Claim} claim */
   async put ({ claim, bytes, content, expiration }) {
+    const hasExpiration = expiration && isFinite(expiration)
     const cmd = new UpdateItemCommand({
       TableName: this.tableName,
       Key: marshall({
@@ -25,9 +26,11 @@ export class ClaimStorage extends DynamoTable {
       }),
       ExpressionAttributeValues: marshall({
         ':by': bytes,
-        ':ex': expiration
+        ':ex': hasExpiration ? expiration : undefined
       }, { removeUndefinedValues: true, convertClassInstanceToMap: true }),
-      UpdateExpression: 'SET bytes=if_not_exists(bytes, :by), expiration=if_not_exists(expiration, :ex)'
+      UpdateExpression: hasExpiration
+        ? 'SET bytes=if_not_exists(bytes, :by), expiration=if_not_exists(expiration, :ex)'
+        : 'SET bytes=if_not_exists(bytes, :by) REMOVE expiration'
     })
     await this.dynamoClient.send(cmd)
   }