handle disconnect error

slamdev · May 23, 2020 · 65f7c80 · 65f7c80
1 parent 05a935a
commit 65f7c80
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/pkg/notifier/websocket.go b/pkg/notifier/websocket.go
@@ -4,6 +4,7 @@ import (
 	"github.com/gorilla/websocket"
 	"github.com/sirupsen/logrus"
 	"net/http"
+	"strings"
 )
 
 var upgrader = websocket.Upgrader{}
@@ -20,10 +21,16 @@ func handleWebsocketRequest(w http.ResponseWriter, r *http.Request) {
 		}
 	}()
 
+	logrus.Info("websocket connection is open")
+
 	for {
 		mt, message, err := conn.ReadMessage()
 		if err != nil {
-			logrus.WithError(err).Error("failed to read message")
+			if strings.Contains(err.Error(), "close 1000") {
+				logrus.Info("websocket connection is closed")
+			} else {
+				logrus.WithError(err).Error("failed to read message")
+			}
 			break
 		}