fix: cloudflare waitUntil not working as inteded with getRuntime (#7419)

Co-authored-by: Bjorn Lu <bjornlu.dev@gmail.com>

1 files changed, 5 insertions, 1 deletions

diff --git a/packages/integrations/cloudflare/src/server.directory.ts b/packages/integrations/cloudflare/src/server.directory.ts
index a0f3299fd..649d0d99b 100644
--- a/packages/integrations/cloudflare/src/server.directory.ts
+++ b/packages/integrations/cloudflare/src/server.directory.ts
@@ -1,4 +1,4 @@
-import type { Request as CFRequest } from '@cloudflare/workers-types';
+import type { Request as CFRequest, EventContext } from '@cloudflare/workers-types';
 import type { SSRManifest } from 'astro';
 import { App } from 'astro/app';
 import { getProcessEnvProxy, isNode } from './util.js';
@@ -17,6 +17,7 @@ export function createExports(manifest: SSRManifest) {
 	}: {
 		request: Request & CFRequest;
 		next: (request: Request) => void;
+		waitUntil: EventContext<unknown, any, unknown>['waitUntil'];
 	} & Record<string, unknown>) => {
 		process.env = runtimeEnv.env as any;
 
@@ -35,6 +36,9 @@ export function createExports(manifest: SSRManifest) {
 			);
 			Reflect.set(request, Symbol.for('runtime'), {
 				...runtimeEnv,
+				waitUntil: (promise: Promise<any>) => {
+					runtimeEnv.waitUntil(promise);
+				},
 				name: 'cloudflare',
 				next,
 				caches,