5
0
mirror of https://github.com/cwinfo/yggdrasil-go.git synced 2024-12-23 04:05:39 +00:00

slight cleanup of dial's timeout

This commit is contained in:
Arceliar 2019-10-21 20:52:16 -05:00
parent 681c8ca6f9
commit ea085663ea

View File

@ -72,19 +72,16 @@ func (d *Dialer) DialByNodeIDandMask(ctx context.Context, nodeID, nodeMask *cryp
return nil, err
}
conn.session.setConn(nil, conn)
var c context.Context
var cancel context.CancelFunc
const timeout = 6 * time.Second
if ctx != nil {
c, cancel = context.WithTimeout(ctx, timeout)
} else {
c, cancel = context.WithTimeout(context.Background(), timeout)
if ctx == nil {
ctx = context.Background()
}
ctx, cancel = context.WithTimeout(ctx, 6*time.Second)
defer cancel()
select {
case <-conn.session.init:
return conn, nil
case <-c.Done():
case <-ctx.Done():
conn.Close()
return nil, errors.New("session handshake timeout")
}