Fix worker are notified to stop with non_graceful shutdown (#333)

actix-server/CHANGES.md

@@ -1,6 +1,10 @@
 # Changes
 
 ## Unreleased - 2021-xx-xx
+* Server shutdown would notify all workers to exit regardless if shutdown is graceful.
+  This would make all worker shutdown immediately in force shutdown case. [#333]
+  
+[#333]: https://github.com/actix/actix-net/pull/333
 
 
 ## 2.0.0-beta.4 - 2021-04-01

actix-server/src/builder.rs

@@ -381,45 +381,29 @@ impl ServerBuilder {
                 let notify = std::mem::take(&mut self.notify);
 
                 // stop workers
-                if !self.handles.is_empty() && graceful {
+                let stop = self
-                    let iter = self
                     .handles
                     .iter()
                     .map(move |worker| worker.1.stop(graceful))
                     .collect();
 
-                    let fut = join_all(iter);
-
                 rt::spawn(async move {
-                        let _ = fut.await;
+                    if graceful {
+                        let _ = join_all(stop).await;
+                    }
+
                     if let Some(tx) = completion {
                         let _ = tx.send(());
                     }
                     for tx in notify {
                         let _ = tx.send(());
                     }
+
                     if exit {
-                            rt::spawn(async {
                         sleep(Duration::from_millis(300)).await;
                         System::current().stop();
-                            });
                     }
                 });
-                } else {
-                    // we need to stop system if server was spawned
-                    if self.exit {
-                        rt::spawn(async {
-                            sleep(Duration::from_millis(300)).await;
-                            System::current().stop();
-                        });
-                    }
-                    if let Some(tx) = completion {
-                        let _ = tx.send(());
-                    }
-                    for tx in notify {
-                        let _ = tx.send(());
-                    }
-                }
             }
             ServerCommand::WorkerFaulted(idx) => {
                 let mut found = false;