package searchsvc import ( "crypto/ecdsa" "github.com/nspcc-dev/neofs-node/pkg/core/client" "github.com/nspcc-dev/neofs-node/pkg/core/netmap" "github.com/nspcc-dev/neofs-node/pkg/local_object_storage/engine" "github.com/nspcc-dev/neofs-node/pkg/services/object/util" "github.com/nspcc-dev/neofs-node/pkg/services/object_manager/placement" "github.com/nspcc-dev/neofs-node/pkg/util/logger" cid "github.com/nspcc-dev/neofs-sdk-go/container/id" oidSDK "github.com/nspcc-dev/neofs-sdk-go/object/id" "github.com/nspcc-dev/neofs-sdk-go/session" "go.uber.org/zap" ) // Service is an utility serving requests // of Object.Search service. type Service struct { *cfg } // Option is a Service's constructor option. type Option func(*cfg) type searchClient interface { searchObjects(*execCtx, client.NodeInfo) ([]oidSDK.ID, error) } type ClientConstructor interface { Get(client.NodeInfo) (client.MultiAddressClient, error) } type cfg struct { log *logger.Logger localStorage interface { search(*execCtx) ([]oidSDK.ID, error) } clientConstructor interface { get(client.NodeInfo) (searchClient, error) } traverserGenerator interface { generateTraverser(*cid.ID, uint64) (*placement.Traverser, error) } currentEpochReceiver interface { currentEpoch() (uint64, error) } keyStore interface { GetKey(token *session.Token) (*ecdsa.PrivateKey, error) } } func defaultCfg() *cfg { return &cfg{ log: zap.L(), clientConstructor: new(clientConstructorWrapper), } } // New creates, initializes and returns utility serving // Object.Get service requests. func New(opts ...Option) *Service { c := defaultCfg() for i := range opts { opts[i](c) } return &Service{ cfg: c, } } // WithLogger returns option to specify Get service's logger. func WithLogger(l *logger.Logger) Option { return func(c *cfg) { c.log = l.With(zap.String("component", "Object.Get service")) } } // WithLocalStorageEngine returns option to set local storage // instance. func WithLocalStorageEngine(e *engine.StorageEngine) Option { return func(c *cfg) { c.localStorage = (*storageEngineWrapper)(e) } } // WithClientConstructor returns option to set constructor of remote node clients. func WithClientConstructor(v ClientConstructor) Option { return func(c *cfg) { c.clientConstructor.(*clientConstructorWrapper).constructor = v } } // WithTraverserGenerator returns option to set generator of // placement traverser to get the objects from containers. func WithTraverserGenerator(t *util.TraverserGenerator) Option { return func(c *cfg) { c.traverserGenerator = (*traverseGeneratorWrapper)(t) } } // WithNetMapSource returns option to set network // map storage to receive current network state. func WithNetMapSource(nmSrc netmap.Source) Option { return func(c *cfg) { c.currentEpochReceiver = &nmSrcWrapper{ nmSrc: nmSrc, } } } // WithKeyStorage returns option to set private // key storage for session tokens and node key. func WithKeyStorage(store *util.KeyStorage) Option { return func(c *cfg) { c.keyStore = store } }