From 8c0ad5ef8834c69920a8254403877826d1959c3f Mon Sep 17 00:00:00 2001
From: skycrab <skycrab68@163.com>
Date: Mon, 6 Jul 2015 21:12:03 +0800
Subject: [PATCH] fix example/chat i/o timeout

---
 example/chat/controllers/ws.go | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/example/chat/controllers/ws.go b/example/chat/controllers/ws.go
index fc3917b3..9e405513 100644
--- a/example/chat/controllers/ws.go
+++ b/example/chat/controllers/ws.go
@@ -54,14 +54,13 @@ func (c *connection) readPump() {
 	}()
 	c.ws.SetReadLimit(maxMessageSize)
 	c.ws.SetReadDeadline(time.Now().Add(readWait))
+	c.ws.SetPongHandler(func(string) error { c.ws.SetReadDeadline(time.Now().Add(readWait)); return nil })
 	for {
 		op, r, err := c.ws.NextReader()
 		if err != nil {
 			break
 		}
 		switch op {
-		case websocket.PongMessage:
-			c.ws.SetReadDeadline(time.Now().Add(readWait))
 		case websocket.TextMessage:
 			message, err := ioutil.ReadAll(r)
 			if err != nil {