small fix

README.md

@@ -7,7 +7,7 @@ A minimal Go WebSocket relay server with SSL support for P2P connections.
 ```bash
 go mod tidy
 # Configure via config.yaml (see config.yaml for options)
-go run main.go
+go run main.go --config-file=./config.yaml
 ```
 
 ## Configuration
@@ -18,16 +18,16 @@ Edit `config.yaml` to configure:
 
 ## Usage
 
-- WebSocket endpoint: `/ws`
+- WebSocket endpoint: `/`
 - All WebSocket messages are relayed to all connected clients
 
 ## Testing
 
 ```javascript
 // For TLS enabled (default config)
-const ws = new WebSocket('wss://localhost:8443/ws');
+const ws = new WebSocket('wss://localhost:8443/');
 // For HTTP only
-// const ws = new WebSocket('ws://localhost:8443/ws');
+// const ws = new WebSocket('ws://localhost:8443/');
 ws.onmessage = (event) => console.log('Received:', event.data);
 ws.send('Hello from client!');
 ```

config.example.yaml

@@ -4,3 +4,7 @@ server:
     enabled: true
     cert_file: cert.pem
     key_file: key.pem
+
+metrics:
+  enabled: true
+  port: 9090

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

@@ -1,17 +1,22 @@
 package main
 
 import (
+	"flag"
 	"fmt"
 	"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() {
-	cfg, err := config.Load("config.yaml")
+	configFile := flag.String("config-file", "config.yaml", "Path to configuration file")
+	flag.Parse()
+
+	cfg, err := config.Load(*configFile)
 	if err != nil {
 		log.Fatal("Failed to load config:", err)
 	}
@@ -30,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 {