Chris Lu
5 years ago
7 changed files with 265 additions and 54 deletions
-
10weed/filesys/dir.go
-
29weed/filesys/dir_rename.go
-
2weed/filesys/file.go
-
147weed/filesys/fscache.go
-
90weed/filesys/fscache_test.go
-
33weed/filesys/wfs.go
-
8weed/util/fullpath.go
@ -0,0 +1,147 @@ |
|||
package filesys |
|||
|
|||
import ( |
|||
"github.com/chrislusf/seaweedfs/weed/util" |
|||
"github.com/seaweedfs/fuse/fs" |
|||
) |
|||
|
|||
type FsCache struct { |
|||
root *FsNode |
|||
} |
|||
type FsNode struct { |
|||
parent *FsNode |
|||
node fs.Node |
|||
name string |
|||
children map[string]*FsNode |
|||
} |
|||
|
|||
func newFsCache(root fs.Node) *FsCache { |
|||
return &FsCache{ |
|||
root: &FsNode{ |
|||
node: root, |
|||
}, |
|||
} |
|||
} |
|||
|
|||
func (c *FsCache) GetFsNode(path util.FullPath) fs.Node { |
|||
t := c.root |
|||
for _, p := range path.Split() { |
|||
t = t.findChild(p) |
|||
if t == nil { |
|||
return nil |
|||
} |
|||
} |
|||
return t.node |
|||
} |
|||
|
|||
func (c *FsCache) SetFsNode(path util.FullPath, node fs.Node) { |
|||
t := c.root |
|||
for _, p := range path.Split() { |
|||
t = t.ensureChild(p) |
|||
} |
|||
t.node = node |
|||
} |
|||
|
|||
func (c *FsCache) EnsureFsNode(path util.FullPath, genNodeFn func() fs.Node) fs.Node { |
|||
t := c.GetFsNode(path) |
|||
if t != nil { |
|||
return t |
|||
} |
|||
t = genNodeFn() |
|||
c.SetFsNode(path, t) |
|||
return t |
|||
} |
|||
|
|||
func (c *FsCache) DeleteFsNode(path util.FullPath) { |
|||
t := c.root |
|||
for _, p := range path.Split() { |
|||
t = t.findChild(p) |
|||
if t == nil { |
|||
return |
|||
} |
|||
} |
|||
if t.parent != nil { |
|||
t.parent.deleteChild(t.name) |
|||
} |
|||
t.deleteSelf() |
|||
} |
|||
|
|||
// oldPath and newPath are full path including the new name
|
|||
func (c *FsCache) Move(oldPath util.FullPath, newPath util.FullPath) *FsNode { |
|||
|
|||
// find old node
|
|||
src := c.root |
|||
for _, p := range oldPath.Split() { |
|||
src = src.findChild(p) |
|||
if src == nil { |
|||
return src |
|||
} |
|||
} |
|||
if src.parent != nil { |
|||
src.parent.deleteChild(src.name) |
|||
} |
|||
src.parent = nil |
|||
|
|||
// find new node
|
|||
target := c.root |
|||
for _, p := range newPath.Split() { |
|||
target = target.ensureChild(p) |
|||
} |
|||
parent := target.parent |
|||
src.name = target.name |
|||
parent.deleteChild(target.name) |
|||
|
|||
target.deleteSelf() |
|||
|
|||
src.connectToParent(parent) |
|||
|
|||
return src |
|||
} |
|||
|
|||
func (n *FsNode) connectToParent(parent *FsNode) { |
|||
n.parent = parent |
|||
oldNode := parent.findChild(n.name) |
|||
if oldNode != nil { |
|||
oldNode.deleteSelf() |
|||
} |
|||
parent.children[n.name] = n |
|||
} |
|||
|
|||
func (n *FsNode) findChild(name string) *FsNode { |
|||
child, found := n.children[name] |
|||
if found { |
|||
return child |
|||
} |
|||
return nil |
|||
} |
|||
|
|||
func (n *FsNode) ensureChild(name string) *FsNode { |
|||
if n.children == nil { |
|||
n.children = make(map[string]*FsNode) |
|||
} |
|||
child, found := n.children[name] |
|||
if found { |
|||
return child |
|||
} |
|||
t := &FsNode{ |
|||
parent: n, |
|||
node: nil, |
|||
name: name, |
|||
children: nil, |
|||
} |
|||
n.children[name] = t |
|||
return t |
|||
} |
|||
|
|||
func (n *FsNode) deleteChild(name string) { |
|||
delete(n.children, name) |
|||
} |
|||
|
|||
func (n *FsNode) deleteSelf() { |
|||
for _, child := range n.children { |
|||
child.deleteSelf() |
|||
} |
|||
n.node = nil |
|||
n.parent = nil |
|||
n.children = nil |
|||
} |
|||
@ -0,0 +1,90 @@ |
|||
package filesys |
|||
|
|||
import ( |
|||
"testing" |
|||
|
|||
"github.com/chrislusf/seaweedfs/weed/util" |
|||
) |
|||
|
|||
func TestPathSplit(t *testing.T) { |
|||
parts := util.FullPath("/").Split() |
|||
if len(parts) != 0 { |
|||
t.Errorf("expecting an empty list, but getting %d", len(parts)) |
|||
} |
|||
} |
|||
|
|||
func TestFsCache(t *testing.T) { |
|||
|
|||
cache := newFsCache(nil) |
|||
|
|||
x := cache.GetFsNode(util.FullPath("/y/x")) |
|||
if x != nil { |
|||
t.Errorf("wrong node!") |
|||
} |
|||
|
|||
p := util.FullPath("/a/b/c") |
|||
cache.SetFsNode(p, &File{Name: "cc"}) |
|||
tNode := cache.GetFsNode(p) |
|||
tFile := tNode.(*File) |
|||
if tFile.Name != "cc" { |
|||
t.Errorf("expecting a FsNode") |
|||
} |
|||
|
|||
cache.SetFsNode(util.FullPath("/a/b/d"), &File{Name: "dd"}) |
|||
cache.SetFsNode(util.FullPath("/a/b/e"), &File{Name: "ee"}) |
|||
cache.SetFsNode(util.FullPath("/a/b/f"), &File{Name: "ff"}) |
|||
cache.SetFsNode(util.FullPath("/z"), &File{Name: "zz"}) |
|||
cache.SetFsNode(util.FullPath("/a"), &File{Name: "aa"}) |
|||
|
|||
b := cache.GetFsNode(util.FullPath("/a/b")) |
|||
if b != nil { |
|||
t.Errorf("unexpected node!") |
|||
} |
|||
|
|||
a := cache.GetFsNode(util.FullPath("/a")) |
|||
if a == nil { |
|||
t.Errorf("missing node!") |
|||
} |
|||
|
|||
cache.DeleteFsNode(util.FullPath("/a")) |
|||
if b != nil { |
|||
t.Errorf("unexpected node!") |
|||
} |
|||
|
|||
a = cache.GetFsNode(util.FullPath("/a")) |
|||
if a != nil { |
|||
t.Errorf("wrong DeleteFsNode!") |
|||
} |
|||
|
|||
z := cache.GetFsNode(util.FullPath("/z")) |
|||
if z == nil { |
|||
t.Errorf("missing node!") |
|||
} |
|||
|
|||
y := cache.GetFsNode(util.FullPath("/x/y")) |
|||
if y != nil { |
|||
t.Errorf("wrong node!") |
|||
} |
|||
|
|||
} |
|||
|
|||
func TestFsCacheMove(t *testing.T) { |
|||
|
|||
cache := newFsCache(nil) |
|||
|
|||
cache.SetFsNode(util.FullPath("/a/b/d"), &File{Name: "dd"}) |
|||
cache.SetFsNode(util.FullPath("/a/b/e"), &File{Name: "ee"}) |
|||
cache.SetFsNode(util.FullPath("/z"), &File{Name: "zz"}) |
|||
cache.SetFsNode(util.FullPath("/a"), &File{Name: "aa"}) |
|||
|
|||
cache.Move(util.FullPath("/a/b"), util.FullPath("/z/x")) |
|||
|
|||
d := cache.GetFsNode(util.FullPath("/z/x/d")) |
|||
if d == nil { |
|||
t.Errorf("unexpected nil node!") |
|||
} |
|||
if d.(*File).Name != "dd" { |
|||
t.Errorf("unexpected non dd node!") |
|||
} |
|||
|
|||
} |
|||
Write
Preview
Loading…
Cancel
Save
Reference in new issue