wisplite/tether
1
0
Fork 0
mirror of https://github.com/wisplite/tether.git synced 2026-05-01 06:22:41 -05:00
Code Issues Packages Projects Releases Wiki Activity

pull websockets out of engine

Browse Source
This commit is contained in:
wisplite
2026-04-08 10:09:43 -05:00
parent c883df4733
commit bdbb2957c1
2 changed files with 16 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files
reactivity/websocket.go
+9 -8
View File
@@ -10,14 +10,14 @@ import (
// MessageReceiver receives decoded WebSocket payloads. Implemented by tether.Engine.
type EngineHandler interface {
OnConnect(conn *websocket.Conn) error
OnDisconnect(conn *websocket.Conn) error
OnReceiveMessage(msg map[string]interface{}) error
OnConnect(clientID string) error
OnDisconnect(clientID string) error
OnReceiveMessage(clientID string, msg map[string]interface{}) error
}
var upgrader = websocket.Upgrader{}
func Handle(w http.ResponseWriter, r *http.Request, e EngineHandler) {
func Handle(w http.ResponseWriter, r *http.Request, e EngineHandler, tracker *Tracker) {
ws, err := upgrader.Upgrade(w, r, nil)
if err != nil {
slog.Error("WS: Failed to upgrade to websocket", "error", err)
@@ -25,13 +25,14 @@ func Handle(w http.ResponseWriter, r *http.Request, e EngineHandler) {
}
defer ws.Close()
err = e.OnConnect(ws)
client := NewClient(ws)
err = e.OnConnect(client.ID)
if err != nil {
slog.Error("WS: Failed to call onConnect handler", "error", err)
return
}
defer e.OnDisconnect(ws)
defer e.OnDisconnect(client.ID)
tracker.Track(client)
for {
_, message, err := ws.ReadMessage()
if err != nil {
@@ -47,7 +48,7 @@ func Handle(w http.ResponseWriter, r *http.Request, e EngineHandler) {
return
}
slog.Debug("WS: Unmarshalled message", "message", msg)
err = e.OnReceiveMessage(msg)
err = e.OnReceiveMessage(client.ID, msg)
if err != nil {
slog.Error("WS: Failed to on receive message", "error", err)
return