xevion/go-ha
1
0
Fork 0
mirror of https://github.com/Xevion/go-ha.git synced 2025-12-06 07:15:09 -06:00
Code Issues Packages Projects Releases Wiki Activity

Fix panics due to concurrent websocket writes

Browse Source
This commit is contained in:
Lubos Dolezel
2023-02-24 11:49:26 +01:00
parent 8bee96aeff
commit f27cbfb299
21 changed files with 131 additions and 141 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
internal/services/homeassistant.go
View File

@@ -3,12 +3,11 @@ package services
import (
"context"
"github.com/gorilla/websocket"
ws "saml.dev/gome-assistant/internal/websocket"
)
type HomeAssistant struct {
conn *websocket.Conn
conn *ws.WebsocketWriter
ctx context.Context
}
@@ -22,7 +21,7 @@ func (ha *HomeAssistant) TurnOn(entityId string, serviceData ...map[string]any)
req.ServiceData = serviceData[0]
}
ws.WriteMessage(req, ha.conn, ha.ctx)
ha.conn.WriteMessage(req, ha.ctx)
}
// Toggle a Home Assistant entity. Takes an entityId and an optional
@@ -35,7 +34,7 @@ func (ha *HomeAssistant) Toggle(entityId string, serviceData ...map[string]any)
req.ServiceData = serviceData[0]
}
ws.WriteMessage(req, ha.conn, ha.ctx)
ha.conn.WriteMessage(req, ha.ctx)
}
func (ha *HomeAssistant) TurnOff(entityId string) {
@@ -43,5 +42,5 @@ func (ha *HomeAssistant) TurnOff(entityId string) {
req.Domain = "homeassistant"
req.Service = "turn_off"
ws.WriteMessage(req, ha.conn, ha.ctx)
ha.conn.WriteMessage(req, ha.ctx)
}