Close Zeroconf server when exiting

devgianlu · Oct 19, 2023 · 55f682e · 55f682e
1 parent df81ee5
commit 55f682e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/zeroconf/zeroconf.go b/zeroconf/zeroconf.go
@@ -225,6 +225,8 @@ func (z *Zeroconf) handleAddUser(writer http.ResponseWriter, request *http.Reque
 type HandleNewRequestFunc func(req NewUserRequest) bool
 
 func (z *Zeroconf) Serve(handler HandleNewRequestFunc) error {
+	defer z.server.Shutdown()
+
 	mux := http.NewServeMux()
 	mux.HandleFunc("/", func(writer http.ResponseWriter, request *http.Request) {
 		if err := request.ParseForm(); err != nil {