use dataFetcher itself as the key for the weakmap

2026-09-13 06:09:21 +00:00 · 2025-06-12 14:24:11 +02:00
2 changed files with 654 additions and 612 deletions
@@ -1,20 +1,17 @@
-import { cache } from 'react';
-
-// This is used to create a context specific to the current request.
-// This version works both in cloudflare and in vercel.
-const getRequestContext = cache(() => ({}));
+import type { GitBookDataFetcher } from './types';

 /**
 * Wrap a function by preventing concurrent executions of the same function.
 * With a logic to work per-request in Cloudflare Workers.
 */
 export function withoutConcurrentExecution<ArgsType extends any[], ReturnType>(
+    dataFetcherObject: GitBookDataFetcher,
    wrapped: (key: string, ...args: ArgsType) => Promise<ReturnType>
 ): (cacheKey: string, ...args: ArgsType) => Promise<ReturnType> {
    const globalPromiseCache = new WeakMap<object, Map<string, Promise<ReturnType>>>();

    return (key: string, ...args: ArgsType) => {
-        const globalContext = getRequestContext();
+        const globalContext = dataFetcherObject;

        /**
         * Cache storage that is scoped to the current request when executed in Cloudflare Workers,