Add a version token on RPCs to read/update volume server states. (#8191)
* 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.
This commit is contained in:
@@ -273,7 +273,7 @@ func (vs *VolumeServer) VolumeStatus(ctx context.Context, req *volume_server_pb.
|
||||
func (vs *VolumeServer) VolumeServerStatus(ctx context.Context, req *volume_server_pb.VolumeServerStatusRequest) (*volume_server_pb.VolumeServerStatusResponse, error) {
|
||||
|
||||
resp := &volume_server_pb.VolumeServerStatusResponse{
|
||||
State: vs.store.State.Pb,
|
||||
State: vs.store.State.Proto(),
|
||||
MemoryStatus: stats.MemStat(),
|
||||
Version: version.Version(),
|
||||
DataCenter: vs.dataCenter,
|
||||
|
||||
@@ -9,7 +9,7 @@ import (
|
||||
// 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.Pb,
|
||||
State: vs.store.State.Proto(),
|
||||
}
|
||||
|
||||
return resp, nil
|
||||
@@ -17,9 +17,9 @@ func (vs *VolumeServer) GetState(ctx context.Context, req *volume_server_pb.GetS
|
||||
|
||||
// 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.State)
|
||||
err := vs.store.State.Update(req.GetState())
|
||||
resp := &volume_server_pb.SetStateResponse{
|
||||
State: vs.store.State.Pb,
|
||||
State: vs.store.State.Proto(),
|
||||
}
|
||||
|
||||
return resp, err
|
||||
|
||||
@@ -178,7 +178,7 @@ func (vs *VolumeServer) MaintenanceMode() bool {
|
||||
if vs.store == nil {
|
||||
return false
|
||||
}
|
||||
return vs.store.State.Pb.GetMaintenance()
|
||||
return vs.store.State.Proto().GetMaintenance()
|
||||
}
|
||||
|
||||
// Checks if a volume server is in maintenance mode, and returns an error explaining why.
|
||||
|
||||
@@ -42,11 +42,8 @@ func TestMaintenanceMode(t *testing.T) {
|
||||
t.Run(tc.name, func(t *testing.T) {
|
||||
vs := VolumeServer{
|
||||
store: &storage.Store{
|
||||
Id: "test_1234",
|
||||
State: &storage.State{
|
||||
FilePath: "/some/path.pb",
|
||||
Pb: tc.pb,
|
||||
},
|
||||
Id: "test_1234",
|
||||
State: storage.NewStateFromProto("/some/path.pb", tc.pb),
|
||||
},
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user