Fix error result from Transport.Connect errors

2019-10-12 17:46:53 +02:00
parent 8fb3e33a1f
commit 01d78a1e5c
2 changed files with 9 additions and 5 deletions
@@ -23,11 +23,12 @@ func (t *WebsocketTransport) Connect() error {
 	defer cancel()

 	wsConn, _, err := websocket.Dial(ctx, t.Config.Address, nil)
-	if err == nil {
-		t.wsConn = wsConn
-		t.netConn = websocket.NetConn(t.ctx, t.wsConn, websocket.MessageText)
+	if err != nil {
+		return NewConnError(err, true)
 	}
-	return err
+	t.wsConn = wsConn
+	t.netConn = websocket.NetConn(t.ctx, t.wsConn, websocket.MessageText)
+	return nil
 }

 func (t WebsocketTransport) StartTLS(domain string) error {