next
Version:
The React Framework
47 lines (46 loc) • 2.15 kB
JavaScript
/** Run a callback, and execute any *new* revalidations added during its runtime. */ export async function withExecuteRevalidates(store, callback) {
if (!store) {
return callback();
}
// If we executed any revalidates during the request, then we don't want to execute them again.
// save the state so we can check if anything changed after we're done running callbacks.
const savedRevalidationState = cloneRevalidationState(store);
try {
return await callback();
} finally{
// Check if we have any new revalidates, and if so, wait until they are all resolved.
const newRevalidates = diffRevalidationState(savedRevalidationState, cloneRevalidationState(store));
await executeRevalidates(store, newRevalidates);
}
}
function cloneRevalidationState(store) {
return {
revalidatedTags: store.revalidatedTags ? [
...store.revalidatedTags
] : [],
pendingRevalidates: {
...store.pendingRevalidates
},
pendingRevalidateWrites: store.pendingRevalidateWrites ? [
...store.pendingRevalidateWrites
] : []
};
}
function diffRevalidationState(prev, curr) {
const prevTags = new Set(prev.revalidatedTags);
const prevRevalidateWrites = new Set(prev.pendingRevalidateWrites);
return {
revalidatedTags: curr.revalidatedTags.filter((tag)=>!prevTags.has(tag)),
pendingRevalidates: Object.fromEntries(Object.entries(curr.pendingRevalidates).filter(([key])=>!(key in prev.pendingRevalidates))),
pendingRevalidateWrites: curr.pendingRevalidateWrites.filter((promise)=>!prevRevalidateWrites.has(promise))
};
}
async function executeRevalidates(workStore, { revalidatedTags, pendingRevalidates, pendingRevalidateWrites }) {
var _workStore_incrementalCache;
return Promise.all([
(_workStore_incrementalCache = workStore.incrementalCache) == null ? void 0 : _workStore_incrementalCache.revalidateTag(revalidatedTags),
...Object.values(pendingRevalidates),
...pendingRevalidateWrites
]);
}
//# sourceMappingURL=revalidation-utils.js.map