lychiyu/bbgo_origin
1
0
Fork 0
mirror of https://github.com/c9s/bbgo.git synced 2024-11-22 23:05:15 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

pkg/exchange: support order book depth 200 on bybit

Browse Source
This commit is contained in:
Edwin 2023-10-12 11:11:26 +08:00
parent d617bf8ae5
commit ef582f6e52
3 changed files with 29 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
pkg/exchange/bybit/stream.go
View File

@ -318,8 +318,12 @@ func (s *Stream) convertSubscription(sub types.Subscription) (string, error) {
case types.BookChannel: case types.BookChannel:
depth := types.DepthLevel1 depth := types.DepthLevel1
if len(sub.Options.Depth) > 0 && sub.Options.Depth == types.DepthLevel50 {
depth = types.DepthLevel50 switch sub.Options.Depth {
case types.DepthLevel50:
depth = sub.Options.Depth
case types.DepthLevel200:
depth = sub.Options.Depth
} }
return genTopic(TopicTypeOrderBook, depth, sub.Symbol), nil return genTopic(TopicTypeOrderBook, depth, sub.Symbol), nil

22
pkg/exchange/bybit/stream_test.go
View File

@ -395,6 +395,28 @@ func Test_convertSubscription(t *testing.T) {
assert.NoError(t, err) assert.NoError(t, err)
assert.Equal(t, genTopic(TopicTypeOrderBook, types.DepthLevel1, "BTCUSDT"), res) assert.Equal(t, genTopic(TopicTypeOrderBook, types.DepthLevel1, "BTCUSDT"), res)
}) })
t.Run("BookChannel.DepthLevel50", func(t *testing.T) {
res, err := s.convertSubscription(types.Subscription{
Symbol: "BTCUSDT",
Channel: types.BookChannel,
Options: types.SubscribeOptions{
Depth: types.DepthLevel50,
},
})
assert.NoError(t, err)
assert.Equal(t, genTopic(TopicTypeOrderBook, types.DepthLevel50, "BTCUSDT"), res)
})
t.Run("BookChannel.DepthLevel200", func(t *testing.T) {
res, err := s.convertSubscription(types.Subscription{
Symbol: "BTCUSDT",
Channel: types.BookChannel,
Options: types.SubscribeOptions{
Depth: types.DepthLevel200,
},
})
assert.NoError(t, err)
assert.Equal(t, genTopic(TopicTypeOrderBook, types.DepthLevel200, "BTCUSDT"), res)
})
t.Run("BookChannel. with default depth", func(t *testing.T) { t.Run("BookChannel. with default depth", func(t *testing.T) {
res, err := s.convertSubscription(types.Subscription{ res, err := s.convertSubscription(types.Subscription{
Symbol: "BTCUSDT", Symbol: "BTCUSDT",

1
pkg/types/stream.go
View File

@ -521,6 +521,7 @@ const (
DepthLevel5 Depth = "5" DepthLevel5 Depth = "5"
DepthLevel20 Depth = "20" DepthLevel20 Depth = "20"
DepthLevel50 Depth = "50" DepthLevel50 Depth = "50"
DepthLevel200 Depth = "200"
) )
type Speed string type Speed string