fix checking visited nodes

This commit is contained in:
Chris Lu 2020-06-19 08:58:48 -07:00
parent 0e7c1a300b
commit 55b6efb755
2 changed files with 16 additions and 10 deletions

View file

@ -1,6 +1,8 @@
package bounded_tree
import (
"fmt"
"github.com/chrislusf/seaweedfs/weed/glog"
"github.com/chrislusf/seaweedfs/weed/util"
)
@ -133,7 +135,7 @@ func (t *BoundedTree) HasVisited(p util.FullPath) bool {
}
components := p.Split()
// fmt.Printf("components %v %d\n", components, len(components))
fmt.Printf("components %v %d\n", components, len(components))
return t.hasVisited(t.root, util.FullPath("/"), components, 0)
}
@ -147,17 +149,17 @@ func (t *BoundedTree) hasVisited(n *Node, currentPath util.FullPath, components
return false
}
// fmt.Printf(" hasVisited child %v %v\n", currentPath, components[i])
fmt.Printf(" hasVisited child %v %+v %d\n", currentPath, components, i)
if i >= len(components) {
return true
}
toVisitNode, found := n.Children[components[i]]
if !found {
return true
}
if i+1 >= len(components) {
return false
}
return t.hasVisited(toVisitNode, currentPath.Child(components[i]), components, i+1)
}

View file

@ -4,6 +4,8 @@ import (
"fmt"
"testing"
"github.com/stretchr/testify/assert"
"github.com/chrislusf/seaweedfs/weed/util"
)
@ -59,6 +61,8 @@ func TestBoundedTree(t *testing.T) {
tree.EnsureVisited(util.FullPath("/a/b/c"), visitFn)
assert.Equal(t, true, tree.HasVisited(util.FullPath("/a/b")))
printMap(tree.root.Children)
a := tree.root.getChild("a")
@ -110,9 +114,9 @@ func TestEmptyBoundedTree(t *testing.T) {
printMap(tree.root.Children)
println(tree.HasVisited(util.FullPath("/a/b")))
println(tree.HasVisited(util.FullPath("/a")))
println(tree.HasVisited(util.FullPath("/g")))
println(tree.HasVisited(util.FullPath("/g/x")))
assert.Equal(t, true, tree.HasVisited(util.FullPath("/a/b")))
assert.Equal(t, true, tree.HasVisited(util.FullPath("/a")))
assert.Equal(t, false, tree.HasVisited(util.FullPath("/g")))
assert.Equal(t, false, tree.HasVisited(util.FullPath("/g/x")))
}