@fortedigital/nextjs-cache-handler
Version:
Next.js cache handlers
52 lines (50 loc) • 1.39 kB
JavaScript
// 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
};