9

diff --git a/ws.go b/ws.go
index 511d8d9d1e2c716d009fa784c17071b001185cad..0b66396065f02deaf1ac07df8d205527d6f921a5 100644 (file)
--- 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
                                }