From f69355d69d25687624c22c441aaf9fd12e20140b Mon Sep 17 00:00:00 2001
From: savinmax <savinmax@amazon.com>
Date: Mon, 4 Aug 2025 11:11:10 +0200
Subject: [PATCH] small fix

---
 internal/metrics/metrics.go | 20 ++++++++++----------
 main.go                     |  5 +++--
 2 files changed, 13 insertions(+), 12 deletions(-)

diff --git a/internal/metrics/metrics.go b/internal/metrics/metrics.go
index c99e2bf..92c0a4e 100644
--- a/internal/metrics/metrics.go
+++ b/internal/metrics/metrics.go
@@ -11,18 +11,18 @@ var (
 		Help: "Number of currently connected WebSocket clients",
 	})
 
-	MessagesTotal = promauto.NewCounter(prometheus.CounterOpts{
-		Name: "websocket_messages_total",
-		Help: "Total number of WebSocket messages processed",
+	MessagesTotal = promauto.NewGauge(prometheus.GaugeOpts{
+		Name: "websocket_message",
+		Help: "Number of WebSocket messages processed",
 	})
 
-	ConnectionsTotal = promauto.NewCounter(prometheus.CounterOpts{
-		Name: "websocket_connections_total",
-		Help: "Total number of WebSocket connections established",
+	ConnectionsTotal = promauto.NewGauge(prometheus.GaugeOpts{
+		Name: "websocket_connection",
+		Help: "Number of WebSocket connections established",
 	})
 
-	DisconnectionsTotal = promauto.NewCounter(prometheus.CounterOpts{
-		Name: "websocket_disconnections_total",
-		Help: "Total number of WebSocket disconnections",
+	DisconnectionsTotal = promauto.NewGauge(prometheus.GaugeOpts{
+		Name: "websocket_disconnection",
+		Help: "Number of WebSocket disconnections",
 	})
-)
\ No newline at end of file
+)
diff --git a/main.go b/main.go
index 9c5fb16..8ead97f 100644
--- a/main.go
+++ b/main.go
@@ -6,9 +6,10 @@ import (
 	"log"
 	"net/http"
 
-	"github.com/prometheus/client_golang/prometheus/promhttp"
 	"websocket-relay/internal/config"
 	"websocket-relay/internal/hub"
+
+	"github.com/prometheus/client_golang/prometheus/promhttp"
 )
 
 func main() {
@@ -34,7 +35,7 @@ func main() {
 	}
 
 	mux := http.NewServeMux()
-	mux.HandleFunc("/ws", h.HandleWebSocket)
+	mux.HandleFunc("/", h.HandleWebSocket)
 
 	addr := fmt.Sprintf(":%d", cfg.Server.Port)
 	if cfg.Server.TLS.Enabled {