frostfs-node/pkg/services/object/get/v2/service.go

package getsvc

import (
	"context"
	"errors"

	clientcore "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/core/client"
	"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/core/container"
	"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/core/netmap"
	objectSvc "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/services/object"
	getsvc "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/services/object/get"
	objutil "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/services/object/util"
	"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/util/logger"
	objectV2 "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/api/object"
	objectSDK "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/object"
	"go.uber.org/zap"
)

// Service implements Get operation of Object service v2.
type Service struct {
	*cfg
}

// Option represents Service constructor option.
type Option func(*cfg)

type clientSource interface {
	Get(info clientcore.NodeInfo) (clientcore.MultiAddressClient, error)
}

type cfg struct {
	svc *getsvc.Service

	keyStorage *objutil.KeyStorage

	clientSource clientSource

	netmapSource netmap.Source

	announcedKeys netmap.AnnouncedKeys

	contSource container.Source

	log *logger.Logger
}

// NewService constructs Service instance from provided options.
func NewService(svc *getsvc.Service,
	keyStorage *objutil.KeyStorage,
	clientSource clientSource,
	netmapSource netmap.Source,
	announcedKeys netmap.AnnouncedKeys,
	contSource container.Source,
	opts ...Option,
) *Service {
	c := &cfg{
		svc:           svc,
		keyStorage:    keyStorage,
		clientSource:  clientSource,
		netmapSource:  netmapSource,
		announcedKeys: announcedKeys,
		contSource:    contSource,
		log:           &logger.Logger{Logger: zap.L()},
	}

	for i := range opts {
		opts[i](c)
	}

	return &Service{
		cfg: c,
	}
}

// Get calls internal service and returns v2 object stream.
func (s *Service) Get(req *objectV2.GetRequest, stream objectSvc.GetObjectStream) error {
	p, err := s.toPrm(req, stream)
	if err != nil {
		return err
	}

	err = s.svc.Get(stream.Context(), *p)

	var splitErr *objectSDK.SplitInfoError
	var ecErr *objectSDK.ECInfoError

	switch {
	case errors.As(err, &splitErr):
		return stream.Send(splitInfoResponse(splitErr.SplitInfo()))
	case errors.As(err, &ecErr):
		return stream.Send(ecInfoResponse(ecErr.ECInfo()))
	default:
		return err
	}
}

// GetRange calls internal service and returns v2 payload range stream.
func (s *Service) GetRange(req *objectV2.GetRangeRequest, stream objectSvc.GetObjectRangeStream) error {
	p, err := s.toRangePrm(req, stream)
	if err != nil {
		return err
	}

	err = s.svc.GetRange(stream.Context(), *p)

	var splitErr *objectSDK.SplitInfoError
	var ecErr *objectSDK.ECInfoError

	switch {
	case errors.As(err, &splitErr):
		return stream.Send(splitInfoRangeResponse(splitErr.SplitInfo()))
	case errors.As(err, &ecErr):
		return stream.Send(ecInfoRangeResponse(ecErr.ECInfo()))
	default:
		return err
	}
}

// Head serves ForstFS API v2 compatible HEAD requests.
func (s *Service) Head(ctx context.Context, req *objectV2.HeadRequest) (*objectV2.HeadResponse, error) {
	resp := new(objectV2.HeadResponse)
	resp.SetBody(new(objectV2.HeadResponseBody))

	p, err := s.toHeadPrm(req, resp)
	if err != nil {
		return nil, err
	}

	err = s.svc.Head(ctx, *p)

	var splitErr *objectSDK.SplitInfoError
	var ecErr *objectSDK.ECInfoError

	if errors.As(err, &splitErr) {
		setSplitInfoHeadResponse(splitErr.SplitInfo(), resp)
		err = nil
	}
	if errors.As(err, &ecErr) {
		setECInfoHeadResponse(ecErr.ECInfo(), resp)
		err = nil
	}

	return resp, err
}

func WithLogger(l *logger.Logger) Option {
	return func(c *cfg) {
		c.log = &logger.Logger{Logger: l.With(zap.String("component", "Object.Get V2 service"))}
	}
}