From c2c566697709998a8aa5383d92758d5b6ca4c906 Mon Sep 17 00:00:00 2001
From: Martin Geno <geno+dev@fireorbit.de>
Date: Fri, 17 Nov 2017 20:38:53 +0100
Subject: [PATCH] [BUGFIX] websocket error handling

---
 websocket/client.go | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/websocket/client.go b/websocket/client.go
index 91a8ce6..a9c2197 100644
--- a/websocket/client.go
+++ b/websocket/client.go
@@ -1,8 +1,6 @@
 package websocket
 
 import (
-	"io"
-
 	"github.com/google/uuid"
 	log "github.com/sirupsen/logrus"
 
@@ -106,10 +104,11 @@ func (c *Client) listenRead() {
 		default:
 			var msg Message
 			err := websocket.ReadJSON(c.ws, &msg)
-			if err == io.EOF {
+			if websocket.IsCloseError(err, websocket.CloseGoingAway) {
 				return
 			} else if err != nil {
-				log.Println(err, c.GetID())
+				log.Infof("error on reading %s: %s", c.GetID(), err)
+				return
 			} else {
 				c.handleInput(&msg)
 			}