improve maxapi websocket reconnect issue

pkg/exchange/max/maxapi/websocket.go

@@ -3,6 +3,7 @@ package max
 import (
 	"context"
 	"fmt"
+	"net"
 	"sync"
 	"time"
 
@@ -160,20 +161,38 @@ func (s *WebSocketService) read(ctx context.Context) {
 
 		default:
 			s.mu.Lock()
+			if err := s.conn.SetReadDeadline(time.Now().Add(time.Second * 5)); err != nil {
+				log.WithError(err).Error("can not set read deadline")
+			}
+
 			mt, msg, err := s.conn.ReadMessage()
 			s.mu.Unlock()
 
 			if err != nil {
-				if websocket.IsUnexpectedCloseError(err, websocket.CloseNormalClosure) {
+				// if it's a network timeout error, we should re-connect
-					s.EmitDisconnect()
+				switch err := err.(type) {
-					// emit reconnect to start a new connection
+
-					s.emitReconnect()
+				// if it's a websocket related error
+				case *websocket.CloseError:
+					if err.Code == websocket.CloseNormalClosure {
 						return
 					}
+					// for unexpected close error, we should re-connect
+					// emit reconnect to start a new connection
+					s.EmitDisconnect()
+					s.emitReconnect()
+					return
 
-				log.WithError(err).Error("websocket error")
+				case net.Error:
+					s.EmitDisconnect()
+					s.emitReconnect()
+					return
+
+				default:
+					log.WithError(err).Error("unexpected websocket connection error")
 					continue
 				}
+			}
 
 			if mt != websocket.TextMessage {
 				continue