feat: support KV key expiration

badrap · Feb 4, 2024 · d4e90be · d4e90be
1 parent 8b91334
commit d4e90be
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/src/api/kv.ts b/src/api/kv.ts
@@ -12,7 +12,7 @@ type KvEntryMaybe<T> =
 type AtomicCheck = { key: KvKey; versionstamp: string | null };
 
 type KvMutation =
-  | { type: "set"; key: KvKey; value: unknown }
+  | { type: "set"; key: KvKey; value: unknown; expireIn?: number }
   | { type: "delete"; key: KvKey };
 
 type KvCommitResult = { ok: true; versionstamp: string };
@@ -123,8 +123,13 @@ class AtomicOperation {
     return this;
   }
 
-  set(key: KvKey, value: unknown): this {
-    this._mutations.push({ type: "set", key, value });
+  set(key: KvKey, value: unknown, options?: { expireIn?: number }): this {
+    this._mutations.push({
+      type: "set",
+      key,
+      value,
+      expireIn: options?.expireIn,
+    });
     return this;
   }