From 2ca4d945bbe24b62ec906198e0e9c9e9f6dd32bf Mon Sep 17 00:00:00 2001
From: qydysky <qydysky@foxmail.com>
Date: Tue, 15 Sep 2020 07:09:03 +0800
Subject: [PATCH] 9

---
 ws.go | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/ws.go b/ws.go
index 511d8d9..0b66396 100644
--- a/ws.go
+++ b/ws.go
@@ -69,8 +69,12 @@ func (i *ws) Handle() (o *ws) {
 			for {
 				_, message, err := c.ReadMessage()
 				if err != nil {
-					if !websocket.IsCloseError(err, websocket.CloseNormalClosure) {
-						l.E(err)
+					if e, ok := err.(*websocket.CloseError); ok {
+						switch e.Code {
+						case websocket.CloseNormalClosure:l.E("服务器关闭连接")
+						case websocket.CloseAbnormalClosure:l.E("服务器意外关闭连接")
+						default:l.E(err);
+						}
 					}
 					return
 				}
-- 
2.39.2