frostfs-node/pkg/services/control/server/syncronize_tree.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
}