Files
gocryptotrader/exchanges/deribit/deribit_websocket.go
Copilot a9f23018e3 Deribit: Remove deprecated RFQ endpoints (#2078)
* Initial plan

* Remove deprecated Deribit RFQ endpoints

Co-authored-by: thrasher- <4685270+thrasher-@users.noreply.github.com>

* Remove extra newline before TestGetTradeVolumes

Co-authored-by: thrasher- <4685270+thrasher-@users.noreply.github.com>

* Remove extra newline before TestSetMMPConfig

Co-authored-by: thrasher- <4685270+thrasher-@users.noreply.github.com>

---------

Co-authored-by: copilot-swe-agent[bot] <198982749+Copilot@users.noreply.github.com>
Co-authored-by: thrasher- <4685270+thrasher-@users.noreply.github.com>
2025-10-14 09:08:45 +11:00

956 lines
30 KiB
Go

package deribit
import (
"context"
"encoding/hex"
"errors"
"fmt"
"net/http"
"strconv"
"strings"
"text/template"
"time"
gws "github.com/gorilla/websocket"
"github.com/thrasher-corp/gocryptotrader/common"
"github.com/thrasher-corp/gocryptotrader/common/crypto"
"github.com/thrasher-corp/gocryptotrader/currency"
"github.com/thrasher-corp/gocryptotrader/encoding/json"
"github.com/thrasher-corp/gocryptotrader/exchange/websocket"
"github.com/thrasher-corp/gocryptotrader/exchanges/asset"
"github.com/thrasher-corp/gocryptotrader/exchanges/kline"
"github.com/thrasher-corp/gocryptotrader/exchanges/nonce"
"github.com/thrasher-corp/gocryptotrader/exchanges/order"
"github.com/thrasher-corp/gocryptotrader/exchanges/orderbook"
"github.com/thrasher-corp/gocryptotrader/exchanges/request"
"github.com/thrasher-corp/gocryptotrader/exchanges/subscription"
"github.com/thrasher-corp/gocryptotrader/exchanges/ticker"
"github.com/thrasher-corp/gocryptotrader/exchanges/trade"
"github.com/thrasher-corp/gocryptotrader/log"
)
var deribitWebsocketAddress = "wss://www.deribit.com/ws" + deribitAPIVersion
const (
rpcVersion = "2.0"
rateLimit = 20
errAuthFailed = 1002
// public websocket channels
announcementsChannel = "announcements"
orderbookChannel = "book"
chartTradesChannel = "chart.trades"
priceIndexChannel = "deribit_price_index"
priceRankingChannel = "deribit_price_ranking"
priceStatisticsChannel = "deribit_price_statistics"
volatilityIndexChannel = "deribit_volatility_index"
estimatedExpirationPriceChannel = "estimated_expiration_price"
incrementalTickerChannel = "incremental_ticker"
instrumentStateChannel = "instrument.state"
markPriceOptionsChannel = "markprice.options"
perpetualChannel = "perpetual."
platformStateChannel = "platform_state"
platformStatePublicMethodsStateChannel = "platform_state.public_methods_state"
quoteChannel = "quote"
requestForQuoteChannel = "rfq"
tickerChannel = "ticker"
tradesChannel = "trades"
// private websocket channels
userAccessLogChannel = "user.access_log"
userChangesInstrumentsChannel = "user.changes."
userChangesCurrencyChannel = "user.changes"
userLockChannel = "user.lock"
userMMPTriggerChannel = "user.mmp_trigger"
userOrdersChannel = "user.orders"
userTradesChannel = "user.trades"
userPortfolioChannel = "user.portfolio"
)
var subscriptionNames = map[string]string{
subscription.TickerChannel: tickerChannel,
subscription.OrderbookChannel: orderbookChannel,
subscription.CandlesChannel: chartTradesChannel,
subscription.AllTradesChannel: tradesChannel,
subscription.MyTradesChannel: userTradesChannel,
subscription.MyOrdersChannel: userOrdersChannel,
announcementsChannel: announcementsChannel,
priceIndexChannel: priceIndexChannel,
priceRankingChannel: priceRankingChannel,
priceStatisticsChannel: priceStatisticsChannel,
volatilityIndexChannel: volatilityIndexChannel,
estimatedExpirationPriceChannel: estimatedExpirationPriceChannel,
incrementalTickerChannel: incrementalTickerChannel,
instrumentStateChannel: instrumentStateChannel,
markPriceOptionsChannel: markPriceOptionsChannel,
perpetualChannel: perpetualChannel,
platformStateChannel: platformStateChannel,
platformStatePublicMethodsStateChannel: platformStatePublicMethodsStateChannel,
quoteChannel: quoteChannel,
requestForQuoteChannel: requestForQuoteChannel,
userAccessLogChannel: userAccessLogChannel,
userChangesInstrumentsChannel: userChangesInstrumentsChannel,
userChangesCurrencyChannel: userChangesCurrencyChannel,
userLockChannel: userLockChannel,
userMMPTriggerChannel: userMMPTriggerChannel,
userPortfolioChannel: userPortfolioChannel,
}
var defaultSubscriptions = subscription.List{
{Enabled: true, Asset: asset.All, Channel: subscription.CandlesChannel, Interval: kline.OneDay},
{Enabled: true, Asset: asset.All, Channel: subscription.OrderbookChannel, Interval: kline.HundredMilliseconds}, // Raw is available for authenticated users
{Enabled: true, Asset: asset.All, Channel: subscription.TickerChannel, Interval: kline.HundredMilliseconds},
{Enabled: true, Asset: asset.All, Channel: subscription.AllTradesChannel, Interval: kline.HundredMilliseconds},
{Enabled: true, Asset: asset.All, Channel: subscription.MyOrdersChannel, Interval: kline.HundredMilliseconds, Authenticated: true},
{Enabled: true, Asset: asset.All, Channel: subscription.MyTradesChannel, Interval: kline.HundredMilliseconds, Authenticated: true},
}
// WsConnect starts a new connection with the websocket API
func (e *Exchange) WsConnect() error {
ctx := context.TODO()
if !e.Websocket.IsEnabled() || !e.IsEnabled() {
return websocket.ErrWebsocketNotEnabled
}
var dialer gws.Dialer
if err := e.Websocket.Conn.Dial(ctx, &dialer, http.Header{}); err != nil {
return err
}
e.Websocket.Wg.Add(1)
go e.wsReadData(ctx)
go e.wsStartHeartbeat(ctx)
if e.Websocket.CanUseAuthenticatedEndpoints() {
if err := e.wsLogin(ctx); err != nil {
log.Errorf(log.ExchangeSys, "%v - authentication failed: %v\n", e.Name, err)
e.Websocket.SetCanUseAuthenticatedEndpoints(false)
}
}
return nil
}
func (e *Exchange) wsStartHeartbeat(ctx context.Context) {
msg := wsInput{
ID: e.MessageID(),
JSONRPCVersion: rpcVersion,
Method: "public/set_heartbeat",
Params: map[string]any{
"interval": 15,
},
}
respRaw, err := e.Websocket.Conn.SendMessageReturnResponse(ctx, request.Unset, msg.ID, msg)
if err != nil {
log.Errorf(log.ExchangeSys, "%v %s: %s\n", e.Name, errStartingHeartbeat, err)
return
}
var resp wsResponse
if err := json.Unmarshal(respRaw, &resp); err != nil {
log.Errorf(log.ExchangeSys, "%v %s: %s\n", e.Name, errStartingHeartbeat, err)
}
if resp.Error.Code != 0 || resp.Error.Message != "" {
log.Errorf(log.ExchangeSys, "%s %s code: %d message: %s", e.Name, errStartingHeartbeat, resp.Error.Code, resp.Error.Message)
}
}
func (e *Exchange) wsLogin(ctx context.Context) error {
if !e.IsWebsocketAuthenticationSupported() {
return fmt.Errorf("%v AuthenticatedWebsocketAPISupport not enabled", e.Name)
}
creds, err := e.GetCredentials(ctx)
if err != nil {
return err
}
e.Websocket.SetCanUseAuthenticatedEndpoints(true)
n := e.Requester.GetNonce(nonce.UnixNano).String()
strTS := strconv.FormatInt(time.Now().UnixMilli(), 10)
str2Sign := strTS + "\n" + n + "\n"
hmac, err := crypto.GetHMAC(crypto.HashSHA256, []byte(str2Sign), []byte(creds.Secret))
if err != nil {
return err
}
req := wsInput{
JSONRPCVersion: rpcVersion,
Method: "public/auth",
ID: e.MessageID(),
Params: map[string]any{
"grant_type": "client_signature",
"client_id": creds.Key,
"timestamp": strTS,
"nonce": n,
"signature": hex.EncodeToString(hmac),
},
}
resp, err := e.Websocket.Conn.SendMessageReturnResponse(ctx, request.Unset, req.ID, req)
if err != nil {
e.Websocket.SetCanUseAuthenticatedEndpoints(false)
return err
}
var response wsLoginResponse
err = json.Unmarshal(resp, &response)
if err != nil {
return fmt.Errorf("%v %v", e.Name, err)
}
if response.Error != nil && (response.Error.Code > 0 || response.Error.Message != "") {
return fmt.Errorf("%v Error:%v Message:%v", e.Name, response.Error.Code, response.Error.Message)
}
return nil
}
// wsReadData receives and passes on websocket messages for processing
func (e *Exchange) wsReadData(ctx context.Context) {
defer e.Websocket.Wg.Done()
for {
resp := e.Websocket.Conn.ReadMessage()
if resp.Raw == nil {
return
}
err := e.wsHandleData(ctx, resp.Raw)
if err != nil {
e.Websocket.DataHandler <- err
}
}
}
func (e *Exchange) wsHandleData(ctx context.Context, respRaw []byte) error {
var response wsResponse
err := json.Unmarshal(respRaw, &response)
if err != nil {
return fmt.Errorf("%s - err %s could not parse websocket data: %s", e.Name, err, respRaw)
}
if response.Method == "heartbeat" {
go e.wsSendHeartbeat(ctx)
return nil
}
if response.ID != "" {
if strings.HasPrefix(response.ID, "hb-") {
return nil
}
return e.Websocket.Match.RequireMatchWithData(response.ID, respRaw)
}
channels := strings.Split(response.Params.Channel, ".")
switch channels[0] {
case "announcements":
announcement := &Announcement{}
response.Params.Data = announcement
err = json.Unmarshal(respRaw, &response)
if err != nil {
return err
}
e.Websocket.DataHandler <- announcement
case "book":
return e.processOrderbook(respRaw, channels)
case "chart":
return e.processCandleChart(respRaw, channels)
case "deribit_price_index":
indexPrice := &wsIndexPrice{}
return e.processData(respRaw, indexPrice)
case "deribit_price_ranking":
priceRankings := &wsRankingPrices{}
return e.processData(respRaw, priceRankings)
case "deribit_price_statistics":
priceStatistics := &wsPriceStatistics{}
return e.processData(respRaw, priceStatistics)
case "deribit_volatility_index":
volatilityIndex := &wsVolatilityIndex{}
return e.processData(respRaw, volatilityIndex)
case "estimated_expiration_price":
estimatedExpirationPrice := &wsEstimatedExpirationPrice{}
return e.processData(respRaw, estimatedExpirationPrice)
case "incremental_ticker":
return e.processIncrementalTicker(respRaw, channels)
case "instrument":
instrumentState := &wsInstrumentState{}
return e.processData(respRaw, instrumentState)
case "markprice":
markPriceOptions := []wsMarkPriceOptions{}
return e.processData(respRaw, markPriceOptions)
case "perpetual":
perpetualInterest := &wsPerpetualInterest{}
return e.processData(respRaw, perpetualInterest)
case platformStateChannel:
platformState := &wsPlatformState{}
return e.processData(respRaw, platformState)
case "quote": // Quote ticker information.
return e.processQuoteTicker(respRaw, channels)
case "ticker":
return e.processInstrumentTicker(respRaw, channels)
case "trades":
return e.processTrades(respRaw, channels)
case "user":
switch channels[1] {
case "access_log":
accessLog := &wsAccessLog{}
return e.processData(respRaw, accessLog)
case "changes":
return e.processUserOrderChanges(respRaw, channels)
case "lock":
userLock := &WsUserLock{}
return e.processData(respRaw, userLock)
case "mmp_trigger":
data := &WsMMPTrigger{
Currency: channels[2],
}
return e.processData(respRaw, data)
case "orders":
return e.processUserOrders(respRaw, channels)
case "portfolio":
portfolio := &wsUserPortfolio{}
return e.processData(respRaw, portfolio)
case "trades":
return e.processTrades(respRaw, channels)
default:
e.Websocket.DataHandler <- websocket.UnhandledMessageWarning{
Message: e.Name + websocket.UnhandledMessage + string(respRaw),
}
return nil
}
case "public/test", "public/set_heartbeat":
default:
switch result := response.Result.(type) {
case string:
if result == "ok" {
return nil
}
default:
e.Websocket.DataHandler <- websocket.UnhandledMessageWarning{
Message: e.Name + websocket.UnhandledMessage + string(respRaw),
}
return nil
}
}
return nil
}
func (e *Exchange) wsSendHeartbeat(ctx context.Context) {
msg := WsSubscriptionInput{
ID: "hb-" + e.MessageID(),
JSONRPCVersion: rpcVersion,
Method: "public/test",
}
if err := e.Websocket.Conn.SendJSONMessage(ctx, request.Unset, msg); err != nil {
log.Errorf(log.ExchangeSys, "%v %s: %s\n", e.Name, errSendingHeartbeat, err)
}
}
func (e *Exchange) processUserOrders(respRaw []byte, channels []string) error {
if len(channels) != 4 && len(channels) != 5 {
return fmt.Errorf("%w, expected format 'user.orders.{instrument_name}.raw, user.orders.{instrument_name}.{interval}, user.orders.{kind}.{currency}.raw, or user.orders.{kind}.{currency}.{interval}', but found %s", errMalformedData, strings.Join(channels, "."))
}
var response wsResponse
orderData := []WsOrder{}
response.Params.Data = orderData
err := json.Unmarshal(respRaw, &response)
if err != nil {
return err
}
orderDetails := make([]order.Detail, len(orderData))
for x := range orderData {
a, cp, err := getAssetPairByInstrument(orderData[x].InstrumentName)
if err != nil {
return err
}
oType, err := order.StringToOrderType(orderData[x].OrderType)
if err != nil {
return err
}
side, err := order.StringToOrderSide(orderData[x].Direction)
if err != nil {
return err
}
status, err := order.StringToOrderStatus(orderData[x].OrderState)
if err != nil {
return err
}
orderDetails[x] = order.Detail{
Price: orderData[x].Price,
Amount: orderData[x].Amount,
ExecutedAmount: orderData[x].FilledAmount,
RemainingAmount: orderData[x].Amount - orderData[x].FilledAmount,
Exchange: e.Name,
OrderID: orderData[x].OrderID,
Type: oType,
Side: side,
Status: status,
AssetType: a,
Date: orderData[x].CreationTimestamp.Time(),
LastUpdated: orderData[x].LastUpdateTimestamp.Time(),
Pair: cp,
}
}
e.Websocket.DataHandler <- orderDetails
return nil
}
func (e *Exchange) processUserOrderChanges(respRaw []byte, channels []string) error {
if len(channels) < 4 || len(channels) > 5 {
return fmt.Errorf("%w, expected format 'trades.{instrument_name}.{interval} or trades.{kind}.{currency}.{interval}', but found %s", errMalformedData, strings.Join(channels, "."))
}
var response wsResponse
changeData := &wsChanges{}
response.Params.Data = changeData
err := json.Unmarshal(respRaw, &response)
if err != nil {
return err
}
td := make([]trade.Data, len(changeData.Trades))
for x := range changeData.Trades {
var side order.Side
side, err = order.StringToOrderSide(changeData.Trades[x].Direction)
if err != nil {
return err
}
var cp currency.Pair
var a asset.Item
a, cp, err = getAssetPairByInstrument(changeData.Trades[x].InstrumentName)
if err != nil {
return err
}
td[x] = trade.Data{
CurrencyPair: cp,
Exchange: e.Name,
Timestamp: changeData.Trades[x].Timestamp.Time(),
Price: changeData.Trades[x].Price,
Amount: changeData.Trades[x].Amount,
Side: side,
TID: changeData.Trades[x].TradeID,
AssetType: a,
}
}
err = trade.AddTradesToBuffer(td...)
if err != nil {
return err
}
orders := make([]order.Detail, len(changeData.Orders))
for x := range orders {
oType, err := order.StringToOrderType(changeData.Orders[x].OrderType)
if err != nil {
return err
}
side, err := order.StringToOrderSide(changeData.Orders[x].Direction)
if err != nil {
return err
}
status, err := order.StringToOrderStatus(changeData.Orders[x].OrderState)
if err != nil {
return err
}
a, cp, err := getAssetPairByInstrument(changeData.Orders[x].InstrumentName)
if err != nil {
return err
}
orders[x] = order.Detail{
Price: changeData.Orders[x].Price,
Amount: changeData.Orders[x].Amount,
ExecutedAmount: changeData.Orders[x].FilledAmount,
RemainingAmount: changeData.Orders[x].Amount - changeData.Orders[x].FilledAmount,
Exchange: e.Name,
OrderID: changeData.Orders[x].OrderID,
Type: oType,
Side: side,
Status: status,
AssetType: a,
Date: changeData.Orders[x].CreationTimestamp.Time(),
LastUpdated: changeData.Orders[x].LastUpdateTimestamp.Time(),
Pair: cp,
}
}
e.Websocket.DataHandler <- orders
e.Websocket.DataHandler <- changeData.Positions
return nil
}
func (e *Exchange) processQuoteTicker(respRaw []byte, channels []string) error {
a, cp, err := getAssetPairByInstrument(channels[1])
if err != nil {
return err
}
var response wsResponse
quoteTicker := &wsQuoteTickerInformation{}
response.Params.Data = quoteTicker
err = json.Unmarshal(respRaw, &response)
if err != nil {
return err
}
e.Websocket.DataHandler <- &ticker.Price{
ExchangeName: e.Name,
Pair: cp,
AssetType: a,
LastUpdated: quoteTicker.Timestamp.Time(),
Bid: quoteTicker.BestBidPrice,
Ask: quoteTicker.BestAskPrice,
BidSize: quoteTicker.BestBidAmount,
AskSize: quoteTicker.BestAskAmount,
}
return nil
}
func (e *Exchange) processTrades(respRaw []byte, channels []string) error {
tradeFeed := e.IsTradeFeedEnabled()
saveTradeData := e.IsSaveTradeDataEnabled()
if !tradeFeed && !saveTradeData {
return nil
}
if len(channels) < 3 || len(channels) > 5 {
return fmt.Errorf("%w, expected format 'trades.{instrument_name}.{interval} or trades.{kind}.{currency}.{interval}', but found %s", errMalformedData, strings.Join(channels, "."))
}
var response wsResponse
var tradeList []wsTrade
response.Params.Data = &tradeList
err := json.Unmarshal(respRaw, &response)
if err != nil {
return err
}
if len(tradeList) == 0 {
return fmt.Errorf("%v, empty list of trades found", common.ErrNoResponse)
}
tradesData := make([]trade.Data, len(tradeList))
for x := range tradesData {
var cp currency.Pair
var a asset.Item
a, cp, err = getAssetPairByInstrument(tradeList[x].InstrumentName)
if err != nil {
return err
}
tradesData[x] = trade.Data{
CurrencyPair: cp,
Exchange: e.Name,
Timestamp: tradeList[x].Timestamp.Time().UTC(),
Price: tradeList[x].Price,
Amount: tradeList[x].Amount,
Side: tradeList[x].Direction,
TID: tradeList[x].TradeID,
AssetType: a,
}
}
if tradeFeed {
for i := range tradesData {
e.Websocket.DataHandler <- tradesData[i]
}
}
if saveTradeData {
return trade.AddTradesToBuffer(tradesData...)
}
return nil
}
func (e *Exchange) processIncrementalTicker(respRaw []byte, channels []string) error {
if len(channels) != 2 {
return fmt.Errorf("%w, expected format 'incremental_ticker.{instrument_name}', but found %s", errMalformedData, strings.Join(channels, "."))
}
a, cp, err := getAssetPairByInstrument(channels[1])
if err != nil {
return err
}
var response wsResponse
incrementalTicker := &WsIncrementalTicker{}
response.Params.Data = incrementalTicker
err = json.Unmarshal(respRaw, &response)
if err != nil {
return err
}
e.Websocket.DataHandler <- &ticker.Price{
ExchangeName: e.Name,
Pair: cp,
AssetType: a,
LastUpdated: incrementalTicker.Timestamp.Time(),
BidSize: incrementalTicker.BestBidAmount,
AskSize: incrementalTicker.BestAskAmount,
High: incrementalTicker.MaxPrice,
Low: incrementalTicker.MinPrice,
Volume: incrementalTicker.Stats.Volume,
QuoteVolume: incrementalTicker.Stats.VolumeUsd,
Ask: incrementalTicker.ImpliedAsk,
Bid: incrementalTicker.ImpliedBid,
}
return nil
}
func (e *Exchange) processInstrumentTicker(respRaw []byte, channels []string) error {
if len(channels) != 3 {
return fmt.Errorf("%w, expected format 'ticker.{instrument_name}.{interval}', but found %s", errMalformedData, strings.Join(channels, "."))
}
return e.processTicker(respRaw, channels)
}
func (e *Exchange) processTicker(respRaw []byte, channels []string) error {
a, cp, err := getAssetPairByInstrument(channels[1])
if err != nil {
return err
}
var response wsResponse
tickerPriceResponse := &wsTicker{}
response.Params.Data = tickerPriceResponse
err = json.Unmarshal(respRaw, &response)
if err != nil {
return err
}
tickerPrice := &ticker.Price{
ExchangeName: e.Name,
Pair: cp,
AssetType: a,
LastUpdated: tickerPriceResponse.Timestamp.Time(),
Bid: tickerPriceResponse.BestBidPrice,
Ask: tickerPriceResponse.BestAskPrice,
BidSize: tickerPriceResponse.BestBidAmount,
AskSize: tickerPriceResponse.BestAskAmount,
Last: tickerPriceResponse.LastPrice,
High: tickerPriceResponse.Stats.High,
Low: tickerPriceResponse.Stats.Low,
Volume: tickerPriceResponse.Stats.Volume,
}
if a != asset.Futures {
tickerPrice.Low = tickerPriceResponse.MinPrice
tickerPrice.High = tickerPriceResponse.MaxPrice
tickerPrice.Last = tickerPriceResponse.MarkPrice
tickerPrice.Ask = tickerPriceResponse.ImpliedAsk
tickerPrice.Bid = tickerPriceResponse.ImpliedBid
}
e.Websocket.DataHandler <- tickerPrice
return nil
}
func (e *Exchange) processData(respRaw []byte, result any) error {
var response wsResponse
response.Params.Data = result
err := json.Unmarshal(respRaw, &response)
if err != nil {
return err
}
e.Websocket.DataHandler <- result
return nil
}
func (e *Exchange) processCandleChart(respRaw []byte, channels []string) error {
if len(channels) != 4 {
return fmt.Errorf("%w, expected format 'chart.trades.{instrument_name}.{resolution}', but found %s", errMalformedData, strings.Join(channels, "."))
}
a, cp, err := getAssetPairByInstrument(channels[2])
if err != nil {
return err
}
var response wsResponse
candleData := &wsCandlestickData{}
response.Params.Data = candleData
err = json.Unmarshal(respRaw, &response)
if err != nil {
return err
}
e.Websocket.DataHandler <- websocket.KlineData{
Timestamp: candleData.Tick.Time(),
Pair: cp,
AssetType: a,
Exchange: e.Name,
OpenPrice: candleData.Open,
HighPrice: candleData.High,
LowPrice: candleData.Low,
ClosePrice: candleData.Close,
Volume: candleData.Volume,
}
return nil
}
func (e *Exchange) processOrderbook(respRaw []byte, channels []string) error {
var response wsResponse
orderbookData := &wsOrderbook{}
response.Params.Data = orderbookData
err := json.Unmarshal(respRaw, &response)
if err != nil {
return err
}
if len(channels) == 3 {
a, cp, err := getAssetPairByInstrument(orderbookData.InstrumentName)
if err != nil {
return err
}
asks := make(orderbook.Levels, 0, len(orderbookData.Asks))
for x := range orderbookData.Asks {
if len(orderbookData.Asks[x]) != 3 {
return errMalformedData
}
price, okay := orderbookData.Asks[x][1].(float64)
if !okay {
return fmt.Errorf("%w, invalid orderbook price", errMalformedData)
}
amount, okay := orderbookData.Asks[x][2].(float64)
if !okay {
return fmt.Errorf("%w, invalid amount", errMalformedData)
}
asks = append(asks, orderbook.Level{
Price: price,
Amount: amount,
})
}
bids := make(orderbook.Levels, 0, len(orderbookData.Bids))
for x := range orderbookData.Bids {
if len(orderbookData.Bids[x]) != 3 {
return errMalformedData
}
price, okay := orderbookData.Bids[x][1].(float64)
if !okay {
return fmt.Errorf("%w, invalid orderbook price", errMalformedData)
} else if price == 0.0 {
continue
}
amount, okay := orderbookData.Bids[x][2].(float64)
if !okay {
return fmt.Errorf("%w, invalid amount", errMalformedData)
}
bids = append(bids, orderbook.Level{
Price: price,
Amount: amount,
})
}
if len(asks) == 0 && len(bids) == 0 {
return nil
}
switch orderbookData.Type {
case "snapshot":
return e.Websocket.Orderbook.LoadSnapshot(&orderbook.Book{
Exchange: e.Name,
ValidateOrderbook: e.ValidateOrderbook,
LastUpdated: orderbookData.Timestamp.Time(),
Pair: cp,
Asks: asks,
Bids: bids,
Asset: a,
LastUpdateID: orderbookData.ChangeID,
})
case "change":
return e.Websocket.Orderbook.Update(&orderbook.Update{
Asks: asks,
Bids: bids,
Pair: cp,
Asset: a,
UpdateID: orderbookData.ChangeID,
UpdateTime: orderbookData.Timestamp.Time(),
})
}
} else if len(channels) == 5 {
a, cp, err := getAssetPairByInstrument(orderbookData.InstrumentName)
if err != nil {
return err
}
asks := make(orderbook.Levels, 0, len(orderbookData.Asks))
for x := range orderbookData.Asks {
if len(orderbookData.Asks[x]) != 2 {
return errMalformedData
}
price, okay := orderbookData.Asks[x][0].(float64)
if !okay {
return fmt.Errorf("%w, invalid orderbook price", errMalformedData)
} else if price == 0 {
continue
}
amount, okay := orderbookData.Asks[x][1].(float64)
if !okay {
return fmt.Errorf("%w, invalid amount", errMalformedData)
}
asks = append(asks, orderbook.Level{
Price: price,
Amount: amount,
})
}
bids := make([]orderbook.Level, 0, len(orderbookData.Bids))
for x := range orderbookData.Bids {
if len(orderbookData.Bids[x]) != 2 {
return errMalformedData
}
price, okay := orderbookData.Bids[x][0].(float64)
if !okay {
return fmt.Errorf("%w, invalid orderbook price", errMalformedData)
} else if price == 0 {
continue
}
amount, okay := orderbookData.Bids[x][1].(float64)
if !okay {
return fmt.Errorf("%w, invalid amount", errMalformedData)
}
bids = append(bids, orderbook.Level{
Price: price,
Amount: amount,
})
}
if len(asks) == 0 && len(bids) == 0 {
return nil
}
return e.Websocket.Orderbook.LoadSnapshot(&orderbook.Book{
Asks: asks,
Bids: bids,
Pair: cp,
Asset: a,
Exchange: e.Name,
LastUpdateID: orderbookData.ChangeID,
LastUpdated: orderbookData.Timestamp.Time(),
})
}
return nil
}
// generateSubscriptions returns a list of configured subscriptions
func (e *Exchange) generateSubscriptions() (subscription.List, error) {
return e.Features.Subscriptions.ExpandTemplates(e)
}
// GetSubscriptionTemplate returns a subscription channel template
func (e *Exchange) GetSubscriptionTemplate(_ *subscription.Subscription) (*template.Template, error) {
return template.New("master.tmpl").Funcs(template.FuncMap{
"channelName": channelName,
"interval": channelInterval,
"isSymbolChannel": isSymbolChannel,
"fmt": formatChannelPair,
}).
Parse(subTplText)
}
// Subscribe sends a websocket message to receive data from the channel
func (e *Exchange) Subscribe(subs subscription.List) error {
ctx := context.TODO()
errs := e.handleSubscription(ctx, "public/subscribe", subs.Public())
return common.AppendError(errs, e.handleSubscription(ctx, "private/subscribe", subs.Private()))
}
// Unsubscribe sends a websocket message to stop receiving data from the channel
func (e *Exchange) Unsubscribe(subs subscription.List) error {
ctx := context.TODO()
errs := e.handleSubscription(ctx, "public/unsubscribe", subs.Public())
return common.AppendError(errs, e.handleSubscription(ctx, "private/unsubscribe", subs.Private()))
}
func (e *Exchange) handleSubscription(ctx context.Context, method string, subs subscription.List) error {
var err error
subs, err = subs.ExpandTemplates(e)
if err != nil || len(subs) == 0 {
return err
}
r := WsSubscriptionInput{
JSONRPCVersion: rpcVersion,
ID: e.MessageID(),
Method: method,
Params: map[string][]string{"channels": subs.QualifiedChannels()},
}
data, err := e.Websocket.Conn.SendMessageReturnResponse(ctx, request.Unset, r.ID, r)
if err != nil {
return err
}
var response wsSubscriptionResponse
err = json.Unmarshal(data, &response)
if err != nil {
return fmt.Errorf("%v %v", e.Name, err)
}
subAck := map[string]bool{}
for _, c := range response.Result {
subAck[c] = true
}
if len(subAck) != len(subs) {
err = websocket.ErrSubscriptionFailure
}
for _, s := range subs {
if _, ok := subAck[s.QualifiedChannel]; ok {
delete(subAck, s.QualifiedChannel)
if !strings.Contains(method, "unsubscribe") {
err = common.AppendError(err, e.Websocket.AddSuccessfulSubscriptions(e.Websocket.Conn, s))
} else {
err = common.AppendError(err, e.Websocket.RemoveSubscriptions(e.Websocket.Conn, s))
}
} else {
err = common.AppendError(err, errors.New(s.String()+" failed to "+method))
}
}
for key := range subAck {
err = common.AppendError(err, fmt.Errorf("unexpected channel %q in result", key))
}
return err
}
func getValidatedCurrencyCode(pair currency.Pair) string {
currencyCode := pair.Base.Upper().String()
switch currencyCode {
case currencyBTC, currencyETH,
currencySOL, currencyUSDT,
currencyUSDC, currencyEURR:
return currencyCode
default:
switch {
case strings.Contains(pair.String(), currencyUSDC):
return currencyUSDC
case strings.Contains(pair.String(), currencyUSDT):
return currencyUSDT
}
return "any"
}
}
func channelName(s *subscription.Subscription) string {
if name, ok := subscriptionNames[s.Channel]; ok {
return name
}
panic(fmt.Errorf("%w: %s", subscription.ErrNotSupported, s.Channel))
}
// channelInterval converts an interval to an exchange specific interval
// We convert 1s to agg2; Docs do not explain agg2 but support explained that it may vary under load but is currently 1 second
func channelInterval(s *subscription.Subscription) string {
if s.Interval != 0 {
if channelName(s) == chartTradesChannel {
if s.Interval == kline.OneDay {
return "1D"
}
m := s.Interval.Duration().Minutes()
switch m {
case 1, 3, 5, 10, 15, 30, 60, 120, 180, 360, 720: // Valid Minute intervals
return strconv.Itoa(int(m))
}
panic(fmt.Errorf("%w: %s", kline.ErrUnsupportedInterval, s.Interval))
}
switch s.Interval {
case kline.ThousandMilliseconds:
return "agg2"
case kline.HundredMilliseconds, kline.Raw:
return s.Interval.Short()
}
panic(fmt.Errorf("%w: %s", kline.ErrUnsupportedInterval, s.Interval))
}
return ""
}
func isSymbolChannel(s *subscription.Subscription) bool {
switch channelName(s) {
case orderbookChannel, chartTradesChannel, tickerChannel, tradesChannel, perpetualChannel, quoteChannel,
userChangesInstrumentsChannel, incrementalTickerChannel, userOrdersChannel, userTradesChannel:
return true
}
return false
}
func formatChannelPair(pair currency.Pair) string {
if str := pair.Quote.String(); strings.Contains(str, "PERPETUAL") && strings.Contains(str, "-") {
pair.Delimiter = "_"
}
return pair.String()
}
const subTplText = `
{{- if isSymbolChannel $.S -}}
{{- range $asset, $pairs := $.AssetPairs }}
{{- range $p := $pairs }}
{{- channelName $.S -}} . {{- fmt $p }}
{{- with $i := interval $.S -}} . {{- $i }}{{ end }}
{{- $.PairSeparator }}
{{- end }}
{{- $.AssetSeparator }}
{{- end }}
{{- else }}
{{- channelName $.S -}}
{{- with $i := interval $.S -}} . {{- $i }}{{ end }}
{{- end }}
`