|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
"use strict"; |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
function getCacheId(type, request, withContext) { |
|
|
return JSON.stringify({ |
|
|
type, |
|
|
context: withContext ? request.context : "", |
|
|
path: request.path, |
|
|
query: request.query, |
|
|
fragment: request.fragment, |
|
|
request: request.request, |
|
|
}); |
|
|
} |
|
|
|
|
|
module.exports = class UnsafeCachePlugin { |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
constructor(source, filterPredicate, cache, withContext, target) { |
|
|
this.source = source; |
|
|
this.filterPredicate = filterPredicate; |
|
|
this.withContext = withContext; |
|
|
this.cache = cache; |
|
|
this.target = target; |
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
apply(resolver) { |
|
|
const target = resolver.ensureHook(this.target); |
|
|
resolver |
|
|
.getHook(this.source) |
|
|
.tapAsync("UnsafeCachePlugin", (request, resolveContext, callback) => { |
|
|
if (!this.filterPredicate(request)) return callback(); |
|
|
const isYield = typeof resolveContext.yield === "function"; |
|
|
const cacheId = getCacheId( |
|
|
isYield ? "yield" : "default", |
|
|
request, |
|
|
this.withContext, |
|
|
); |
|
|
const cacheEntry = this.cache[cacheId]; |
|
|
if (cacheEntry) { |
|
|
if (isYield) { |
|
|
const yield_ = |
|
|
|
|
|
(resolveContext.yield); |
|
|
if (Array.isArray(cacheEntry)) { |
|
|
for (const result of cacheEntry) yield_(result); |
|
|
} else { |
|
|
yield_(cacheEntry); |
|
|
} |
|
|
return callback(null, null); |
|
|
} |
|
|
return callback(null, (cacheEntry)); |
|
|
} |
|
|
|
|
|
|
|
|
let yieldFn; |
|
|
|
|
|
let yield_; |
|
|
|
|
|
const yieldResult = []; |
|
|
if (isYield) { |
|
|
yieldFn = resolveContext.yield; |
|
|
yield_ = (result) => { |
|
|
yieldResult.push(result); |
|
|
}; |
|
|
} |
|
|
|
|
|
resolver.doResolve( |
|
|
target, |
|
|
request, |
|
|
null, |
|
|
yield_ ? { ...resolveContext, yield: yield_ } : resolveContext, |
|
|
(err, result) => { |
|
|
if (err) return callback(err); |
|
|
if (isYield) { |
|
|
if (result) yieldResult.push(result); |
|
|
for (const result of yieldResult) { |
|
|
|
|
|
(yieldFn)(result); |
|
|
} |
|
|
this.cache[cacheId] = yieldResult; |
|
|
return callback(null, null); |
|
|
} |
|
|
if (result) return callback(null, (this.cache[cacheId] = result)); |
|
|
callback(); |
|
|
}, |
|
|
); |
|
|
}); |
|
|
} |
|
|
}; |
|
|
|