chore(server): close before flush

axiomhq · Apr 16, 2024 · c5550cc · c5550cc
1 parent 18e65c0
commit c5550cc
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/server/server.go b/server/server.go
@@ -122,6 +122,8 @@ func (srv *Server) Run(ctx context.Context) {
 	for {
 		select {
 		case <-ctx.Done():
+			srv.tcpCloser.Close()
+			srv.udpCloser.Close()
 			srv.Flush()
 			return
 		case <-ticker.C: