two things:

fix bug with newer HA versions by sending int id instead of string
add warning to logs when receiving unsuccessful websocket response
This commit is contained in:
Sam Lewis
2023-06-11 14:09:13 -04:00
parent d16f646423
commit eb26a44459
2 changed files with 21 additions and 12 deletions

View File

@@ -3,20 +3,23 @@ package websocket
import (
"context"
"encoding/json"
"fmt"
"log"
"github.com/gorilla/websocket"
)
type BaseMessage struct {
Type string `json:"type"`
Id int64 `json:"id"`
Type string `json:"type"`
Id int64 `json:"id"`
Success bool `json:"success"`
}
type ChanMsg struct {
Id int64
Type string
Raw []byte
Id int64
Type string
Success bool
Raw []byte
}
func ListenWebsocket(conn *websocket.Conn, ctx context.Context, c chan ChanMsg) {
@@ -29,12 +32,19 @@ func ListenWebsocket(conn *websocket.Conn, ctx context.Context, c chan ChanMsg)
break
}
base := BaseMessage{}
base := BaseMessage{
// default to true for messages that don't include "success" at all
Success: true,
}
json.Unmarshal(bytes, &base)
if !base.Success {
fmt.Println("WARNING: received unsuccessful response:", string(bytes))
}
chanMsg := ChanMsg{
Type: base.Type,
Id: base.Id,
Raw: bytes,
Type: base.Type,
Id: base.Id,
Success: base.Success,
Raw: bytes,
}
c <- chanMsg