Ver Fonte

Improve IsUnexpectedCloseError examples

Gary Burd há 10 anos atrás
pai
commit
5434f8b69b
2 ficheiros alterados com 4 adições e 1 exclusões
  1. 1 1
      example_test.go
  2. 3 0
      examples/chat/conn.go

+ 1 - 1
example_test.go

@@ -25,7 +25,7 @@ func ExampleIsUnexpectedCloseError(err error, c *websocket.Conn, req *http.Reque
 		messageType, p, err := c.ReadMessage()
 		if err != nil {
 			if websocket.IsUnexpectedCloseError(err, websocket.CloseGoingAway) {
-				log.Println("error: %v, user-agent: %v", err, req.Header.Get("User-Agent"))
+				log.Printf("error: %v, user-agent: %v", err, req.Header.Get("User-Agent"))
 			}
 			return
 		}

+ 3 - 0
examples/chat/conn.go

@@ -51,6 +51,9 @@ func (c *connection) readPump() {
 	for {
 		_, message, err := c.ws.ReadMessage()
 		if err != nil {
+			if websocket.IsUnexpectedCloseError(err, websocket.CloseGoingAway) {
+				log.Printf("error: %v", err)
+			}
 			break
 		}
 		h.broadcast <- message