give access to the server context in onResultProcess

.changeset/cold-mails-film.md

@@ -0,0 +1,6 @@
+---
+'graphql-yoga': minor
+---
+
+Expose server context in `onResultProcessHook`. In particular, this gives access to the `waitUntil`
+method to cleanly handle hanging promises.

packages/graphql-yoga/src/process-request.ts

@@ -4,19 +4,21 @@ import { ExecutionArgs } from '@graphql-tools/executor';
 import { OnResultProcess, ResultProcessor, ResultProcessorInput } from './plugins/types.js';
 import { FetchAPI, GraphQLParams } from './types.js';
 
-export async function processResult({
+export async function processResult<TServerContext>({
   request,
   result,
   fetchAPI,
   onResultProcessHooks,
+  serverContext,
 }: {
   request: Request;
   result: ResultProcessorInput;
   fetchAPI: FetchAPI;
   /**
    * Response Hooks
    */
-  onResultProcessHooks: OnResultProcess[];
+  onResultProcessHooks: OnResultProcess<TServerContext>[];
+  serverContext: TServerContext;
 }) {
   let resultProcessor: ResultProcessor | undefined;
 
@@ -36,6 +38,7 @@ export async function processResult({
         resultProcessor = newResultProcessor;
         acceptedMediaType = newAcceptedMimeType;
       },
+      serverContext,
     });
   }
 

packages/graphql-yoga/src/server.ts

@@ -21,6 +21,7 @@ import {
   ServerAdapterRequestHandler,
   useCORS,
   useErrorHandling,
+  type ServerAdapterInitialContext,
 } from '@whatwg-node/server';
 import { handleError, isAbortError } from './error.js';
 import { isGETRequest, parseGETRequest } from './plugins/request-parser/get.js';
@@ -350,20 +351,22 @@ export class YogaServer<
       // Middlewares after the GraphQL execution
       useResultProcessors(),
       useErrorHandling<TServerContext & YogaInitialContext & ServerAdapterInitialContext>(
-        (error, request) => {
-        const errors = handleError(error, this.maskedErrorsOpts, this.logger);
+        (error, request, serverContext) => {
+          const errors = handleError(error, this.maskedErrorsOpts, this.logger);
 
-        const result = {
-          errors,
-        };
+          const result = {
+            errors,
+          };
 
-        return processResult({
-          request,
-          result,
-          fetchAPI: this.fetchAPI,
-          onResultProcessHooks: this.onResultProcessHooks,
-        });
-      }),
+          return processResult({
+            request,
+            result,
+            fetchAPI: this.fetchAPI,
+            onResultProcessHooks: this.onResultProcessHooks,
+            serverContext,
+          });
+        },
+      ),
 
       ...(options?.plugins ?? []),
       // To make sure those are called at the end
@@ -621,6 +624,7 @@ export class YogaServer<
       result,
       fetchAPI: this.fetchAPI,
       onResultProcessHooks: this.onResultProcessHooks,
+      serverContext,
     });
   };
 }