From 7b82725e35b5020f9a57979bb350ce0e21f406ab Mon Sep 17 00:00:00 2001
From: FishFish <fishfish@FishFishdeMacBook-Pro.local>
Date: Fri, 3 Aug 2018 16:50:36 +0800
Subject: [PATCH] websocket close

---
 utils/net/websocket.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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)