mirror of https://github.com/gorilla/websocket.git
perf: reduce timer in write_control
Signed-off-by: rfyiamcool <rfyiamcool@163.com>
This commit is contained in:
parent
85fb2d8136
commit
f01629e7ea
13
conn.go
13
conn.go
|
@ -446,13 +446,18 @@ func (c *Conn) WriteControl(messageType int, data []byte, deadline time.Time) er
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
timer := time.NewTimer(d)
|
|
||||||
select {
|
select {
|
||||||
case <-c.mu:
|
case <-c.mu:
|
||||||
timer.Stop()
|
default:
|
||||||
case <-timer.C:
|
timer := time.NewTimer(d)
|
||||||
return errWriteTimeout
|
select {
|
||||||
|
case <-c.mu:
|
||||||
|
timer.Stop()
|
||||||
|
case <-timer.C:
|
||||||
|
return errWriteTimeout
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
defer func() { c.mu <- struct{}{} }()
|
defer func() { c.mu <- struct{}{} }()
|
||||||
|
|
||||||
c.writeErrMu.Lock()
|
c.writeErrMu.Lock()
|
||||||
|
|
Loading…
Reference in New Issue