init code

internal/data/market/future_price.go

@@ -0,0 +1,50 @@
+package market
+
+import (
+	"github.com/adshao/go-binance/v2/futures"
+	"github.com/rs/zerolog/log"
+	"strconv"
+)
+
+func (ms MarketData) GetFuturePrice(symbol string) (float64, float64, int64, bool) {
+	p, ok := ms.futureMarkPrice[symbol]
+	if !ok {
+		return 0, 0, 0, false
+	}
+	return p, ms.futureFundingRate[symbol], ms.futureNextFundingTime[symbol], true
+}
+
+func (ms MarketData) StartFutureWsMarkPrice() error {
+	_, _, err := futures.WsAllMarkPriceServe(ms.futureWsMarkPriceHandler, ms.futureWsErrHandler)
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+func (ms MarketData) futureWsMarkPriceHandler(event futures.WsAllMarkPriceEvent) {
+	for _, priceEvent := range event {
+		price, err := strconv.ParseFloat(priceEvent.MarkPrice, 64)
+		if err != nil {
+			continue
+		}
+
+		fundingRate, err := strconv.ParseFloat(priceEvent.FundingRate, 64)
+		if err != nil {
+			continue
+		}
+
+		ms.futureMarkPrice[priceEvent.Symbol] = price
+		ms.futureFundingRate[priceEvent.Symbol] = fundingRate
+		ms.futureNextFundingTime[priceEvent.Symbol] = priceEvent.NextFundingTime
+	}
+}
+
+func (ms MarketData) futureWsErrHandler(err error) {
+	log.Debug().Err(err).Msg("Ws Error. Restart socket")
+	_ = ms.StartFutureWsMarkPrice()
+}
+
+func (ms MarketData) GetAllFundRate() (map[string]float64, map[string]int64) {
+	return ms.futureFundingRate, ms.futureNextFundingTime
+}