feat(cache): add shouldCache parameter

davidbonnet · May 29, 2022 · f156eee · f156eee
1 parent 75a97fa
commit f156eee
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/cache.js b/src/cache.js
@@ -4,7 +4,7 @@ import { Buffer } from 'buffer'
 
 const { isPromise } = types
 
-export function cache({ store }) {
+export function cache({ store, shouldCache = defaultShouldCache }) {
   return async function cache(request, next) {
     if (await store.has(request)) {
       const cache = await store.get(request)
@@ -15,7 +15,7 @@ export function cache({ store }) {
       return await request.respond(cache)
     }
     const response = await next(request)
-    if (response.cache) {
+    if (shouldCache(response)) {
       const tube = new PassThrough()
       response.tube = tube
       await store.set(request, {
@@ -29,6 +29,10 @@ export function cache({ store }) {
   }
 }
 
+function defaultShouldCache(request) {
+  return request.cache
+}
+
 function buffer(tube) {
   return new Promise(function (resolve, reject) {
     const chunks = []