diff --git a/utils/net/websocket.go b/utils/net/websocket.go index de58570f..7a669843 100644 --- a/utils/net/websocket.go +++ b/utils/net/websocket.go @@ -75,7 +75,7 @@ func NewWebScoketConn(conn net.Conn) (c *WebsocketConn) { } func (p *WebsocketConn) Close() error { - if atomic.LoadInt32(&p.closed) == 1 { + if atomic.SwapInt32(&p.closed, 1) == 1 { return nil } close(p.wait)