shutdown gracefully after build

crates/napi/src/next_api/project.rs

@@ -432,6 +432,13 @@ pub async fn project_update(
     Ok(())
 }
 
+#[napi(ts_return_type = "{ __napiType: \"Project\" }")]
+pub async fn project_shutdown(
+    #[napi(ts_arg_type = "{ __napiType: \"Project\" }")] project: External<ProjectInstance>,
+) {
+    project.turbo_tasks.stop_and_wait().await;
+}
+
 #[napi(object)]
 #[derive(Default)]
 struct AppPageNapiRoute {

packages/next/src/build/index.ts

@@ -1249,6 +1249,7 @@ export default async function build(
       async function turbopackBuild(): Promise<{
         duration: number
         buildTraceContext: undefined
+        shutdownPromise: Promise<void>
       }> {
         if (!IS_TURBOPACK_BUILD) {
           throw new Error("next build doesn't support turbopack yet")
@@ -1464,6 +1465,8 @@ export default async function build(
           }
         }
 
+        const shutdownPromise = project.shutdown()
+
         if (warnings.length > 0) {
           Log.warn(
             `Turbopack build collected ${warnings.length} warnings:\n${warnings
@@ -1487,6 +1490,7 @@ export default async function build(
         return {
           duration: process.hrtime(startTime)[0],
           buildTraceContext: undefined,
+          shutdownPromise,
         }
       }
 
@@ -1531,9 +1535,15 @@ export default async function build(
         },
       })
 
+      let shutdownPromise = Promise.resolve()
       if (!isGenerateMode) {
         if (turboNextBuild) {
-          const { duration: compilerDuration, ...rest } = await turbopackBuild()
+          const {
+            duration: compilerDuration,
+            shutdownPromise: p,
+            ...rest
+          } = await turbopackBuild()
+          shutdownPromise = p
           traceMemoryUsage('Finished build', nextBuildSpan)
 
           buildTraceContext = rest.buildTraceContext
@@ -3566,6 +3576,8 @@ export default async function build(
       await nextBuildSpan
         .traceChild('telemetry-flush')
         .traceAsyncFn(() => telemetry.flush())
+
+      await shutdownPromise
     })
   } finally {
     // Ensure we wait for lockfile patching if present

packages/next/src/build/swc/index.ts

@@ -679,6 +679,8 @@ export interface Project {
     aggregationMs: number
   ): AsyncIterableIterator<TurbopackResult<UpdateMessage>>
 
+  shutdown(): Promise<void>
+
   onExit(): Promise<void>
 }
 
@@ -1100,6 +1102,10 @@ function bindingToApi(
       return subscription
     }
 
+    shutdown(): Promise<void> {
+      return binding.projectShutdown(this._nativeProject)
+    }
+
     onExit(): Promise<void> {
       return binding.projectOnExit(this._nativeProject)
     }