fix: gracefully shutting down (#4145)

webpack · Apr 10, 2024 · 90720e2 · 90720e2
1 parent 3818d7c
commit 90720e2
Showing 1 changed file with 30 additions and 4 deletions.
diff --git a/packages/webpack-cli/src/webpack-cli.ts b/packages/webpack-cli/src/webpack-cli.ts
@@ -72,6 +72,8 @@ const WEBPACK_DEV_SERVER_PACKAGE = WEBPACK_DEV_SERVER_PACKAGE_IS_CUSTOM
   ? (process.env.WEBPACK_DEV_SERVER_PACKAGE as string)
   : "webpack-dev-server";
 
+const EXIT_SIGNALS = ["SIGINT", "SIGTERM"];
+
 interface Information {
   Binaries?: string[];
   Browsers?: string[];
@@ -2532,11 +2534,35 @@ class WebpackCLI implements IWebpackCLI {
           : compiler.options.watch,
       );
 
-    if (isWatch(compiler) && this.needWatchStdin(compiler)) {
-      process.stdin.on("end", () => {
-        process.exit(0);
+    if (isWatch(compiler)) {
+      let needForceShutdown = false;
+
+      EXIT_SIGNALS.forEach((signal) => {
+        const listener = () => {
+          if (needForceShutdown) {
+            process.exit(0);
+          }
+
+          this.logger.info(
+            "Gracefully shutting down. To force exit, press ^C again. Please wait...",
+          );
+
+          needForceShutdown = true;
+
+          compiler.close(() => {
+            process.exit(0);
+          });
+        };
+
+        process.on(signal, listener);
       });
-      process.stdin.resume();
+
+      if (this.needWatchStdin(compiler)) {
+        process.stdin.on("end", () => {
+          process.exit(0);
+        });
+        process.stdin.resume();
+      }
     }
   }
 }