watsonb8/fayec
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fixing error where connection was re-established incorrectly

Browse Source
This commit is contained in:
Brandon Watson 2023-09-25 15:25:43 -05:00
parent 896bafe70b
commit 12bd5e6a01
2 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
internal/dispatcher/dispatcher.go
View File

@ -113,7 +113,15 @@ func (d *Dispatcher) dispatchMessage(msg *message.Message) {
return
}
log.Println("Websocket terminated: reconnecting")
err := d.Disconnect()
if err != nil {
log.Println(err)
}
d.Connect()
if err != nil {
log.Println(err)
}
subsList := d.store.GetAll()
for i := range subsList {
sub := subsList[i]

4
transport/websocket/websocket.go
View File

@ -46,6 +46,10 @@ func (w *Websocket) Init(endpoint string, options *transport.Options) error {
w.stopCh = make(chan error)
w.conn, _, err = websocket.DefaultDialer.Dial(endpoint, options.Headers)
if err != nil {
return err
}
err = w.conn.UnderlyingConn().(*tls.Conn).NetConn().(*net.TCPConn).SetKeepAlive(true)
if err != nil {
return err