Fix logout process.

tetafro · Sep 6, 2023 · 24f76de · 24f76de
1 parent ed38a1e
commit 24f76de
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/collector.go b/collector.go
@@ -34,6 +34,11 @@ func (c *Collector) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 		http500(w, "Collector error")
 		return
 	}
+	defer func() {
+		if err := client.Logout(r.Context()); err != nil {
+			log.Fatalf("Failed to logout: %v", err)
+		}
+	}()
 
 	reg := prometheus.NewRegistry()
 	c.collectCMState(r.Context(), reg, client)