Commit 8c0ad5ef authored by skycrab's avatar skycrab

fix example/chat i/o timeout

parent 079993b9
...@@ -54,14 +54,13 @@ func (c *connection) readPump() { ...@@ -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 {
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment