Fixing error where connection was re-established incorrectly

2023-09-25 15:25:43 -05:00
parent 896bafe70b
commit e41fdd3561
2 changed files with 12 additions and 0 deletions
--- a/transport/websocket/websocket.go
+++ b/transport/websocket/websocket.go
@ -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