68 lines
1.4 KiB
Go
68 lines
1.4 KiB
Go
package swadmin
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"regexp"
|
|
"strings"
|
|
|
|
"github.com/chrislusf/seaweedfs/weed/shell"
|
|
"google.golang.org/grpc"
|
|
)
|
|
|
|
type SeaweedAdmin struct {
|
|
commandReg *regexp.Regexp
|
|
commandEnv *shell.CommandEnv
|
|
Output io.Writer
|
|
}
|
|
|
|
func NewSeaweedAdmin(masters string, output io.Writer) *SeaweedAdmin {
|
|
var shellOptions shell.ShellOptions
|
|
shellOptions.GrpcDialOption = grpc.WithInsecure()
|
|
shellOptions.Masters = &masters
|
|
|
|
commandEnv := shell.NewCommandEnv(shellOptions)
|
|
reg, _ := regexp.Compile(`'.*?'|".*?"|\S+`)
|
|
|
|
go commandEnv.MasterClient.KeepConnectedToMaster()
|
|
|
|
return &SeaweedAdmin{
|
|
commandEnv: commandEnv,
|
|
commandReg: reg,
|
|
Output: output,
|
|
}
|
|
}
|
|
|
|
// ProcessCommands cmds can be semi-colon separated commands
|
|
func (sa *SeaweedAdmin) ProcessCommands(cmds string) error {
|
|
for _, c := range strings.Split(cmds, ";") {
|
|
if err := sa.ProcessCommand(c); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (sa *SeaweedAdmin) ProcessCommand(cmd string) error {
|
|
sa.commandEnv.MasterClient.WaitUntilConnected()
|
|
cmds := sa.commandReg.FindAllString(cmd, -1)
|
|
if len(cmds) == 0 {
|
|
return nil
|
|
}
|
|
|
|
args := make([]string, len(cmds[1:]))
|
|
|
|
for i := range args {
|
|
args[i] = strings.Trim(string(cmds[1+i]), "\"'")
|
|
}
|
|
|
|
for _, c := range shell.Commands {
|
|
if c.Name() == cmd || c.Name() == "fs."+cmd {
|
|
return c.Do(args, sa.commandEnv, sa.Output)
|
|
}
|
|
}
|
|
|
|
return fmt.Errorf("unknown command: %v", cmd)
|
|
|
|
}
|