mirror of
https://github.com/Xevion/go-ha.git
synced 2025-12-06 23:15:09 -06:00
43 lines
635 B
Go
43 lines
635 B
Go
package websocket
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"log"
|
|
|
|
"github.com/gorilla/websocket"
|
|
)
|
|
|
|
type BaseMessage struct {
|
|
Type string `json:"type"`
|
|
Id int64 `json:"id"`
|
|
}
|
|
|
|
type ChanMsg struct {
|
|
Id int64
|
|
Type string
|
|
Raw []byte
|
|
}
|
|
|
|
func ListenWebsocket(conn *websocket.Conn, ctx context.Context, c chan ChanMsg) {
|
|
for {
|
|
bytes, err := ReadMessage(conn, ctx)
|
|
|
|
if err != nil {
|
|
log.Default().Println("Error reading from websocket:", err)
|
|
close(c)
|
|
break
|
|
}
|
|
|
|
base := BaseMessage{}
|
|
json.Unmarshal(bytes, &base)
|
|
chanMsg := ChanMsg{
|
|
Type: base.Type,
|
|
Id: base.Id,
|
|
Raw: bytes,
|
|
}
|
|
|
|
c <- chanMsg
|
|
}
|
|
}
|