From 1ca1f3dff9a9929e094d82217bc75f13f9fc00d7 Mon Sep 17 00:00:00 2001
From: Simon Cruanes <simon.cruanes.2007@m4x.org>
Date: Fri, 4 Aug 2023 11:01:24 -0400
Subject: [PATCH] fix: wait for threads to terminate when shutting down server

---
 src/Tiny_httpd_server.ml | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/Tiny_httpd_server.ml b/src/Tiny_httpd_server.ml
index e59fa360..b1d6fcd0 100644
--- a/src/Tiny_httpd_server.ml
+++ b/src/Tiny_httpd_server.ml
@@ -970,6 +970,9 @@ module Unix_tcp_server_ = struct
                   k "Unix.accept or Thread.create raised an exception: %s"
                     (Printexc.to_string e))
           done;
+
+          (* Wait for all threads to be done: this only works if all threads are done. *)
+          Sem_.acquire self.sem_max_connections.max self.sem_max_connections;
           ());
     }
 end