frostfs-node/pkg/services/object/search/local.go

package searchsvc

import (
	"context"

	"github.com/nspcc-dev/neofs-api-go/pkg/container"
	objectSDK "github.com/nspcc-dev/neofs-api-go/pkg/object"
	"github.com/nspcc-dev/neofs-node/pkg/local_object_storage/localstore"
	"github.com/nspcc-dev/neofs-node/pkg/services/object/search/query"
	"github.com/pkg/errors"
)

type localStream struct {
	query query.Query

	storage *localstore.Storage

	cid *container.ID
}

func (s *localStream) stream(ctx context.Context, ch chan<- []*objectSDK.ID) error {
	fs := s.query.ToSearchFilters()
	fs.AddObjectContainerIDFilter(objectSDK.MatchStringEqual, s.cid)

	addrList, err := s.storage.Select(fs)
	if err != nil {
		return errors.Wrapf(err, "(%T) could not select objects from local storage", s)
	}

	idList := make([]*objectSDK.ID, 0, len(addrList))

	for i := range addrList {
		idList = append(idList, addrList[i].ObjectID())
	}

	select {
	case <-ctx.Done():
		return ctx.Err()
	case ch <- idList:
		return nil
	}
}