Keep seat reconciliation alive after the response

Author: RhysSullivan

apps/cloud/src/extensions/billing/member-seats.ts

@@ -3,6 +3,7 @@
 // ---------------------------------------------------------------------------
 
 import { Effect } from "effect";
+import { waitUntil } from "cloudflare:workers";
 
 import { WorkOSClient } from "../../auth/workos";
 import { AutumnService } from "./service";
@@ -38,16 +39,16 @@ export const reportMemberSeats = (
 /**
  * Fork `reportMemberSeats` off the calling request, mirroring how execution
  * tracking is forked: billing must never stall or fail a user-facing
- * request. Only boot-scoped services are captured (WorkOS + Autumn — no
- * request-scoped resources), so the forked fiber cannot outlive anything it
- * depends on.
+ * request. Cloudflare owns the promise through waitUntil, so the recount can
+ * finish after the response. Only boot-scoped WorkOS and Autumn services are
+ * captured.
  */
 export const forkReportMemberSeats = (
   organizationId: string,
 ): Effect.Effect<void, never, WorkOSClient | AutumnService> =>
   Effect.gen(function* () {
     const ctx = yield* Effect.context<WorkOSClient | AutumnService>();
     yield* Effect.sync(() => {
-      Effect.runForkWith(ctx)(reportMemberSeats(organizationId));
+      waitUntil(Effect.runPromiseWith(ctx)(reportMemberSeats(organizationId)));
     });
   });

apps/cloud/test-stubs/cloudflare-workers.ts

@@ -7,6 +7,10 @@
 // from `process.env` into this stub at import time. Without this bridge
 // the test DbService would dial the default postgres port instead of the
 // PGlite socket server started by `scripts/test-globalsetup.ts`.
+import { vi } from "vitest";
+
+export const waitUntil = vi.fn<(work: Promise<unknown>) => void>();
+
 export const env: Record<string, unknown> = {
   DATABASE_URL: process.env.DATABASE_URL,
   WORKOS_API_KEY: process.env.WORKOS_API_KEY,