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) { websocketHandler := func(ctx *Context) {
if err := server.Upgrade(ctx); err != nil { 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 { 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)
} }
} }
} }