Fix websocket: close 1006 (abnormal closure): unexpected EOF errors when calling binance.WsCombinedBookTickerServe()

v2/futures/websocket.go

@@ -87,19 +87,26 @@ func keepAlive(c *websocket.Conn, timeout time.Duration) {
 	ticker := time.NewTicker(timeout)
 
 	lastResponse := time.Now()
-	c.SetPongHandler(func(msg string) error {
+
+	c.SetPingHandler(func(pingData string) error {
+		// Respond with Pong using the server's PING payload
+		err := c.WriteControl(
+			websocket.PongMessage,
+			[]byte(pingData),
+			time.Now().Add(time.Second*10), // Short deadline to ensure timely response
+		)
+		if err != nil {
+			return err
+		}
+
 		lastResponse = time.Now()
+
 		return nil
 	})
 
 	go func() {
 		defer ticker.Stop()
 		for {
-			deadline := time.Now().Add(10 * time.Second)
-			err := c.WriteControl(websocket.PingMessage, []byte{}, deadline)
-			if err != nil {
-				return
-			}
 			<-ticker.C
 			if time.Since(lastResponse) > timeout {
 				c.Close()

v2/futures/websocket_service.go

@@ -25,7 +25,7 @@ var (
 	// WebsocketTimeout is an interval for sending ping/pong messages if WebsocketKeepalive is enabled
 	WebsocketTimeout = time.Second * 60
 	// WebsocketKeepalive enables sending ping/pong messages to check the connection stability
-	WebsocketKeepalive = false
+	WebsocketKeepalive = true
 	// UseTestnet switch all the WS streams from production to the testnet
 	UseTestnet = false
 	// WebsocketTimeoutReadWriteConnection is an interval for sending ping/pong messages if WebsocketKeepalive is enabled

v2/options/websocket.go

@@ -88,19 +88,26 @@ func keepAlive(c *websocket.Conn, timeout time.Duration) {
 	ticker := time.NewTicker(timeout)
 
 	lastResponse := time.Now()
-	c.SetPongHandler(func(msg string) error {
+
+	c.SetPingHandler(func(pingData string) error {
+		// Respond with Pong using the server's PING payload
+		err := c.WriteControl(
+			websocket.PongMessage,
+			[]byte(pingData),
+			time.Now().Add(time.Second*10), // Short deadline to ensure timely response
+		)
+		if err != nil {
+			return err
+		}
+
 		lastResponse = time.Now()
+
 		return nil
 	})
 
 	go func() {
 		defer ticker.Stop()
 		for {
-			deadline := time.Now().Add(10 * time.Second)
-			err := c.WriteControl(websocket.PingMessage, []byte{}, deadline)
-			if err != nil {
-				return
-			}
 			<-ticker.C
 			if time.Since(lastResponse) > timeout {
 				c.Close()

v2/options/websocket_service.go

@@ -20,7 +20,7 @@ var (
 	// WebsocketTimeout is an interval for sending ping/pong messages if WebsocketKeepalive is enabled
 	WebsocketTimeout = time.Second * 60
 	// WebsocketKeepalive enables sending ping/pong messages to check the connection stability
-	WebsocketKeepalive = false
+	WebsocketKeepalive = true
 	// UseTestnet switch all the WS streams from production to the testnet
 	UseTestnet = false
 

v2/websocket.go

@@ -53,10 +53,14 @@ var wsServe = func(cfg *WsConfig, handler WsHandler, errHandler ErrHandler) (don
 		// This function will exit either on error from
 		// websocket.Conn.ReadMessage or when the stopC channel is
 		// closed by the client.
+
 		defer close(doneC)
 		if WebsocketKeepalive {
+			// This function overwrites the default ping frame handler
+			// sent by the websocket API server
 			keepAlive(c, WebsocketTimeout)
 		}
+
 		// Wait for the stopC channel to be closed.  We do that in a
 		// separate goroutine because ReadMessage is a blocking
 		// operation.
@@ -87,19 +91,26 @@ func keepAlive(c *websocket.Conn, timeout time.Duration) {
 	ticker := time.NewTicker(timeout)
 
 	lastResponse := time.Now()
-	c.SetPongHandler(func(msg string) error {
+
+	c.SetPingHandler(func(pingData string) error {
+		// Respond with Pong using the server's PING payload
+		err := c.WriteControl(
+			websocket.PongMessage,
+			[]byte(pingData),
+			time.Now().Add(time.Second*10), // Short deadline to ensure timely response
+		)
+		if err != nil {
+			return err
+		}
+
 		lastResponse = time.Now()
+
 		return nil
 	})
 
 	go func() {
 		defer ticker.Stop()
 		for {
-			deadline := time.Now().Add(10 * time.Second)
-			err := c.WriteControl(websocket.PingMessage, []byte{}, deadline)
-			if err != nil {
-				return
-			}
 			<-ticker.C
 			if time.Since(lastResponse) > timeout {
 				c.Close()

v2/websocket_service.go

@@ -22,7 +22,7 @@ var (
 	WebsocketTimeout = time.Second * 60
 
 	// WebsocketKeepalive enables sending ping/pong messages to check the connection stability
-	WebsocketKeepalive = false
+	WebsocketKeepalive = true
 	// WebsocketTimeoutReadWriteConnection is an interval for sending ping/pong messages if WebsocketKeepalive is enabled
 	// using for websocket API (read/write)
 	WebsocketTimeoutReadWriteConnection = time.Second * 10