fix: Heartbeat message for bybit

Author: metarsit

pkg/connector/bybit/connector.go

@@ -6,6 +6,7 @@ import (
 	"fmt"
 	"log/slog"
 	"strings"
+	"time"
 
 	"github.com/google/uuid"
 	"github.com/gorilla/websocket"
@@ -42,19 +43,23 @@ func NewConnector(ctx context.Context, wsUrl string, pairs []string) (connector.
 
 	req := SubscriptionMessage{
 		Op:    "subscribe",
-		ReqId: uuid.New(),
+		ReqId: uuid.New().String(),
 		Args:  args,
 	}
 
 	if err := ws.WriteJSON(req); err != nil {
 		return nil, err
 	}
 
-	return &Connector{
+	connector := &Connector{
 		ctx:   ctx,
 		pairs: pairs,
 		ws:    ws,
-	}, nil
+	}
+
+	// We send heart beat every 10 seconds
+	connector.sendHeartbeat()
+	return connector, nil
 }
 
 func (c *Connector) Close() error {
@@ -107,3 +112,20 @@ func (c *Connector) Name() string {
 func (c *Connector) Tickers() []string {
 	return c.pairs
 }
+
+func (c *Connector) sendHeartbeat() {
+	req := SubscriptionMessage{
+		Op:    "ping",
+		ReqId: "100001",
+	}
+
+	// We send heart beat every 10 seconds
+	go func() {
+		for {
+			time.Sleep(10 * time.Second)
+			if err := c.ws.WriteJSON(req); err != nil {
+				slog.Warn("error sending heartbeat", "error", err, "exchange", Name)
+			}
+		}
+	}()
+}

pkg/connector/bybit/types.go

@@ -2,14 +2,12 @@ package bybit
 
 import (
 	"math/big"
-
-	"github.com/google/uuid"
 )
 
 type SubscriptionMessage struct {
-	ReqId uuid.UUID `json:"req_id"`
-	Op    string    `json:"op"`
-	Args  []string  `json:"args"`
+	ReqId string   `json:"req_id"`
+	Op    string   `json:"op"`
+	Args  []string `json:"args"`
 }
 
 type TickerResponse struct {