Add fs rm options

This commit is contained in:
byunghwa.yun 2021-07-21 21:10:36 +09:00
parent b23b307e08
commit 41d559f54e

View file

@ -4,6 +4,7 @@ import (
"context" "context"
"fmt" "fmt"
"io" "io"
"strings"
"github.com/chrislusf/seaweedfs/weed/pb/filer_pb" "github.com/chrislusf/seaweedfs/weed/pb/filer_pb"
"github.com/chrislusf/seaweedfs/weed/util" "github.com/chrislusf/seaweedfs/weed/util"
@ -21,47 +22,79 @@ func (c *commandFsRm) Name() string {
} }
func (c *commandFsRm) Help() string { func (c *commandFsRm) Help() string {
return `remove a file or a folder, recursively delete all files and folders return `remove file and directory entries
fs.rm <entry1> fs.rm [-rf] <entry1> <entry2> ...
fs.rm /dir/file_name fs.rm /dir/file_name1 dir/file_name2
fs.rm /dir fs.rm /dir
The option "-r" can be recursive.
The option "-f" can be ignored by recursive error.
` `
} }
func (c *commandFsRm) Do(args []string, commandEnv *CommandEnv, writer io.Writer) (err error) { func (c *commandFsRm) Do(args []string, commandEnv *CommandEnv, writer io.Writer) (err error) {
isRecursive := false
if len(args) != 1 { ignoreRecursiveError := false
var entiries []string
for _, arg := range args {
if !strings.HasPrefix(arg, "-") {
entiries = append(entiries, arg)
continue
}
for _, t := range arg {
switch t {
case 'r':
isRecursive = true
case 'f':
ignoreRecursiveError = true
}
}
}
if len(entiries) < 1 {
return fmt.Errorf("need to have arguments") return fmt.Errorf("need to have arguments")
} }
targetPath, err := commandEnv.parseUrl(args[0]) commandEnv.WithFilerClient(func(client filer_pb.SeaweedFilerClient) error {
if err != nil { for _, entry := range entiries {
return err targetPath, err := commandEnv.parseUrl(entry)
} if err != nil {
fmt.Fprintf(writer, "rm: %s: %v\n", targetPath, err)
continue
}
targetDir, targetName := util.FullPath(targetPath).DirAndName() targetDir, targetName := util.FullPath(targetPath).DirAndName()
return commandEnv.WithFilerClient(func(client filer_pb.SeaweedFilerClient) error { lookupRequest := &filer_pb.LookupDirectoryEntryRequest{
Directory: targetDir,
Name: targetName,
}
_, err = filer_pb.LookupEntry(client, lookupRequest)
if err != nil {
fmt.Fprintf(writer, "rm: %s: %v\n", targetPath, err)
continue
}
request := &filer_pb.DeleteEntryRequest{ request := &filer_pb.DeleteEntryRequest{
Directory: targetDir, Directory: targetDir,
Name: targetName, Name: targetName,
IgnoreRecursiveError: true, IgnoreRecursiveError: ignoreRecursiveError,
IsDeleteData: true, IsDeleteData: true,
IsRecursive: true, IsRecursive: isRecursive,
IsFromOtherCluster: false, IsFromOtherCluster: false,
Signatures: nil, Signatures: nil,
}
if resp, err := client.DeleteEntry(context.Background(), request); err != nil {
fmt.Fprintf(writer, "rm: %s: %v\n", targetPath, err)
} else {
if resp.Error != "" {
fmt.Fprintf(writer, "rm: %s: %v\n", targetPath, resp.Error)
}
}
} }
_, err = client.DeleteEntry(context.Background(), request) return nil
if err == nil {
fmt.Fprintf(writer, "remove: %s\n", targetPath)
}
return err
}) })
return
} }