Websocket: Fix silly panic when no needed https://github.com/kataras/iris/issues/367

This commit is contained in:
Gerasimos Maropoulos 2016-08-17 23:35:46 +03:00
parent befa43ba2a
commit 8f1bf8dcc7

View File

@ -36,7 +36,10 @@ func RegisterWebsocketServer(station FrameworkAPI, server WebsocketServer, logge
websocketHandler := func(ctx *Context) {
if err := server.Upgrade(ctx); err != nil {
logger.Panic(err)
if ctx.framework.Config.IsDevelopment {
logger.Printf("Websocket error while trying to Upgrade the connection. Trace: %s", err.Error())
}
ctx.EmitError(StatusBadRequest)
}
}
@ -47,7 +50,10 @@ func RegisterWebsocketServer(station FrameworkAPI, server WebsocketServer, logge
}
if err := server.Upgrade(ctx); err != nil {
logger.Panic(err)
if ctx.framework.Config.IsDevelopment {
logger.Printf("Websocket error while trying to Upgrade the connection. Trace: %s", err.Error())
}
ctx.EmitError(StatusBadRequest)
}
}
}