From e1ef49c46a472f72c52c8b32da896b6331218328 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Miroslav=20=C5=A0ediv=C3=BD?= Date: Wed, 28 Oct 2020 22:28:10 +0100 Subject: [PATCH] ws upgrade fail error message. --- internal/http/http.go | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/internal/http/http.go b/internal/http/http.go index 735bf17a..daf88ae8 100644 --- a/internal/http/http.go +++ b/internal/http/http.go @@ -33,7 +33,9 @@ func New(conf *config.Server, webSocketHandler types.WebSocketHandler) *Server { router.Use(customMiddleware.Logger) // Log API request calls using custom logger function router.Get("/ws", func(w http.ResponseWriter, r *http.Request) { - webSocketHandler.Upgrade(w, r) + if webSocketHandler.Upgrade(w, r) != nil { + w.Write([]byte("unable to upgrade your connection to a websocket")) + } }) fs := http.FileServer(http.Dir(conf.Static))