1
0
mirror of https://github.com/astaxie/beego.git synced 2024-06-26 18:14:12 +00:00

fix example/chat i/o timeout

This commit is contained in:
skycrab 2015-07-06 21:12:03 +08:00
parent 079993b9f7
commit 8c0ad5ef88

View File

@ -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 {