jshiffer/go-xmpp
1
0
Fork 1
mirror of https://github.com/FluuxIO/go-xmpp.git synced 2024-11-21 10:02:00 -08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix websocket connect timeout

Browse Source
This commit is contained in:
Wichert Akkerman 2019-10-12 17:50:00 +02:00 committed by Mickaël Rémond
parent 01d78a1e5c
commit 87ff01ac68
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
websocket_transport.go
View File

@ -19,10 +19,13 @@ type WebsocketTransport struct {
func (t *WebsocketTransport) Connect() error { func (t *WebsocketTransport) Connect() error {
t.ctx = context.Background() t.ctx = context.Background()
if t.Config.ConnectTimeout > 0 {
ctx, cancel := context.WithTimeout(t.ctx, time.Duration(t.Config.ConnectTimeout)*time.Second) ctx, cancel := context.WithTimeout(t.ctx, time.Duration(t.Config.ConnectTimeout)*time.Second)
t.ctx = ctx
defer cancel() defer cancel()
}
wsConn, _, err := websocket.Dial(ctx, t.Config.Address, nil) wsConn, _, err := websocket.Dial(t.ctx, t.Config.Address, nil)
if err != nil { if err != nil {
return NewConnError(err, true) return NewConnError(err, true)
} }