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

fix rbtree memory error, check neel

Browse Source
This commit is contained in:
c9s 2021-05-23 01:12:16 +08:00
parent 9c70e36e1b
commit de768296f1
1 changed files with 12 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
pkg/types/rbtree.go
View File

@ -336,14 +336,14 @@ func (tree *RBTree) Rightmost() *RBNode {
} }
func (tree *RBTree) RightmostOf(current *RBNode) *RBNode { func (tree *RBTree) RightmostOf(current *RBNode) *RBNode {
for current.Right != Neel {
current = current.Right
}
if current == Neel { if current == Neel {
return nil return nil
} }
for current.Right != Neel {
current = current.Right
}
return current return current
} }
@ -352,18 +352,22 @@ func (tree *RBTree) Leftmost() *RBNode {
} }
func (tree *RBTree) LeftmostOf(current *RBNode) *RBNode { func (tree *RBTree) LeftmostOf(current *RBNode) *RBNode {
for current.Left != Neel {
current = current.Left
}
if current == Neel { if current == Neel {
return nil return nil
} }
for current.Left != Neel {
current = current.Left
}
return current return current
} }
func (tree *RBTree) Successor(current *RBNode) *RBNode { func (tree *RBTree) Successor(current *RBNode) *RBNode {
if current == Neel {
return nil
}
if current.Right != Neel { if current.Right != Neel {
return tree.LeftmostOf(current.Right) return tree.LeftmostOf(current.Right)
} }