UNPKG

@fortedigital/nextjs-cache-handler

Version:
52 lines (50 loc) 1.39 kB
// src/helpers/withAbortSignal.ts function withAbortSignal(promiseFn, signal) { if (!signal) return promiseFn(); return new Promise((resolve, reject) => { if (signal.aborted) { return reject(new Error("Aborted")); } let settled = false; const onAbort = () => { if (!settled) { settled = true; reject(new Error("Operation aborted")); } }; signal.addEventListener("abort", onAbort, { once: true }); promiseFn().then((res) => { if (!settled) { settled = true; signal.removeEventListener("abort", onAbort); resolve(res); } }).catch((err) => { if (!settled) { settled = true; signal.removeEventListener("abort", onAbort); reject(err); } }); }); } // src/helpers/withAbortSignalProxy.ts function withAbortSignalProxy(obj, defaultSignal) { function createProxy(signal) { const handler = { get(target, prop, receiver) { if (prop === "withAbortSignal") { return (s) => createProxy(s); } const orig = Reflect.get(target, prop, receiver); if (typeof orig !== "function") return orig; return (...args) => withAbortSignal(() => orig.apply(receiver, args), signal); } }; return new Proxy(obj, handler); } return createProxy(defaultSignal); } export { withAbortSignalProxy };