2cda4289f4
* Add a version token on `GetState()`/`SetState()` RPCs for volume server states. * Make state version a property ov `VolumeServerState` instead of an in-memory counter. Also extend state atomicity to reads, instead of just writes.
27 lines
736 B
Go
27 lines
736 B
Go
package weed_server
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/seaweedfs/seaweedfs/weed/pb/volume_server_pb"
|
|
)
|
|
|
|
// GetState returns a volume server's state flags.
|
|
func (vs *VolumeServer) GetState(ctx context.Context, req *volume_server_pb.GetStateRequest) (*volume_server_pb.GetStateResponse, error) {
|
|
resp := &volume_server_pb.GetStateResponse{
|
|
State: vs.store.State.Proto(),
|
|
}
|
|
|
|
return resp, nil
|
|
}
|
|
|
|
// SetState updates state flags for volume servers.
|
|
func (vs *VolumeServer) SetState(ctx context.Context, req *volume_server_pb.SetStateRequest) (*volume_server_pb.SetStateResponse, error) {
|
|
err := vs.store.State.Update(req.GetState())
|
|
resp := &volume_server_pb.SetStateResponse{
|
|
State: vs.store.State.Proto(),
|
|
}
|
|
|
|
return resp, err
|
|
}
|