fix: add isColdStart to request context

webiny · Apr 29, 2020 · 7807688 · 7807688
1 parent c2486e5
commit 7807688
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/packages/http-handler-apollo-server/src/plugins/apolloHandler.ts b/packages/http-handler-apollo-server/src/plugins/apolloHandler.ts
@@ -13,12 +13,14 @@ function normalizeEvent(event) {
 }
 
 let cache;
+let isColdStart = true;
 
 const plugin: CreateApolloHandlerPlugin = {
     name: "handler-apollo-server-create-handler",
     type: "handler-apollo-server-create-handler",
     async create({ context, options }) {
         if (cache) {
+            isColdStart = false;
             return cache;
         }
 
@@ -46,7 +48,8 @@ const plugin: CreateApolloHandlerPlugin = {
             schema,
             context: async ({ event }) => ({
                 event,
-                plugins: context.plugins
+                plugins: context.plugins,
+                isColdStart
             })
         });
 

diff --git a/packages/http-handler-apollo-server/src/plugins/createHandlerApolloServer.ts b/packages/http-handler-apollo-server/src/plugins/createHandlerApolloServer.ts
@@ -16,6 +16,7 @@ export default (options: HttpHandlerApolloServerOptions = {}): HttpHandlerPlugin
     },
     async handle({ args, context }) {
         const [event, handlerContext] = args;
+
         try {
             const createApolloHandlerPlugin = context.plugins.byName<CreateApolloHandlerPlugin>(
                 "handler-apollo-server-create-handler"

diff --git a/packages/http-handler-apollo-server/src/types.ts b/packages/http-handler-apollo-server/src/types.ts
@@ -31,5 +31,5 @@ export type CreateApolloHandlerPlugin = Plugin & {
 export type CreateSchemaPlugin = Plugin & {
     name: "handler-apollo-server-create-schema";
     type: "handler-apollo-server-create-schema";
-    create(params: { plugins: PluginsContainer }): { schema: GraphQLSchema };
+    create(params: { plugins: PluginsContainer; }): { schema: GraphQLSchema };
 };