1 files changed, 13 insertions, 0 deletions

diff --git a/internal/handler/websocket/handler.go b/internal/handler/websocket/handler.go
index a7b73ec..cb089b1 100644
--- a/internal/handler/websocket/handler.go
+++ b/internal/handler/websocket/handler.go
@@ -26,6 +26,8 @@ type AuthStore interface {
 }
 
 type MetricsRecorder interface {
+        IncrementConnections()
+        DecrementConnections()
         IncrementSubscriptions()
         DecrementSubscriptions()
         SetActiveSubscriptions(count int)
@@ -114,6 +116,11 @@ func (h *Handler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
         }
         defer conn.Close(websocket.StatusNormalClosure, "")
 
+        // Track connection metrics
+        if h.metrics != nil {
+                h.metrics.IncrementConnections()
+        }
+
         ctx := r.Context()
         clientSubs := make(map[string]*subscription.Subscription)
         state := &connState{
@@ -121,6 +128,12 @@ func (h *Handler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
         }
 
         defer func() {
+                // Decrement connection count
+                if h.metrics != nil {
+                        h.metrics.DecrementConnections()
+                }
+
+                // Clean up subscriptions
                 count := len(clientSubs)
                 for subID := range clientSubs {
                         h.subs.Remove(subID)