Make Shutdown blocking to ensure outstanding requests processing comp…

…letion
pin · Oct 28, 2023 · 4b9cd87 · 4b9cd87
1 parent 4baa433
commit 4b9cd87
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/server.go b/server.go
@@ -318,7 +318,9 @@ func (s *Server) Shutdown() {
 		s.Unlock()
 	}
 	s.cancelFn()
-	s.wg.Wait()
+	if !s.singlePort {
+		s.wg.Wait()
+	}
 }
 
 func (s *Server) handlePacket(localAddr net.IP, remoteAddr *net.UDPAddr, buffer []byte, n, maxBlockLen int, listener chan []byte) error {