lychiyu/bbgo_origin
1
0
Fork 0
mirror of https://github.com/c9s/bbgo.git synced 2024-11-26 08:45:16 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix rbtree price volume order

Browse Source
This commit is contained in:
c9s 2021-05-23 00:42:27 +08:00
parent 7a653affa6
commit 9fa10ee1fd
1 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
pkg/types/rbtorderbook.go
View File

@ -137,15 +137,17 @@ func (b *RBTOrderBook) CopyDepth(limit int) OrderBook {
return book
}
func (b *RBTOrderBook) convertTreeToPriceVolumeSlice(tree *RBTree, descending bool) (pvs PriceVolumeSlice) {
func (b *RBTOrderBook) convertTreeToPriceVolumeSlice(tree *RBTree, limit int, descending bool) (pvs PriceVolumeSlice) {
if descending {
tree.InorderReverse(func(n *RBNode) bool {
pvs = append(pvs, PriceVolume{
Price: n.Key,
Volume: n.Value,
})
return true
return !(limit > 0 && len(pvs) >= limit)
})
return pvs
}
@ -154,7 +156,8 @@ func (b *RBTOrderBook) convertTreeToPriceVolumeSlice(tree *RBTree, descending bo
Price: n.Key,
Volume: n.Value,
})
return true
return !(limit > 0 && len(pvs) >= limit)
})
return pvs
}
@ -163,10 +166,10 @@ func (b *RBTOrderBook) SideBook(sideType SideType) PriceVolumeSlice {
switch sideType {
case SideTypeBuy:
return b.convertTreeToPriceVolumeSlice(b.Bids, false)
return b.convertTreeToPriceVolumeSlice(b.Bids, 0, true)
case SideTypeSell:
return b.convertTreeToPriceVolumeSlice(b.Asks, true)
return b.convertTreeToPriceVolumeSlice(b.Asks, 0, false)
default:
return nil