lychiyu/bbgo_origin
1
0
Fork 0
mirror of https://github.com/c9s/bbgo.git synced 2024-11-14 02:53:50 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

types: avoid using copy node for rbtree

Browse Source
This commit is contained in:
c9s 2022-01-12 14:45:05 +08:00
parent c3356fa694
commit 0c7710c91b
2 changed files with 8 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
pkg/types/rbtree.go
View File

@ -435,17 +435,6 @@ func (tree *RBTree) PostorderOf(current *RBNode, cb func(n *RBNode) bool) {
} }
} }
func (tree *RBTree) copyNode(node *RBNode) *RBNode {
if node == neel {
return neel
}
newNode := *node
newNode.left = tree.copyNode(node.left)
newNode.right = tree.copyNode(node.right)
return &newNode
}
func (tree *RBTree) CopyInorderReverse(limit int) *RBTree { func (tree *RBTree) CopyInorderReverse(limit int) *RBTree {
cnt := 0 cnt := 0
newTree := NewRBTree() newTree := NewRBTree()
@ -483,9 +472,3 @@ func (tree *RBTree) Print() {
return true return true
}) })
} }
func (tree *RBTree) Copy() *RBTree {
newTree := NewRBTree()
newTree.Root = tree.copyNode(tree.Root)
return newTree
}

5
pkg/types/rbtree_test.go
View File

@ -4,8 +4,9 @@ import (
"math/rand" "math/rand"
"testing" "testing"
"github.com/c9s/bbgo/pkg/fixedpoint"
"github.com/stretchr/testify/assert" "github.com/stretchr/testify/assert"
"github.com/c9s/bbgo/pkg/fixedpoint"
) )
func TestRBTree_InsertAndDelete(t *testing.T) { func TestRBTree_InsertAndDelete(t *testing.T) {
@ -94,7 +95,7 @@ func TestTree_Copy(t *testing.T) {
tree.Insert(fixedpoint.NewFromFloat(4000.0), fixedpoint.NewFromFloat(2.0)) tree.Insert(fixedpoint.NewFromFloat(4000.0), fixedpoint.NewFromFloat(2.0))
tree.Insert(fixedpoint.NewFromFloat(2000.0), fixedpoint.NewFromFloat(3.0)) tree.Insert(fixedpoint.NewFromFloat(2000.0), fixedpoint.NewFromFloat(3.0))
newTree := tree.Copy() newTree := tree.CopyInorder(0)
node1 := newTree.Search(fixedpoint.NewFromFloat(2000.0)) node1 := newTree.Search(fixedpoint.NewFromFloat(2000.0))
assert.NotNil(t, node1) assert.NotNil(t, node1)
assert.Equal(t, fixedpoint.NewFromFloat(2000.0), node1.key) assert.Equal(t, fixedpoint.NewFromFloat(2000.0), node1.key)