Reduce async/await context

mzyy94 · May 25, 2024 · 4d60c26 · 4d60c26
1 parent 6c86ef7
commit 4d60c26
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 7 deletions.
diff --git a/src/api/register.ts b/src/api/register.ts
@@ -85,7 +85,7 @@ export const registerAccount = factory.createHandlers(
         target: users.user,
         set: { handle: identifier, issuedAt: now },
       });
-    await savePubkey(c, didDoc.id, findPubkey(didDoc) ?? '');
+    savePubkey(c, didDoc.id, findPubkey(didDoc) ?? '');
 
     const { JWT_SECRET } = env<{ JWT_SECRET: string }>(c);
     const token = await sign(

diff --git a/src/middleware/auth.ts b/src/middleware/auth.ts
@@ -78,7 +78,7 @@ export const XrpcAuth = (opt: Option) =>
       if (!verified) {
         throw authError(c, 401, 'unauthorized', 'token verification failure');
       }
-      await savePubkey(c, iss, pubkey);
+      savePubkey(c, iss, pubkey);
     }
 
     c.set('iss', iss);

diff --git a/src/pubkey.ts b/src/pubkey.ts
@@ -13,13 +13,13 @@ export function findPubkey(didDoc: DidDoc): string | null {
   return null;
 }
 
-export async function savePubkey(c: Context, did: string, pubkey: string) {
+export function savePubkey(c: Context, did: string, pubkey: string) {
   if (!did || !pubkey) {
     return;
   }
-  await putPubkeyToCache(did, pubkey);
+  c.executionCtx.waitUntil(putPubkeyToCache(did, pubkey));
   const { did_key_store } = env<{ did_key_store: KVNamespace }>(c);
-  await did_key_store.put(did, pubkey);
+  c.executionCtx.waitUntil(did_key_store.put(did, pubkey));
 }
 
 export async function getPubkey(c: Context, did: string) {
@@ -31,7 +31,7 @@ export async function getPubkey(c: Context, did: string) {
   }
   pubkey = await did_key_store.get(did);
   if (pubkey) {
-    await putPubkeyToCache(did, pubkey);
+    c.executionCtx.waitUntil(putPubkeyToCache(did, pubkey));
   }
   return pubkey;
 }

diff --git a/src/xrpc/feed.ts b/src/xrpc/feed.ts
@@ -193,7 +193,9 @@ export const getFeedSkeletonHandlers = factory.createHandlers(
     const feed = result.map(({ post }) => ({ post }));
     const lastPost = result.at(-1);
     const lastCur = createCursor(lastPost);
-    await putFeedToCache(c, user, feedItems, opId, range, !cursor);
+    c.executionCtx.waitUntil(
+      putFeedToCache(c, user, feedItems, opId, range, !cursor),
+    );
 
     return c.json<TFeed, 200>({ cursor: lastCur, feed });
   },