connections metric

kdudkov · Nov 10, 2023 · 98b68ca · 98b68ca
1 parent 6bf6823
commit 98b68ca
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/cmd/goatak_server/main.go b/cmd/goatak_server/main.go
@@ -173,12 +173,15 @@ func (app *App) NewCotMessage(msg *cot.CotMessage) {
 
 func (app *App) AddClientHandler(ch client.ClientHandler) {
 	app.handlers.Store(ch.GetName(), ch)
+	connectionsMetric.Inc()
 }
 
 func (app *App) RemoveClientHandler(name string) {
 	if _, ok := app.handlers.Load(name); ok {
 		app.Logger.Infof("remove handler: %s", name)
-		app.handlers.Delete(name)
+		if _, ok := app.handlers.LoadAndDelete(name); ok {
+			connectionsMetric.Dec()
+		}
 	}
 }
 

diff --git a/cmd/goatak_server/metrics.go b/cmd/goatak_server/metrics.go
@@ -11,4 +11,10 @@ var (
 		Name:      "cots_processed",
 		Help:      "The total number of cots processed",
 	})
+
+	connectionsMetric = promauto.NewGauge(prometheus.GaugeOpts{
+		Namespace: "goatak",
+		Name:      "connections",
+		Help:      "The total number of connections",
+	})
 )