messages are being received on both ends now!

reactivity/tracker.go

@@ -1,6 +1,7 @@
 package reactivity
 
 import (
+	"log/slog"
 	"sync"
 )
 
@@ -56,3 +57,18 @@ func (t *Tracker) GetQuerySubscriptions(queryHash string) []string {
 	}
 	return subscriptionIDs
 }
+
+func (t *Tracker) SendMessage(clientID string, message []byte) {
+	t.mu.RLock()
+	defer t.mu.RUnlock()
+	client := t.clients[clientID]
+	if client == nil {
+		slog.Error("Tracker: Client not found", "clientID", clientID)
+		return
+	}
+	select {
+	case client.Send <- message:
+	default:
+		slog.Error("Tracker: Client send channel is full", "clientID", clientID)
+	}
+}

reactivity/websocket.go

@@ -23,7 +23,6 @@ func Handle(w http.ResponseWriter, r *http.Request, e EngineHandler, tracker *Tr
 		slog.Error("WS: Failed to upgrade to websocket", "error", err)
 		return
 	}
-	defer ws.Close()
 
 	client := NewClient(ws)
 	err = e.OnConnect(client.ID)
@@ -33,6 +32,11 @@ func Handle(w http.ResponseWriter, r *http.Request, e EngineHandler, tracker *Tr
 	}
 	defer e.OnDisconnect(client.ID)
 	tracker.Track(client)
+	defer func() {
+		tracker.Untrack(client)
+		close(client.Send)
+	}()
+	go client.WritePump()
 	for {
 		_, message, err := ws.ReadMessage()
 		if err != nil {