frostfs-node/pkg/local_object_storage/shard/select.go
Alex Vanin f585772746 [] shard: Support ContainerID argument in metabase selects
Signed-off-by: Alex Vanin <alexey@nspcc.ru>
2020-12-11 17:19:37 +03:00

57 lines
1.4 KiB
Go

package shard
import (
"github.com/nspcc-dev/neofs-api-go/pkg/container"
objectSDK "github.com/nspcc-dev/neofs-api-go/pkg/object"
meta "github.com/nspcc-dev/neofs-node/pkg/local_object_storage/metabase"
"github.com/pkg/errors"
)
// SelectPrm groups the parameters of Select operation.
type SelectPrm struct {
cid *container.ID
filters objectSDK.SearchFilters
}
// SelectRes groups resulting values of Select operation.
type SelectRes struct {
addrList []*objectSDK.Address
}
// WithContainerID is a Select option to set the container id to search in.
func (p *SelectPrm) WithContainerID(cid *container.ID) *SelectPrm {
if p != nil {
p.cid = cid
}
return p
}
// WithFilters is a Select option to set the object filters.
func (p *SelectPrm) WithFilters(fs objectSDK.SearchFilters) *SelectPrm {
if p != nil {
p.filters = fs
}
return p
}
// AddressList returns list of addresses of the selected objects.
func (r *SelectRes) AddressList() []*objectSDK.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) {
addrList, err := meta.Select(s.metaBase, prm.cid, prm.filters)
if err != nil {
return nil, errors.Wrap(err, "could not select objects from metabase")
}
return &SelectRes{
addrList: addrList,
}, nil
}