package shell import ( "flag" "fmt" "github.com/chrislusf/seaweedfs/weed/pb" "io" "github.com/chrislusf/seaweedfs/weed/storage/needle" ) func init() { Commands = append(Commands, &commandVolumeMark{}) } type commandVolumeMark struct { } func (c *commandVolumeMark) Name() string { return "volume.mark" } func (c *commandVolumeMark) Help() string { return `Mark volume writable or readonly from one volume server volume.mark -node -volumeId -writable or -readonly ` } func (c *commandVolumeMark) Do(args []string, commandEnv *CommandEnv, writer io.Writer) (err error) { if err = commandEnv.confirmIsLocked(); err != nil { return } volMarkCommand := flag.NewFlagSet(c.Name(), flag.ContinueOnError) volumeIdInt := volMarkCommand.Int("volumeId", 0, "the volume id") nodeStr := volMarkCommand.String("node", "", "the volume server :") writable := volMarkCommand.Bool("writable", false, "volume mark writable") readonly := volMarkCommand.Bool("readonly", false, "volume mark readonly") if err = volMarkCommand.Parse(args); err != nil { return nil } markWritable := false if (*writable && *readonly) || (!*writable && !*readonly) { return fmt.Errorf("use -readonly or -writable") } else if *writable { markWritable = true } sourceVolumeServer := pb.ServerAddress(*nodeStr) volumeId := needle.VolumeId(*volumeIdInt) return markVolumeWritable(commandEnv.option.GrpcDialOption, volumeId, sourceVolumeServer, markWritable) }