mirror of
https://github.com/seaweedfs/seaweedfs.git
synced 2024-01-19 02:48:24 +00:00
Add option to recursively delete a folder.
This commit is contained in:
parent
e378f9892d
commit
83c0c9843d
|
@ -14,6 +14,6 @@ type Filer interface {
|
||||||
FindFile(filePath string) (fid string, err error)
|
FindFile(filePath string) (fid string, err error)
|
||||||
ListDirectories(dirPath string) (dirs []DirectoryEntry, err error)
|
ListDirectories(dirPath string) (dirs []DirectoryEntry, err error)
|
||||||
ListFiles(dirPath string, lastFileName string, limit int) (files []FileEntry, err error)
|
ListFiles(dirPath string, lastFileName string, limit int) (files []FileEntry, err error)
|
||||||
DeleteDirectory(dirPath string) (err error)
|
DeleteDirectory(dirPath string, recursive bool) (err error)
|
||||||
DeleteFile(filePath string) (fid string, err error)
|
DeleteFile(filePath string) (fid string, err error)
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,16 +1,20 @@
|
||||||
package filer
|
package filer
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"code.google.com/p/weed-fs/go/operation"
|
||||||
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
|
"strings"
|
||||||
)
|
)
|
||||||
|
|
||||||
type FilerEmbedded struct {
|
type FilerEmbedded struct {
|
||||||
|
master string
|
||||||
directories *DirectoryManagerInMap
|
directories *DirectoryManagerInMap
|
||||||
files *FileListInLevelDb
|
files *FileListInLevelDb
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewFilerEmbedded(dir string) (filer *FilerEmbedded, err error) {
|
func NewFilerEmbedded(master string, dir string) (filer *FilerEmbedded, err error) {
|
||||||
dm, de := NewDirectoryManagerInMap(filepath.Join(dir, "dir.log"))
|
dm, de := NewDirectoryManagerInMap(filepath.Join(dir, "dir.log"))
|
||||||
if de != nil {
|
if de != nil {
|
||||||
return nil, de
|
return nil, de
|
||||||
|
@ -20,6 +24,7 @@ func NewFilerEmbedded(dir string) (filer *FilerEmbedded, err error) {
|
||||||
return nil, fe
|
return nil, fe
|
||||||
}
|
}
|
||||||
filer = &FilerEmbedded{
|
filer = &FilerEmbedded{
|
||||||
|
master: master,
|
||||||
directories: dm,
|
directories: dm,
|
||||||
files: fl,
|
files: fl,
|
||||||
}
|
}
|
||||||
|
@ -52,16 +57,47 @@ func (filer *FilerEmbedded) ListFiles(dirPath string, lastFileName string, limit
|
||||||
}
|
}
|
||||||
return filer.files.ListFiles(dirId, lastFileName, limit), nil
|
return filer.files.ListFiles(dirId, lastFileName, limit), nil
|
||||||
}
|
}
|
||||||
func (filer *FilerEmbedded) DeleteDirectory(dirPath string) (err error) {
|
func (filer *FilerEmbedded) DeleteDirectory(dirPath string, recursive bool) (err error) {
|
||||||
dirId, e := filer.directories.FindDirectory(dirPath)
|
dirId, e := filer.directories.FindDirectory(dirPath)
|
||||||
if e != nil {
|
if e != nil {
|
||||||
return e
|
return e
|
||||||
}
|
}
|
||||||
if len(filer.files.ListFiles(dirId, "", 1)) > 0 {
|
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)
|
return fmt.Errorf("Fail to delete non-empty directory %s!", dirPath)
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
for {
|
||||||
|
if len(list) == 0 {
|
||||||
return filer.directories.DeleteDirectory(dirPath)
|
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) {
|
func (filer *FilerEmbedded) DeleteFile(filePath string) (fid string, err error) {
|
||||||
dir, file := filepath.Split(filePath)
|
dir, file := filepath.Split(filePath)
|
||||||
dirId, e := filer.directories.FindDirectory(dir)
|
dirId, e := filer.directories.FindDirectory(dir)
|
||||||
|
|
|
@ -21,7 +21,7 @@ func NewFilerServer(r *http.ServeMux, port int, master string, dir string, colle
|
||||||
port: ":" + strconv.Itoa(port),
|
port: ":" + strconv.Itoa(port),
|
||||||
}
|
}
|
||||||
|
|
||||||
if fs.filer, err = filer.NewFilerEmbedded(dir); err != nil {
|
if fs.filer, err = filer.NewFilerEmbedded(master, dir); err != nil {
|
||||||
glog.Fatal("Can not start filer in dir", dir, ": ", err.Error())
|
glog.Fatal("Can not start filer in dir", dir, ": ", err.Error())
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
|
@ -171,11 +171,14 @@ func (fs *FilerServer) PostHandler(w http.ResponseWriter, r *http.Request) {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// curl -X DELETE http://localhost:8888/path/to
|
||||||
|
// curl -X DELETE http://localhost:8888/path/to?recursive=true
|
||||||
func (fs *FilerServer) DeleteHandler(w http.ResponseWriter, r *http.Request) {
|
func (fs *FilerServer) DeleteHandler(w http.ResponseWriter, r *http.Request) {
|
||||||
var err error
|
var err error
|
||||||
var fid string
|
var fid string
|
||||||
if strings.HasSuffix(r.URL.Path, "/") {
|
if strings.HasSuffix(r.URL.Path, "/") {
|
||||||
err = fs.filer.DeleteDirectory(r.URL.Path)
|
isRecursive := r.FormValue("recursive") == "true"
|
||||||
|
err = fs.filer.DeleteDirectory(r.URL.Path, isRecursive)
|
||||||
} else {
|
} else {
|
||||||
fid, err = fs.filer.DeleteFile(r.URL.Path)
|
fid, err = fs.filer.DeleteFile(r.URL.Path)
|
||||||
if err == nil {
|
if err == nil {
|
||||||
|
|
Loading…
Reference in a new issue