forked from TrueCloudLab/frostfs-node
Alex Vanin
20de74a505
Due to source code relocation from GitHub. Signed-off-by: Alex Vanin <a.vanin@yadro.com>
48 lines
1.2 KiB
Go
48 lines
1.2 KiB
Go
package control
|
|
|
|
import (
|
|
"context"
|
|
|
|
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/services/control"
|
|
cid "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/container/id"
|
|
"google.golang.org/grpc/codes"
|
|
"google.golang.org/grpc/status"
|
|
)
|
|
|
|
// TreeService represents a tree service instance.
|
|
type TreeService interface {
|
|
Synchronize(ctx context.Context, cnr cid.ID, treeID string) error
|
|
}
|
|
|
|
func (s *Server) SynchronizeTree(ctx context.Context, req *control.SynchronizeTreeRequest) (*control.SynchronizeTreeResponse, error) {
|
|
err := s.isValidRequest(req)
|
|
if err != nil {
|
|
return nil, status.Error(codes.PermissionDenied, err.Error())
|
|
}
|
|
|
|
if s.treeService == nil {
|
|
return nil, status.Error(codes.Internal, "tree service is disabled")
|
|
}
|
|
|
|
b := req.GetBody()
|
|
|
|
var cnr cid.ID
|
|
if err := cnr.Decode(b.GetContainerId()); err != nil {
|
|
return nil, status.Error(codes.Internal, err.Error())
|
|
}
|
|
|
|
err = s.treeService.Synchronize(ctx, cnr, b.GetTreeId())
|
|
if err != nil {
|
|
return nil, status.Error(codes.Internal, err.Error())
|
|
}
|
|
|
|
resp := new(control.SynchronizeTreeResponse)
|
|
resp.SetBody(new(control.SynchronizeTreeResponse_Body))
|
|
|
|
err = SignMessage(s.key, resp)
|
|
if err != nil {
|
|
return nil, status.Error(codes.Internal, err.Error())
|
|
}
|
|
|
|
return resp, nil
|
|
}
|