Initial commit

Initial public review release v0.10.0

lib/implementations/locator.go

@@ -0,0 +1,78 @@
+package implementations
+
+import (
+	"context"
+
+	"github.com/multiformats/go-multiaddr"
+	"github.com/nspcc-dev/neofs-api-go/query"
+	"github.com/nspcc-dev/neofs-api-go/refs"
+	"github.com/nspcc-dev/neofs-api-go/service"
+	"github.com/nspcc-dev/neofs-node/lib/replication"
+	"github.com/nspcc-dev/neofs-node/lib/transport"
+	"github.com/pkg/errors"
+	"go.uber.org/zap"
+)
+
+type (
+	locator struct {
+		executor SelectiveContainerExecutor
+		log      *zap.Logger
+	}
+
+	// LocatorParams groups the parameters of ObjectLocator constructor.
+	LocatorParams struct {
+		SelectiveContainerExecutor SelectiveContainerExecutor
+		Logger                     *zap.Logger
+	}
+)
+
+const locatorInstanceFailMsg = "could not create object locator"
+
+var errEmptyObjectsContainerHandler = errors.New("empty container objects container handler")
+
+func (s *locator) LocateObject(ctx context.Context, addr Address) (res []multiaddr.Multiaddr, err error) {
+	queryBytes, err := (&query.Query{
+		Filters: []query.Filter{
+			{
+				Type:  query.Filter_Exact,
+				Name:  transport.KeyID,
+				Value: addr.ObjectID.String(),
+			},
+		},
+	}).Marshal()
+	if err != nil {
+		return nil, errors.Wrap(err, "locate object failed on query marshal")
+	}
+
+	err = s.executor.Search(ctx, &SearchParams{
+		SelectiveParams: SelectiveParams{
+			CID:    addr.CID,
+			TTL:    service.NonForwardingTTL,
+			IDList: make([]ObjectID, 1),
+		},
+		SearchCID:   addr.CID,
+		SearchQuery: queryBytes,
+		Handler: func(node multiaddr.Multiaddr, addrList []refs.Address) {
+			if len(addrList) > 0 {
+				res = append(res, node)
+			}
+		},
+	})
+
+	return
+}
+
+// NewObjectLocator constructs replication.ObjectLocator from SelectiveContainerExecutor.
+func NewObjectLocator(p LocatorParams) (replication.ObjectLocator, error) {
+	switch {
+	case p.SelectiveContainerExecutor == nil:
+		return nil, errors.Wrap(errEmptyObjectsContainerHandler, locatorInstanceFailMsg)
+	case p.Logger == nil:
+		return nil, errors.Wrap(errEmptyLogger, locatorInstanceFailMsg)
+	}
+
+	return &locator{
+		executor: p.SelectiveContainerExecutor,
+		log:      p.Logger,
+	}, nil
+}