diff options
Diffstat (limited to 'weed/filer/embedded_filer/filer_embedded.go')
| -rw-r--r-- | weed/filer/embedded_filer/filer_embedded.go | 141 |
1 files changed, 141 insertions, 0 deletions
diff --git a/weed/filer/embedded_filer/filer_embedded.go b/weed/filer/embedded_filer/filer_embedded.go new file mode 100644 index 000000000..27299eb40 --- /dev/null +++ b/weed/filer/embedded_filer/filer_embedded.go @@ -0,0 +1,141 @@ +package embedded_filer + +import ( + "errors" + "fmt" + "path/filepath" + "strings" + + "github.com/chrislusf/seaweedfs/weed/filer" + "github.com/chrislusf/seaweedfs/weed/operation" +) + +type FilerEmbedded struct { + master string + directories *DirectoryManagerInMap + files *FileListInLevelDb +} + +func NewFilerEmbedded(master string, dir string) (filer *FilerEmbedded, err error) { + dm, de := NewDirectoryManagerInMap(filepath.Join(dir, "dir.log")) + if de != nil { + return nil, de + } + fl, fe := NewFileListInLevelDb(dir) + if fe != nil { + return nil, fe + } + filer = &FilerEmbedded{ + master: master, + directories: dm, + files: fl, + } + return +} + +func (filer *FilerEmbedded) CreateFile(filePath string, fid string) (err error) { + dir, file := filepath.Split(filePath) + dirId, e := filer.directories.MakeDirectory(dir) + if e != nil { + return e + } + return filer.files.CreateFile(dirId, file, fid) +} +func (filer *FilerEmbedded) FindFile(filePath string) (fid string, err error) { + dir, file := filepath.Split(filePath) + dirId, e := filer.directories.FindDirectory(dir) + if e != nil { + return "", e + } + return filer.files.FindFile(dirId, file) +} +func (filer *FilerEmbedded) FindDirectory(dirPath string) (dirId filer.DirectoryId, err error) { + return filer.directories.FindDirectory(dirPath) +} +func (filer *FilerEmbedded) ListDirectories(dirPath string) (dirs []filer.DirectoryEntry, err error) { + return filer.directories.ListDirectories(dirPath) +} +func (filer *FilerEmbedded) ListFiles(dirPath string, lastFileName string, limit int) (files []filer.FileEntry, err error) { + dirId, e := filer.directories.FindDirectory(dirPath) + if e != nil { + return nil, e + } + return filer.files.ListFiles(dirId, lastFileName, limit), nil +} +func (filer *FilerEmbedded) DeleteDirectory(dirPath string, recursive bool) (err error) { + dirId, e := filer.directories.FindDirectory(dirPath) + if e != nil { + return e + } + if sub_dirs, sub_err := filer.directories.ListDirectories(dirPath); sub_err == nil { + if len(sub_dirs) > 0 && !recursive { + return fmt.Errorf("Fail to delete directory %s: %d sub directories found!", dirPath, len(sub_dirs)) + } + for _, sub := range sub_dirs { + if delete_sub_err := filer.DeleteDirectory(filepath.Join(dirPath, sub.Name), recursive); delete_sub_err != nil { + return delete_sub_err + } + } + } + list := filer.files.ListFiles(dirId, "", 100) + if len(list) != 0 && !recursive { + if !recursive { + return fmt.Errorf("Fail to delete non-empty directory %s!", dirPath) + } + } + for { + if len(list) == 0 { + return filer.directories.DeleteDirectory(dirPath) + } + var fids []string + for _, fileEntry := range list { + fids = append(fids, string(fileEntry.Id)) + } + if result_list, delete_file_err := operation.DeleteFiles(filer.master, fids); delete_file_err != nil { + return delete_file_err + } else { + if len(result_list.Errors) > 0 { + return errors.New(strings.Join(result_list.Errors, "\n")) + } + } + lastFile := list[len(list)-1] + list = filer.files.ListFiles(dirId, lastFile.Name, 100) + } + +} + +func (filer *FilerEmbedded) DeleteFile(filePath string) (fid string, err error) { + dir, file := filepath.Split(filePath) + dirId, e := filer.directories.FindDirectory(dir) + if e != nil { + return "", e + } + return filer.files.DeleteFile(dirId, file) +} + +/* +Move a folder or a file, with 4 Use cases: +mv fromDir toNewDir +mv fromDir toOldDir +mv fromFile toDir +mv fromFile toFile +*/ +func (filer *FilerEmbedded) Move(fromPath string, toPath string) error { + if _, dir_err := filer.FindDirectory(fromPath); dir_err == nil { + if _, err := filer.FindDirectory(toPath); err == nil { + // move folder under an existing folder + return filer.directories.MoveUnderDirectory(fromPath, toPath, "") + } + // move folder to a new folder + return filer.directories.MoveUnderDirectory(fromPath, filepath.Dir(toPath), filepath.Base(toPath)) + } + if fid, file_err := filer.DeleteFile(fromPath); file_err == nil { + if _, err := filer.FindDirectory(toPath); err == nil { + // move file under an existing folder + return filer.CreateFile(filepath.Join(toPath, filepath.Base(fromPath)), fid) + } + // move to a folder with new name + return filer.CreateFile(toPath, fid) + } + return fmt.Errorf("File %s is not found!", fromPath) +} |
