lbsadmin/Beego
1
0
Fork 0
mirror of https://github.com/astaxie/beego.git synced 2024-11-22 14:50:55 +00:00
Code Issues Releases Activity

fix example/chat i/o timeout

Browse Source
This commit is contained in:
skycrab 2015-07-06 21:12:03 +08:00
parent 079993b9f7
commit 8c0ad5ef88
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
example/chat/controllers/ws.go
View File

@ -54,14 +54,13 @@ func (c *connection) readPump() {
}() }()
c.ws.SetReadLimit(maxMessageSize) c.ws.SetReadLimit(maxMessageSize)
c.ws.SetReadDeadline(time.Now().Add(readWait)) c.ws.SetReadDeadline(time.Now().Add(readWait))
c.ws.SetPongHandler(func(string) error { c.ws.SetReadDeadline(time.Now().Add(readWait)); return nil })
for { for {
op, r, err := c.ws.NextReader() op, r, err := c.ws.NextReader()
if err != nil { if err != nil {
break break
} }
switch op { switch op {
case websocket.PongMessage:
c.ws.SetReadDeadline(time.Now().Add(readWait))
case websocket.TextMessage: case websocket.TextMessage:
message, err := ioutil.ReadAll(r) message, err := ioutil.ReadAll(r)
if err != nil { if err != nil {