mirror of
https://github.com/wisplite/tether.git
synced 2026-05-01 06:22:41 -05:00
send incoming websocket data to engine
This commit is contained in:
@@ -36,5 +36,10 @@ func (e *Engine) CreateTable(name string, schema interface{}) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (e *Engine) Handle(w http.ResponseWriter, r *http.Request) {
|
func (e *Engine) Handle(w http.ResponseWriter, r *http.Request) {
|
||||||
reactivity.Handle(w, r)
|
reactivity.Handle(w, r, e)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (e *Engine) OnReceiveMessage(msg map[string]interface{}) error {
|
||||||
|
slog.Debug("Received message", "message", msg)
|
||||||
|
return nil
|
||||||
}
|
}
|
||||||
|
|||||||
+23
-4
@@ -1,18 +1,24 @@
|
|||||||
package reactivity
|
package reactivity
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"encoding/json"
|
||||||
"log/slog"
|
"log/slog"
|
||||||
"net/http"
|
"net/http"
|
||||||
|
|
||||||
"github.com/gorilla/websocket"
|
"github.com/gorilla/websocket"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
// MessageReceiver receives decoded WebSocket payloads. Implemented by tether.Engine.
|
||||||
|
type MessageReceiver interface {
|
||||||
|
OnReceiveMessage(msg map[string]interface{}) error
|
||||||
|
}
|
||||||
|
|
||||||
var upgrader = websocket.Upgrader{}
|
var upgrader = websocket.Upgrader{}
|
||||||
|
|
||||||
func Handle(w http.ResponseWriter, r *http.Request) {
|
func Handle(w http.ResponseWriter, r *http.Request, e MessageReceiver) {
|
||||||
ws, err := upgrader.Upgrade(w, r, nil)
|
ws, err := upgrader.Upgrade(w, r, nil)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
slog.Error("Failed to upgrade to websocket", "error", err)
|
slog.Error("WS: Failed to upgrade to websocket", "error", err)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
defer ws.Close()
|
defer ws.Close()
|
||||||
@@ -20,9 +26,22 @@ func Handle(w http.ResponseWriter, r *http.Request) {
|
|||||||
for {
|
for {
|
||||||
_, message, err := ws.ReadMessage()
|
_, message, err := ws.ReadMessage()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
slog.Error("Failed to read message", "error", err)
|
slog.Error("WS: Failed to read message", "error", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
slog.Debug("WS: Received message", "message", string(message))
|
||||||
|
|
||||||
|
var msg map[string]interface{}
|
||||||
|
err = json.Unmarshal(message, &msg)
|
||||||
|
if err != nil {
|
||||||
|
slog.Error("WS: Failed to unmarshal message", "error", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
slog.Debug("WS: Unmarshalled message", "message", msg)
|
||||||
|
err = e.OnReceiveMessage(msg)
|
||||||
|
if err != nil {
|
||||||
|
slog.Error("WS: Failed to on receive message", "error", err)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
slog.Debug("Received message", "message", string(message))
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user