small fix

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",
 	})
 )

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 {