Wait for lifecycle shutdown to finish (#170)

hummingbird-project · Feb 21, 2023 · b9bdaf7 · b9bdaf7
1 parent 9042192
commit b9bdaf7
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/Sources/Hummingbird/Application.swift b/Sources/Hummingbird/Application.swift
@@ -154,7 +154,12 @@ public final class HBApplication: HBExtensible {
 
     /// Shutdown application
     public func stop() {
-        self.lifecycle.shutdown()
+        let stopSemaphore = DispatchSemaphore(value: 0)
+
+        self.lifecycle.shutdown { _ in
+            stopSemaphore.signal()
+        }
+        stopSemaphore.wait()
     }
 
     /// middleware applied to requests