@orion-js/graphql 2.8.0 → 2.8.1

package/lib/getOperationName.js

@@ -0,0 +1,24 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.default = getOperationName;
+
+async function getOperationName(params) {
+  const {
+    request,
+    getBodyJSON
+  } = params;
+
+  if (request.method !== 'GET') {
+    const data = await getBodyJSON();
+    return data.operationName;
+  }
+
+  if (request.method === 'GET') {
+    // handle persisted queries
+    const queryParams = new URLSearchParams(request.url.split('?')[1]);
+    return queryParams.get('operationName');
+  }
+}

package/lib/startGraphQL.js

@@ -19,6 +19,10 @@ var _startWebsocket = _interopRequireDefault(require("./startWebsocket"));
 
 var _apolloServerCore = require("apollo-server-core");
 
+var _apolloServerCaching = require("apollo-server-caching");
+
+var _getOperationName = _interopRequireDefault(require("./getOperationName"));
+
 async function _default(options) {
   const apolloOptions = await (0, _getApolloOptions.default)(options);
   (0, _startGraphiQL.default)(apolloOptions, options);
@@ -31,6 +35,7 @@ async function _default(options) {
   await apolloServer.start();
   const handler = apolloServer.createHandler(); // highlight-line
 
+  const cacheControlCache = apolloOptions.persistedQueries?.cache ? apolloOptions.persistedQueries.cache : new _apolloServerCaching.InMemoryLRUCache();
   (0, _app.route)('/graphql', async function (params) {
     const {
       request,
@@ -38,6 +43,8 @@ async function _default(options) {
       viewer,
       getBodyJSON
     } = params;
+    const operationName = await (0, _getOperationName.default)(params);
+    const cacheKey = operationName ? `orion-cache-control-${operationName}` : null;
 
     if (options.executeGraphQLCache) {
       try {
@@ -57,6 +64,11 @@ async function _default(options) {
         const result = await options.executeGraphQLCache(params, fallback);
 
         if (result) {
+          if (result.cacheControl && cacheKey) {
+            cacheControlCache.set(cacheKey, result.cacheControl);
+            response.setHeader('Cache-Control', result.cacheControl);
+          }
+
           return result;
         }
       } catch (error) {
@@ -64,6 +76,22 @@ async function _default(options) {
       }
     }
 
+    if (cacheKey) {
+      let shouldCache = true;
+
+      if (apolloOptions?.shouldAddCacheControl) {
+        shouldCache = apolloOptions.shouldAddCacheControl(operationName);
+      }
+
+      if (shouldCache) {
+        const cacheControl = await cacheControlCache.get(cacheKey);
+
+        if (cacheControl) {
+          response.setHeader('Cache-Control', cacheControl);
+        }
+      }
+    }
+
     request._orionjsViewer = viewer;
     handler(request, response);
   });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@orion-js/graphql",
-  "version": "2.8.0",
+  "version": "2.8.1",
   "main": "lib/index.js",
   "author": "nicolaslopezj",
   "license": "MIT",
@@ -13,7 +13,7 @@
   "dependencies": {
     "@orion-js/schema": "^2.6.6",
     "apollo-server-core": "3.4.0",
-    "apollo-server-micro": "3.13.0",
+    "apollo-server-micro": "3.8.0",
     "deep-sort-object": "^1.0.2",
     "graphql-iso-date": "^3.6.1",
     "graphql-subscriptions": "^1.2.1",