frostfs-node/pkg/local_object_storage/shard/select.go

package shard

import (
	"fmt"

	meta "github.com/TrueCloudLab/frostfs-node/pkg/local_object_storage/metabase"
	cid "github.com/TrueCloudLab/frostfs-sdk-go/container/id"
	"github.com/TrueCloudLab/frostfs-sdk-go/object"
	oid "github.com/TrueCloudLab/frostfs-sdk-go/object/id"
)

// SelectPrm groups the parameters of Select operation.
type SelectPrm struct {
	cnr     cid.ID
	filters object.SearchFilters
}

// SelectRes groups the resulting values of Select operation.
type SelectRes struct {
	addrList []oid.Address
}

// SetContainerID is a Select option to set the container id to search in.
func (p *SelectPrm) SetContainerID(cnr cid.ID) {
	p.cnr = cnr
}

// SetFilters is a Select option to set the object filters.
func (p *SelectPrm) SetFilters(fs object.SearchFilters) {
	p.filters = fs
}

// AddressList returns list of addresses of the selected objects.
func (r SelectRes) AddressList() []oid.Address {
	return r.addrList
}

// Select selects the objects from shard that match select parameters.
//
// Returns any error encountered that
// did not allow to completely select the objects.
func (s *Shard) Select(prm SelectPrm) (SelectRes, error) {
	s.m.RLock()
	defer s.m.RUnlock()

	if s.info.Mode.NoMetabase() {
		return SelectRes{}, ErrDegradedMode
	}

	var selectPrm meta.SelectPrm
	selectPrm.SetFilters(prm.filters)
	selectPrm.SetContainerID(prm.cnr)

	mRes, err := s.metaBase.Select(selectPrm)
	if err != nil {
		return SelectRes{}, fmt.Errorf("could not select objects from metabase: %w", err)
	}

	return SelectRes{
		addrList: mRes.AddressList(),
	}, nil
}