package shell import ( "context" "flag" "fmt" "github.com/seaweedfs/seaweedfs/weed/operation" "github.com/seaweedfs/seaweedfs/weed/pb" "github.com/seaweedfs/seaweedfs/weed/pb/volume_server_pb" "google.golang.org/grpc" "io" ) func init() { Commands = append(Commands, &commandVolumeServerLeave{}) } type commandVolumeServerLeave struct { } func (c *commandVolumeServerLeave) Name() string { return "volumeServer.leave" } func (c *commandVolumeServerLeave) Help() string { return `stop a volume server from sending heartbeats to the master volumeServer.leave -node -force This command enables gracefully shutting down the volume server. The volume server will stop sending heartbeats to the master. After draining the traffic for a few seconds, you can safely shut down the volume server. This operation is not revocable unless the volume server is restarted. ` } func (c *commandVolumeServerLeave) Do(args []string, commandEnv *CommandEnv, writer io.Writer) (err error) { vsLeaveCommand := flag.NewFlagSet(c.Name(), flag.ContinueOnError) volumeServer := vsLeaveCommand.String("node", "", ": of the volume server") if err = vsLeaveCommand.Parse(args); err != nil { return nil } if err = commandEnv.confirmIsLocked(args); err != nil { return } if *volumeServer == "" { return fmt.Errorf("need to specify volume server by -node=:") } return volumeServerLeave(commandEnv.option.GrpcDialOption, pb.ServerAddress(*volumeServer), writer) } func volumeServerLeave(grpcDialOption grpc.DialOption, volumeServer pb.ServerAddress, writer io.Writer) (err error) { return operation.WithVolumeServerClient(false, volumeServer, grpcDialOption, func(volumeServerClient volume_server_pb.VolumeServerClient) error { _, leaveErr := volumeServerClient.VolumeServerLeave(context.Background(), &volume_server_pb.VolumeServerLeaveRequest{}) if leaveErr != nil { fmt.Fprintf(writer, "ask volume server %s to leave: %v\n", volumeServer, leaveErr) } else { fmt.Fprintf(writer, "stopped heartbeat in volume server %s. After a few seconds to drain traffic, it will be safe to stop the volume server.\n", volumeServer) } return leaveErr }) }