seaweedfs/weed/filesys/dir.go

203 lines
4.3 KiB
Go
Raw Normal View History

2018-05-06 05:47:16 +00:00
package filesys
import (
"context"
"fmt"
2018-05-06 06:39:29 +00:00
"os"
2018-05-06 05:47:16 +00:00
"path"
2018-05-10 06:18:02 +00:00
"sync"
2018-05-06 06:39:29 +00:00
"bazil.org/fuse/fs"
"bazil.org/fuse"
2018-05-08 08:59:43 +00:00
"github.com/chrislusf/seaweedfs/weed/glog"
2018-05-10 06:18:02 +00:00
"github.com/chrislusf/seaweedfs/weed/pb/filer_pb"
"time"
2018-05-06 05:47:16 +00:00
)
type Dir struct {
2018-05-06 06:50:34 +00:00
Path string
NodeMap map[string]fs.Node
NodeMapLock sync.Mutex
wfs *WFS
2018-05-06 05:47:16 +00:00
}
func (dir *Dir) Attr(context context.Context, attr *fuse.Attr) error {
2018-05-08 08:59:43 +00:00
attr.Mode = os.ModeDir | 0777
2018-05-06 05:47:16 +00:00
return nil
}
func (dir *Dir) Create(ctx context.Context, req *fuse.CreateRequest,
resp *fuse.CreateResponse) (fs.Node, fs.Handle, error) {
err := dir.wfs.withFilerClient(func(client filer_pb.SeaweedFilerClient) error {
request := &filer_pb.CreateEntryRequest{
Directory: dir.Path,
Entry: &filer_pb.Entry{
Name: req.Name,
IsDirectory: req.Mode&os.ModeDir > 0,
Attributes: &filer_pb.FuseAttributes{
Mtime: time.Now().Unix(),
FileMode: uint32(req.Mode),
Uid: req.Uid,
Gid: req.Gid,
},
},
}
glog.V(1).Infof("create: %v", request)
if _, err := client.CreateEntry(ctx, request); err != nil {
return fmt.Errorf("create file: %v", err)
}
return nil
})
if err == nil {
node := &File{Name: req.Name, dir: dir, wfs: dir.wfs}
dir.NodeMap[req.Name] = node
return node, node, nil
}
return nil, nil, err
}
2018-05-08 08:59:43 +00:00
func (dir *Dir) Mkdir(ctx context.Context, req *fuse.MkdirRequest) (fs.Node, error) {
dir.NodeMapLock.Lock()
defer dir.NodeMapLock.Unlock()
err := dir.wfs.withFilerClient(func(client filer_pb.SeaweedFilerClient) error {
2018-05-08 08:59:43 +00:00
request := &filer_pb.CreateEntryRequest{
Directory: dir.Path,
Entry: &filer_pb.Entry{
Name: req.Name,
IsDirectory: true,
Attributes: &filer_pb.FuseAttributes{
Mtime: time.Now().Unix(),
FileMode: uint32(req.Mode),
Uid: req.Uid,
Gid: req.Gid,
},
},
}
2018-05-08 08:59:43 +00:00
glog.V(1).Infof("mkdir: %v", request)
if _, err := client.CreateEntry(ctx, request); err != nil {
return fmt.Errorf("make dir: %v", err)
}
return nil
})
if err == nil {
node := &Dir{Path: path.Join(dir.Path, req.Name), wfs: dir.wfs}
dir.NodeMap[req.Name] = node
return node, nil
}
return nil, err
2018-05-08 08:59:43 +00:00
}
2018-05-06 06:50:34 +00:00
func (dir *Dir) Lookup(ctx context.Context, name string) (node fs.Node, err error) {
dir.NodeMapLock.Lock()
defer dir.NodeMapLock.Unlock()
if dir.NodeMap == nil {
dir.NodeMap = make(map[string]fs.Node)
}
if node, ok := dir.NodeMap[name]; ok {
return node, nil
}
2018-05-10 06:18:02 +00:00
var entry *filer_pb.Entry
err = dir.wfs.withFilerClient(func(client filer_pb.SeaweedFilerClient) error {
2018-05-08 08:59:43 +00:00
2018-05-10 06:18:02 +00:00
request := &filer_pb.LookupDirectoryEntryRequest{
2018-05-08 08:59:43 +00:00
Directory: dir.Path,
Name: name,
2018-05-06 05:47:16 +00:00
}
2018-05-08 08:59:43 +00:00
glog.V(1).Infof("lookup directory entry: %v", request)
resp, err := client.LookupDirectoryEntry(ctx, request)
if err != nil {
return err
}
entry = resp.Entry
return nil
})
if entry != nil {
if entry.IsDirectory {
2018-05-06 06:50:34 +00:00
node = &Dir{Path: path.Join(dir.Path, name), wfs: dir.wfs}
2018-05-08 08:59:43 +00:00
} else {
node = &File{Chunks: entry.Chunks, Name: name, dir: dir, wfs: dir.wfs}
2018-05-06 05:47:16 +00:00
}
2018-05-06 06:50:34 +00:00
dir.NodeMap[name] = node
return node, nil
2018-05-06 05:47:16 +00:00
}
return nil, fuse.ENOENT
2018-05-06 05:47:16 +00:00
}
2018-05-08 08:59:43 +00:00
func (dir *Dir) ReadDirAll(ctx context.Context) (ret []fuse.Dirent, err error) {
2018-05-10 06:18:02 +00:00
err = dir.wfs.withFilerClient(func(client filer_pb.SeaweedFilerClient) error {
2018-05-08 08:59:43 +00:00
2018-05-10 06:18:02 +00:00
request := &filer_pb.ListEntriesRequest{
2018-05-08 08:59:43 +00:00
Directory: dir.Path,
2018-05-06 05:47:16 +00:00
}
2018-05-08 08:59:43 +00:00
glog.V(1).Infof("read directory: %v", request)
resp, err := client.ListEntries(ctx, request)
if err != nil {
return err
2018-05-06 05:47:16 +00:00
}
2018-05-08 08:59:43 +00:00
for _, entry := range resp.Entries {
if entry.IsDirectory {
dirent := fuse.Dirent{Name: entry.Name, Type: fuse.DT_Dir}
ret = append(ret, dirent)
} else {
dirent := fuse.Dirent{Name: entry.Name, Type: fuse.DT_File}
ret = append(ret, dirent)
dir.wfs.listDirectoryEntriesCache.Set(dir.Path+"/"+entry.Name, entry.Attributes, 3*time.Second)
2018-05-08 08:59:43 +00:00
}
}
return nil
})
return ret, err
2018-05-06 05:47:16 +00:00
}
func (dir *Dir) Remove(ctx context.Context, req *fuse.RemoveRequest) error {
2018-05-06 06:50:34 +00:00
dir.NodeMapLock.Lock()
defer dir.NodeMapLock.Unlock()
2018-05-10 06:18:02 +00:00
return dir.wfs.withFilerClient(func(client filer_pb.SeaweedFilerClient) error {
2018-05-08 08:59:43 +00:00
2018-05-10 06:18:02 +00:00
request := &filer_pb.DeleteEntryRequest{
2018-05-08 08:59:43 +00:00
Directory: dir.Path,
Name: req.Name,
IsDirectory: req.Dir,
}
glog.V(1).Infof("remove directory entry: %v", request)
_, err := client.DeleteEntry(ctx, request)
if err != nil {
return err
}
2018-05-06 06:50:34 +00:00
delete(dir.NodeMap, req.Name)
2018-05-08 08:59:43 +00:00
return nil
})
2018-05-06 05:47:16 +00:00
}