forked from TrueCloudLab/frostfs-node
51 lines
1.5 KiB
Go
51 lines
1.5 KiB
Go
package control
|
|
|
|
import (
|
|
"context"
|
|
|
|
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/services/control"
|
|
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/services/tree"
|
|
cid "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/container/id"
|
|
netmapSDK "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/netmap"
|
|
"google.golang.org/grpc/codes"
|
|
"google.golang.org/grpc/status"
|
|
)
|
|
|
|
// TreeService represents a tree service instance.
|
|
type TreeService interface {
|
|
SynchronizeTree(ctx context.Context, cnr cid.ID, treeID string) error
|
|
ReplicateTreeOp(ctx context.Context, n netmapSDK.NodeInfo, req *tree.ApplyRequest) 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.SynchronizeTree(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
|
|
}
|