Fix: tzkt event close error handling

Author: aopoltorzhicky

tzkt/events/signalr/errors.go

@@ -3,13 +3,13 @@ package signalr
 import "errors"
 
 var (
-	ErrUnknownMessageType = errors.New("Unknown message type")
-	ErrMessageParsing     = errors.New("Can't parse message")
-	ErrEmptyResponse      = errors.New("Empty response from server")
-	ErrHandshake          = errors.New("Handshake error")
-	ErrInvalidStatusCode  = errors.New("Invalid status code")
-	ErrNegotiate          = errors.New("Negotiate error")
-	ErrInvalidScheme      = errors.New("Invalid URL scheme. Expected https or http. Got")
-	ErrConnectionClose    = errors.New("Connection is closed")
-	ErrTimeout            = errors.New("Connection timeout")
+	ErrUnknownMessageType = errors.New("unknown message type")
+	ErrMessageParsing     = errors.New("can't parse message")
+	ErrEmptyResponse      = errors.New("empty response from server")
+	ErrHandshake          = errors.New("handshake error")
+	ErrInvalidStatusCode  = errors.New("invalid status code")
+	ErrNegotiate          = errors.New("negotiate error")
+	ErrInvalidScheme      = errors.New("invalid URL scheme. Expected https or http. Got")
+	ErrConnectionClose    = errors.New("connection is closed")
+	ErrTimeout            = errors.New("connection timeout")
 )

tzkt/events/signalr/hub.go

@@ -138,7 +138,7 @@ func (hub *Hub) listen() {
 			default:
 				if err := hub.readAllMessages(); err != nil {
 					switch {
-					case errors.Is(err, ErrTimeout):
+					case errors.Is(err, ErrTimeout) || errors.Is(err, &websocket.CloseError{}):
 						if err := hub.reconnect(); err != nil {
 							log.Errorf("reconnect: %s", err.Error())
 							log.Warn("retry after 5 seconds")
@@ -247,6 +247,7 @@ func (hub *Hub) getScanner() (*JSONReader, error) {
 		if e, ok := err.(net.Error); ok && e.Timeout() {
 			return nil, ErrTimeout
 		}
+
 		return nil, errors.Wrap(err, "NextReader")
 	}
 	return NewJSONReader(r), nil