Fix metrics

gatewayd-io · Oct 15, 2023 · 8013cb9 · 8013cb9
1 parent f02a31f
commit 8013cb9
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/network/client.go b/network/client.go
@@ -236,6 +236,8 @@ func (c *Client) Reconnect() error {
 
 	if c.conn != nil {
 		c.Close()
+	} else {
+		metrics.ServerConnections.Dec()
 	}
 	c.connected.Store(false)
 

diff --git a/network/server_test.go b/network/server_test.go
@@ -16,6 +16,7 @@ import (
 	"github.com/gatewayd-io/gatewayd/logging"
 	"github.com/gatewayd-io/gatewayd/plugin"
 	"github.com/gatewayd-io/gatewayd/pool"
+	"github.com/prometheus/client_golang/prometheus"
 	"github.com/rs/zerolog"
 	"github.com/stretchr/testify/assert"
 	"google.golang.org/grpc"
@@ -25,6 +26,9 @@ import (
 func TestRunServer(t *testing.T) {
 	errs := make(chan error)
 
+	// Reset prometheus metrics.
+	prometheus.DefaultRegisterer = prometheus.NewRegistry()
+
 	logger := logging.NewLogger(context.Background(), logging.LoggerConfig{
 		Output: []config.LogOutput{
 			config.File,