try close disconnected ws #1

websocketserver/client.go

@@ -63,7 +63,16 @@ func (c *Client) listen() {
 	for {
 		select {
 		case msg := <-c.ch:
-			websocket.JSON.Send(c.ws, msg)
+			err := websocket.JSON.Send(c.ws, msg)
+			if err != nil {
+				c.doneCh <- true
+			}
+		case gone := <-c.doneCh:
+			if gone {
+				c.server.Del(c)
+				err := fmt.Errorf("Client %d is disconnected.", c.id)
+				c.server.Err(err)
+			}
 		}
 	}
 }

websocketserver/server.go

@@ -1,6 +1,7 @@
 package websocketserver
 
 import (
+	"fmt"
 	"log"
 	"net/http"
 
@@ -86,6 +87,11 @@ func (s *Server) Listen() {
 
 		client := NewClient(ws, s)
 		s.Add(client)
+		defer func() {
+			s.Del(client)
+			err := fmt.Errorf("Client %d is disconnected.", client.id)
+			s.Err(err)
+		}()
 		client.Listen()
 	}
 	http.Handle(s.pattern, websocket.Handler(onConnected))