go-packages/iris
1
0
Fork 0
mirror of https://github.com/kataras/iris.git synced 2025-01-23 10:41:03 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Try third to solve https://github.com/kataras/iris/issues/175#issuecomment-226634061

Browse Source
This commit is contained in:
Makis Maropoulos 2016-06-17 01:44:16 +03:00
parent 0d482508fc
commit 2942d384c4
2 changed files with 18 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
http.go
View File

@ -1157,8 +1157,8 @@ func (mux *serveMux) fireError(statusCode int, ctx *Context) {
if errHandler == nil {
errHandler = HandlerFunc(func(ctx *Context) {
ctx.ResetBody()
ctx.SetBodyString(statusText[statusCode])
ctx.SetStatusCode(statusCode)
ctx.SetBodyString(statusText[statusCode])
})
mux.errorHandlers[statusCode] = errHandler
}

18
websocket/connection.go
View File

@ -98,9 +98,25 @@ func (c *connection) writer() {
return
}
if err := c.write(websocket.TextMessage, msg); err != nil {
c.underline.SetWriteDeadline(time.Now().Add(c.server.config.WriteTimeout))
res, err := c.underline.NextWriter(websocket.TextMessage)
if err != nil {
return
}
res.Write(msg)
n := len(c.send)
for i := 0; i < n; i++ {
res.Write(<-c.send)
}
if err := res.Close(); err != nil {
return
}
// if err := c.write(websocket.TextMessage, msg); err != nil {
// return
// }
case <-ticker.C:
if err := c.write(websocket.PingMessage, []byte{}); err != nil {