mirror of
https://github.com/d0zingcat/gocryptotrader.git
synced 2026-05-18 15:10:03 +00:00
e007f69f7c
* Websockets: Move Subscription to its own package This allows the small type to be imported from both `config` and from `stream` without an import cycle, so we don't have to repeat ourselves * Subs: Renamed Currency to Pair This was being mis-used through much of the code, and since we're already touching everything, we might as well fix it * Websockets: Add Subscription configuration * Binance: Add subscription configuration * Kucoin: Subscription configuration * Simplify GenerateDefaultSubs * Improve TestGenSubs coverage * Test Candle Sub generation * Support Candle intervals * Full responsibility for formatting Channel name on GenerateDefaultSubs OR consumer of Subscribe * Simplify generatePayloads as a result * Fix test coverage of asset types in processMarketSnapshot * Exchanges: Abstract ParallelChanOp * Tests: Generic ws mock instances * Kucoin: Fix intermittent conflict in test currs Use isolated test instance for `TestGetOpenInterest`. `TestGetOpenInterest` would occassionally change pairs before GenerateDefault Subs.
100 lines
3.2 KiB
Go
100 lines
3.2 KiB
Go
package bybit
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
|
|
"github.com/gorilla/websocket"
|
|
"github.com/thrasher-corp/gocryptotrader/currency"
|
|
"github.com/thrasher-corp/gocryptotrader/exchanges/asset"
|
|
"github.com/thrasher-corp/gocryptotrader/exchanges/stream"
|
|
"github.com/thrasher-corp/gocryptotrader/exchanges/subscription"
|
|
)
|
|
|
|
// WsLinearConnect connects to linear a websocket feed
|
|
func (by *Bybit) WsLinearConnect() error {
|
|
if !by.Websocket.IsEnabled() || !by.IsEnabled() || !by.IsAssetWebsocketSupported(asset.LinearContract) {
|
|
return errWebsocketNotEnabled
|
|
}
|
|
by.Websocket.Conn.SetURL(linearPublic)
|
|
var dialer websocket.Dialer
|
|
err := by.Websocket.Conn.Dial(&dialer, http.Header{})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
by.Websocket.Conn.SetupPingHandler(stream.PingHandler{
|
|
MessageType: websocket.TextMessage,
|
|
Message: []byte(`{"op": "ping"}`),
|
|
Delay: bybitWebsocketTimer,
|
|
})
|
|
|
|
by.Websocket.Wg.Add(1)
|
|
go by.wsReadData(asset.LinearContract, by.Websocket.Conn)
|
|
if by.IsWebsocketAuthenticationSupported() {
|
|
err = by.WsAuth(context.TODO())
|
|
if err != nil {
|
|
by.Websocket.DataHandler <- err
|
|
by.Websocket.SetCanUseAuthenticatedEndpoints(false)
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// GenerateLinearDefaultSubscriptions generates default subscription
|
|
func (by *Bybit) GenerateLinearDefaultSubscriptions() ([]subscription.Subscription, error) {
|
|
var subscriptions []subscription.Subscription
|
|
var channels = []string{chanOrderbook, chanPublicTrade, chanPublicTicker}
|
|
pairs, err := by.GetEnabledPairs(asset.USDTMarginedFutures)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
linearPairMap := map[asset.Item]currency.Pairs{
|
|
asset.USDTMarginedFutures: pairs,
|
|
}
|
|
usdcPairs, err := by.GetEnabledPairs(asset.USDCMarginedFutures)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
linearPairMap[asset.USDCMarginedFutures] = usdcPairs
|
|
pairs = append(pairs, usdcPairs...)
|
|
for a := range linearPairMap {
|
|
for p := range linearPairMap[a] {
|
|
for x := range channels {
|
|
subscriptions = append(subscriptions,
|
|
subscription.Subscription{
|
|
Channel: channels[x],
|
|
Pair: pairs[p],
|
|
Asset: a,
|
|
})
|
|
}
|
|
}
|
|
}
|
|
return subscriptions, nil
|
|
}
|
|
|
|
// LinearSubscribe sends a subscription message to linear public channels.
|
|
func (by *Bybit) LinearSubscribe(channelSubscriptions []subscription.Subscription) error {
|
|
return by.handleLinearPayloadSubscription("subscribe", channelSubscriptions)
|
|
}
|
|
|
|
// LinearUnsubscribe sends an unsubscription messages through linear public channels.
|
|
func (by *Bybit) LinearUnsubscribe(channelSubscriptions []subscription.Subscription) error {
|
|
return by.handleLinearPayloadSubscription("unsubscribe", channelSubscriptions)
|
|
}
|
|
|
|
func (by *Bybit) handleLinearPayloadSubscription(operation string, channelSubscriptions []subscription.Subscription) error {
|
|
payloads, err := by.handleSubscriptions(asset.USDTMarginedFutures, operation, channelSubscriptions)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
for a := range payloads {
|
|
// The options connection does not send the subscription request id back with the subscription notification payload
|
|
// therefore the code doesn't wait for the response to check whether the subscription is successful or not.
|
|
err = by.Websocket.Conn.SendJSONMessage(payloads[a])
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|