feat(worker): prefetch resources after cache

packages/worker/package.json

@@ -18,6 +18,7 @@
     "anitomy": "workspace:*",
     "hono": "^3.4.1",
     "memofunc": "^0.0.5",
+    "ohash": "^1.1.3",
     "simptrad": "^0.1.0"
   },
   "devDependencies": {

packages/worker/src/query.ts

@@ -1,6 +1,7 @@
 import type { Context } from 'hono';
 import type { Resource, Team, User } from '@prisma/client/edge';
 
+import { hash } from 'ohash';
 import { memoAsync } from 'memofunc';
 import {
   parseSearchURL,
@@ -36,7 +37,13 @@ export async function queryResourceDetail(ctx: Context<{ Bindings: Env }>) {
   return ctx.json({ detail });
 }
 
-export const getSearchResources = memoAsync(
+export const PrefetchFilter = [
+  parseSearchURL(new URLSearchParams('?page=1')),
+  parseSearchURL(new URLSearchParams('?page=2')),
+  parseSearchURL(new URLSearchParams('?page=3'))
+];
+
+export const findResourcesFromDB = memoAsync(
   async (env: Env, options: ResolvedFilterOptions) => {
     const prisma = makePrisma(env);
 
@@ -118,38 +125,33 @@ export const getSearchResources = memoAsync(
   {
     external: {
       async get([env, params]) {
-        return getResourcesStore(env).get(JSON.stringify({ params }));
+        return getResourcesStore(env).get(hash(params));
       },
       async set([env, params], value) {
-        return getResourcesStore(env).put(JSON.stringify({ params }), value, {
+        return getResourcesStore(env).put(hash(params), value, {
           expirationTtl: 300
         });
       },
       async remove([env, params]) {
-        return getResourcesStore(env).remove(JSON.stringify({ params }));
+        return getResourcesStore(env).remove(hash(params));
       },
       async clear() {}
     }
   }
 );
 
-async function getTimestamp(ctx: Context<{ Bindings: Env }>) {
-  const timestamp = await getRefreshTimestamp(ctx.env);
-  return timestamp;
-}
-
 export async function searchResources(ctx: Context<{ Bindings: Env }>) {
   const url = new URL(ctx.req.url);
   const filter = parseSearchURL(url.searchParams, await ctx.req.json().catch(() => undefined));
   if (!filter) {
     return ctx.json({ message: 'Request is not valid' }, 400);
   }
 
-  const timestampPromise = getTimestamp(ctx);
+  const timestampPromise = getRefreshTimestamp(ctx.env);
 
   const result = !isNoCache(ctx)
-    ? await getSearchResources(ctx.env, filter)
-    : await getSearchResources.raw(ctx.env, filter);
+    ? await findResourcesFromDB(ctx.env, filter)
+    : await findResourcesFromDB.raw(ctx.env, filter);
 
   const complete = result.length > filter.pageSize;
   const resources = resolveQueryResult(result.slice(0, filter.pageSize));

packages/worker/src/scheduled.ts

@@ -7,6 +7,7 @@ import type { Env } from './types';
 
 import { makePrisma } from './prisma';
 import { updateRefreshTimestamp } from './state';
+import { findResourcesFromDB, PrefetchFilter } from './query';
 
 const teams = new Set<number>();
 const users = new Set<number>();
@@ -73,6 +74,12 @@ export async function handleScheduled(env: Env) {
 
   if (sum > 0) {
     await updateRefreshTimestamp(env);
+    await Promise.all(
+      PrefetchFilter.map(async (filter) => {
+        await findResourcesFromDB.clear(env, filter);
+        await findResourcesFromDB(env, filter);
+      })
+    );
   }
 
   return { count: sum };