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

package searchsvc

import (
	"context"

	"github.com/nspcc-dev/neofs-api-go/pkg/client"
	"github.com/nspcc-dev/neofs-api-go/pkg/object"
	"github.com/nspcc-dev/neofs-node/pkg/network"
	"github.com/nspcc-dev/neofs-node/pkg/network/cache"
	"github.com/nspcc-dev/neofs-node/pkg/services/object/util"
	"github.com/pkg/errors"
)

type remoteStream struct {
	prm *Prm

	keyStorage *util.KeyStorage

	addr *network.Address

	clientCache *cache.ClientCache
}

func (s *remoteStream) stream(ctx context.Context, ch chan<- []*object.ID) error {
	key, err := s.keyStorage.GetKey(s.prm.common.SessionToken())
	if err != nil {
		return errors.Wrapf(err, "(%T) could not receive private key", s)
	}

	addr, err := s.addr.IPAddrString()
	if err != nil {
		return err
	}

	c, err := s.clientCache.Get(key, addr)
	if err != nil {
		return errors.Wrapf(err, "(%T) could not create SDK client %s", s, addr)
	}

	// TODO: add writer parameter to SDK client
	id, err := c.SearchObject(ctx, new(client.SearchObjectParams).
		WithContainerID(s.prm.cid).
		WithSearchFilters(s.prm.query.ToSearchFilters()),
		client.WithTTL(1), // FIXME: use constant
		client.WithSession(s.prm.common.SessionToken()),
		client.WithBearer(s.prm.common.BearerToken()),
	)
	if err != nil {
		return errors.Wrapf(err, "(%T) could not search objects in %s", s, addr)
	}

	ch <- id

	return nil
}