fix(prerender): call nitroApp close hook when done prerendering (#2287

)
unjs · Mar 21, 2024 · 5184e45 · 5184e45
1 parent dc83a2e
commit 5184e45
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/src/prerender.ts b/src/prerender.ts
@@ -81,9 +81,9 @@ export async function prerender(nitro: Nitro) {
  nitroRenderer.options.output.serverDir,
  "index.mjs"
  );
- const { localFetch } = (await import(
+ const { closePrerenderer, localFetch } = (await import(
  pathToFileURL(serverEntrypoint).href
- )) as { localFetch: $Fetch };
+ )) as { closePrerenderer: () => Promise<void>; localFetch: $Fetch };
 
  // Create route rule matcher
  const _routeRulesMatcher = toRouteMatcher(
@@ -296,6 +296,8 @@ export async function prerender(nitro: Nitro) {
  interval: nitro.options.prerender.interval,
  });
 
+ await closePrerenderer();
+
  await nitro.hooks.callHook("prerender:done", {
  prerenderedRoutes: nitro._prerenderedRoutes,
  failedRoutes: [...failedRoutes],

diff --git a/src/runtime/entries/nitro-prerenderer.ts b/src/runtime/entries/nitro-prerenderer.ts
@@ -3,6 +3,7 @@ import { nitroApp } from "../app";
 import { trapUnhandledNodeErrors } from "../utils";
 
 export const localFetch = nitroApp.localFetch;
+export const closePrerenderer = () => nitroApp.hooks.callHook("close");
 
 // Trap unhandled errors
 trapUnhandledNodeErrors();