Fix error result from Transport.Connect errors

This commit is contained in:
Wichert Akkerman 2019-10-12 17:46:53 +02:00 committed by Mickaël Rémond
parent 8fb3e33a1f
commit 01d78a1e5c
2 changed files with 9 additions and 5 deletions

View file

@ -23,11 +23,12 @@ func (t *WebsocketTransport) Connect() error {
defer cancel()
wsConn, _, err := websocket.Dial(ctx, t.Config.Address, nil)
if err == nil {
if err != nil {
return NewConnError(err, true)
}
t.wsConn = wsConn
t.netConn = websocket.NetConn(t.ctx, t.wsConn, websocket.MessageText)
}
return err
return nil
}
func (t WebsocketTransport) StartTLS(domain string) error {

View file

@ -19,7 +19,10 @@ func (t *XMPPTransport) Connect() error {
var err error
t.conn, err = net.DialTimeout("tcp", t.Config.Address, time.Duration(t.Config.ConnectTimeout)*time.Second)
return err
if err != nil {
return NewConnError(err, true)
}
return nil
}
func (t XMPPTransport) DoesStartTLS() bool {