seaweedfs/weed/shell/commands.go

139 lines
3.0 KiB
Go
Raw Normal View History

package shell
import (
"fmt"
"io"
"net/url"
"strconv"
"strings"
2019-12-13 16:22:37 +08:00
"google.golang.org/grpc"
2020-03-24 13:54:02 +08:00
"github.com/chrislusf/seaweedfs/weed/pb"
"github.com/chrislusf/seaweedfs/weed/pb/filer_pb"
2020-03-23 15:01:34 +08:00
"github.com/chrislusf/seaweedfs/weed/util"
"github.com/chrislusf/seaweedfs/weed/wdclient"
2020-05-26 15:03:31 +08:00
"github.com/chrislusf/seaweedfs/weed/wdclient/exclusive_locks"
)
type ShellOptions struct {
2019-10-25 22:45:12 +08:00
Masters *string
GrpcDialOption grpc.DialOption
// shell transient context
2021-08-05 03:30:18 +08:00
FilerHost string
FilerPort int64
FilerAddress pb.ServerAddress
2021-08-05 03:30:18 +08:00
Directory string
}
type CommandEnv struct {
env map[string]string
MasterClient *wdclient.MasterClient
option ShellOptions
2020-05-26 15:03:31 +08:00
locker *exclusive_locks.ExclusiveLocker
}
type command interface {
Name() string
Help() string
Do([]string, *CommandEnv, io.Writer) error
}
var (
Commands = []command{}
)
func NewCommandEnv(options ShellOptions) *CommandEnv {
2020-04-23 17:31:04 +08:00
ce := &CommandEnv{
2020-03-02 14:39:08 +08:00
env: make(map[string]string),
MasterClient: wdclient.NewMasterClient(options.GrpcDialOption, pb.AdminShellClient, "", "", pb.ServerAddresses(*options.Masters).ToAddresses()),
2020-03-02 14:39:08 +08:00
option: options,
}
2021-09-20 03:02:23 +08:00
ce.locker = exclusive_locks.NewExclusiveLocker(ce.MasterClient, "admin")
2020-04-23 17:31:04 +08:00
return ce
}
2020-03-24 12:26:15 +08:00
func (ce *CommandEnv) parseUrl(input string) (path string, err error) {
if strings.HasPrefix(input, "http") {
err = fmt.Errorf("http://<filer>:<port> prefix is not supported any more")
return
}
if !strings.HasPrefix(input, "/") {
2020-04-06 04:11:43 +08:00
input = util.Join(ce.option.Directory, input)
}
2020-03-24 12:26:15 +08:00
return input, err
}
2020-03-24 12:26:15 +08:00
func (ce *CommandEnv) isDirectory(path string) bool {
2020-03-24 12:26:15 +08:00
return ce.checkDirectory(path) == nil
}
2021-12-11 05:24:38 +08:00
func (ce *CommandEnv) confirmIsLocked(args []string) error {
2020-05-26 15:03:31 +08:00
if ce.locker.IsLocking() {
return nil
}
2021-12-11 05:24:38 +08:00
ce.locker.SetMessage(fmt.Sprintf("%v", args))
2021-08-06 12:06:55 +08:00
return fmt.Errorf("need to run \"lock\" first to continue")
}
2020-03-24 12:26:15 +08:00
func (ce *CommandEnv) checkDirectory(path string) error {
2020-03-23 15:01:34 +08:00
dir, name := util.FullPath(path).DirAndName()
2020-03-24 12:36:39 +08:00
exists, err := filer_pb.Exists(ce, dir, name, true)
if !exists {
return fmt.Errorf("%s is not a directory", path)
}
2020-03-24 12:26:15 +08:00
return err
}
2020-04-30 04:26:02 +08:00
var _ = filer_pb.FilerClient(&CommandEnv{})
2020-03-24 13:54:02 +08:00
func (ce *CommandEnv) WithFilerClient(fn func(filer_pb.SeaweedFilerClient) error) error {
return pb.WithGrpcFilerClient(ce.option.FilerAddress, ce.option.GrpcDialOption, fn)
2020-03-24 13:54:02 +08:00
}
2021-01-29 06:36:29 +08:00
func (ce *CommandEnv) AdjustedUrl(location *filer_pb.Location) string {
return location.Url
}
func parseFilerUrl(entryPath string) (filerServer string, filerPort int64, path string, err error) {
if strings.HasPrefix(entryPath, "http") {
var u *url.URL
u, err = url.Parse(entryPath)
if err != nil {
return
}
filerServer = u.Hostname()
portString := u.Port()
if portString != "" {
filerPort, err = strconv.ParseInt(portString, 10, 32)
}
path = u.Path
} else {
err = fmt.Errorf("path should have full url /path/to/dirOrFile : %s", entryPath)
}
return
}
func findInputDirectory(args []string) (input string) {
input = "."
if len(args) > 0 {
input = args[len(args)-1]
if strings.HasPrefix(input, "-") {
input = "."
}
}
return input
}