64 lines
1.3 KiB
Go
64 lines
1.3 KiB
Go
package main
|
|
|
|
import (
|
|
"log"
|
|
"time"
|
|
|
|
"github.com/gorilla/websocket"
|
|
)
|
|
|
|
type WSConnection struct {
|
|
url string
|
|
conn *websocket.Conn
|
|
errChan chan error
|
|
ReadChan chan string
|
|
}
|
|
|
|
func (ws *WSConnection) readMessage() {
|
|
log.Printf("Reading messages")
|
|
for {
|
|
_, message, err := ws.conn.ReadMessage()
|
|
if err != nil {
|
|
ws.errChan <- err
|
|
return
|
|
}
|
|
log.Printf("Received: %s", message)
|
|
ws.ReadChan <- string(message)
|
|
}
|
|
}
|
|
|
|
func (ws *WSConnection) writeMessage(message string) {
|
|
err := ws.conn.WriteMessage(websocket.TextMessage, []byte(message))
|
|
if err != nil {
|
|
log.Printf("Error during message writing: %v", err)
|
|
ws.errChan <- err
|
|
return
|
|
}
|
|
}
|
|
|
|
func (ws *WSConnection) handleError() {
|
|
for {
|
|
err := <-ws.errChan
|
|
log.Println("Error during message reading:", err)
|
|
|
|
time.Sleep(5 * time.Second)
|
|
ws.Open()
|
|
}
|
|
}
|
|
|
|
func (ws *WSConnection) Open() {
|
|
log.Printf("Connecting to %s", ws.url)
|
|
conn, _, err := websocket.DefaultDialer.Dial(ws.url, nil)
|
|
if err != nil {
|
|
log.Println("Error during connection:", err)
|
|
ws.errChan <- err
|
|
return
|
|
}
|
|
log.Printf("Connected")
|
|
ws.conn = conn
|
|
ws.errChan = make(chan error)
|
|
ws.ReadChan = make(chan string, 1024)
|
|
go ws.readMessage()
|
|
go ws.handleError()
|
|
}
|