forked from TrueCloudLab/frostfs-node
Dmitrii Stepanov
e4889e06ba
Now it's possible to run evacuate shard in async. Also only one evacuate process can be in progress. Signed-off-by: Dmitrii Stepanov <d.stepanov@yadro.com>
97 lines
2.7 KiB
Go
97 lines
2.7 KiB
Go
package control
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
|
|
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/engine"
|
|
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/util/logicerr"
|
|
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/services/control"
|
|
"google.golang.org/grpc/codes"
|
|
"google.golang.org/grpc/status"
|
|
)
|
|
|
|
func (s *Server) StartShardEvacuation(ctx context.Context, req *control.StartShardEvacuationRequest) (*control.StartShardEvacuationResponse, error) {
|
|
err := s.isValidRequest(req)
|
|
if err != nil {
|
|
return nil, status.Error(codes.PermissionDenied, err.Error())
|
|
}
|
|
|
|
var prm engine.EvacuateShardPrm
|
|
prm.WithShardIDList(s.getShardIDList(req.GetBody().GetShard_ID()))
|
|
prm.WithIgnoreErrors(req.GetBody().GetIgnoreErrors())
|
|
prm.WithFaultHandler(s.replicate)
|
|
prm.WithAsync(true)
|
|
|
|
_, err = s.s.Evacuate(ctx, prm)
|
|
if err != nil {
|
|
var logicalErr logicerr.Logical
|
|
if errors.As(err, &logicalErr) {
|
|
return nil, status.Error(codes.Aborted, err.Error())
|
|
}
|
|
return nil, status.Error(codes.Internal, err.Error())
|
|
}
|
|
|
|
resp := &control.StartShardEvacuationResponse{
|
|
Body: &control.StartShardEvacuationResponse_Body{},
|
|
}
|
|
|
|
err = SignMessage(s.key, resp)
|
|
if err != nil {
|
|
return nil, status.Error(codes.Internal, err.Error())
|
|
}
|
|
return resp, nil
|
|
}
|
|
|
|
func (s *Server) GetShardEvacuationStatus(ctx context.Context, req *control.GetShardEvacuationStatusRequest) (*control.GetShardEvacuationStatusResponse, error) {
|
|
err := s.isValidRequest(req)
|
|
if err != nil {
|
|
return nil, status.Error(codes.PermissionDenied, err.Error())
|
|
}
|
|
|
|
state, err := s.s.GetEvacuationState(ctx)
|
|
if err != nil {
|
|
var logicalErr logicerr.Logical
|
|
if errors.As(err, &logicalErr) {
|
|
return nil, status.Error(codes.Aborted, err.Error())
|
|
}
|
|
return nil, status.Error(codes.Internal, err.Error())
|
|
}
|
|
|
|
resp, err := stateToResponse(state)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
err = SignMessage(s.key, resp)
|
|
if err != nil {
|
|
return nil, status.Error(codes.Internal, err.Error())
|
|
}
|
|
return resp, nil
|
|
}
|
|
|
|
func (s *Server) StopShardEvacuation(ctx context.Context, req *control.StopShardEvacuationRequest) (*control.StopShardEvacuationResponse, error) {
|
|
err := s.isValidRequest(req)
|
|
if err != nil {
|
|
return nil, status.Error(codes.PermissionDenied, err.Error())
|
|
}
|
|
|
|
err = s.s.EnqueRunningEvacuationStop(ctx)
|
|
if err != nil {
|
|
var logicalErr logicerr.Logical
|
|
if errors.As(err, &logicalErr) {
|
|
return nil, status.Error(codes.Aborted, err.Error())
|
|
}
|
|
return nil, status.Error(codes.Internal, err.Error())
|
|
}
|
|
|
|
resp := &control.StopShardEvacuationResponse{
|
|
Body: &control.StopShardEvacuationResponse_Body{},
|
|
}
|
|
|
|
err = SignMessage(s.key, resp)
|
|
if err != nil {
|
|
return nil, status.Error(codes.Internal, err.Error())
|
|
}
|
|
return resp, nil
|
|
}
|