feat: close server on sigint/sigterm

close #1525
vitejs · Jan 14, 2021 · 4338d7d · 4338d7d
1 parent 25e9c44
commit 4338d7d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/packages/vite/src/node/server/index.ts b/packages/vite/src/node/server/index.ts
@@ -248,6 +248,9 @@ export async function createServer(
     }
   }
 
+  process.once('SIGINT', server.close)
+  process.once('SIGTERM', server.close)
+
   watcher.on('change', async (file) => {
     file = normalizePath(file)
     // invalidate module graph cache on file change

diff --git a/packages/vite/src/node/server/pluginContainer.ts b/packages/vite/src/node/server/pluginContainer.ts
@@ -359,6 +359,7 @@ export async function createPluginContainer(
   }
 
   let nestedResolveCall = 0
+  let closed = false
 
   const container: PluginContainer = {
     options: await (async () => {
@@ -573,13 +574,15 @@ export async function createPluginContainer(
     },
 
     async close() {
+      if (closed) return
       const ctx = new Context()
       await Promise.all(
         plugins.map((p) => p.buildEnd && p.buildEnd.call(ctx as any))
       )
       await Promise.all(
         plugins.map((p) => p.closeBundle && p.closeBundle.call(ctx as any))
       )
+      closed = true
     }
   }