feat(getters/shrex): enable endpoints in shrex

Author: vgonkivs

nodebuilder/tests/share_test.go

@@ -1,5 +1,3 @@
-//go:build share || integration
-
 package tests
 
 import (

share/shwap/accessor.go

@@ -5,6 +5,7 @@ import (
 	"io"
 
 	libshare "github.com/celestiaorg/go-square/v2/share"
+	"github.com/celestiaorg/rsmt2d"
 
 	"github.com/celestiaorg/celestia-node/share"
 )
@@ -13,5 +14,11 @@ import (
 type Accessor interface {
 	AxisRoots(context.Context) (*share.AxisRoots, error)
 	RowNamespaceData(context.Context, libshare.Namespace, int) (RowNamespaceData, error)
+	Sample(ctx context.Context, idx SampleCoords) (Sample, error)
+	AxisHalf(ctx context.Context, axisType rsmt2d.Axis, axisIdx int) (AxisHalf, error)
 	Reader() (io.Reader, error)
+	RangeNamespaceData(
+		ctx context.Context,
+		from, to int,
+	) (RangeNamespaceData, error)
 }

share/shwap/p2p/shrex/request.go

@@ -21,6 +21,15 @@ var registry = []newRequestID{
 	func() request {
 		return &shwap.EdsID{}
 	},
+	func() request {
+		return &shwap.SampleID{}
+	},
+	func() request {
+		return &shwap.RowID{}
+	},
+	func() request {
+		return &shwap.RangeNamespaceDataID{}
+	},
 }
 
 // request represents compatible generalised interface for requests.

share/shwap/p2p/shrex/shrex_getter/metrics.go

@@ -0,0 +1,70 @@
+package shrex_getter
+
+import (
+	"context"
+	"fmt"
+
+	"go.opentelemetry.io/otel/attribute"
+	"go.opentelemetry.io/otel/metric"
+
+	"github.com/celestiaorg/celestia-node/libs/utils"
+)
+
+type requestType int
+
+var (
+	edsRequest           requestType = 1
+	namespaceDataRequest requestType = 2
+	rowRequest           requestType = 3
+	samplesRequest       requestType = 4
+	rangeRequest         requestType = 5
+)
+
+type metrics struct {
+	requestAttempt metric.Int64Histogram
+}
+
+func (r requestType) String() string {
+	switch r {
+	case edsRequest:
+		return "edsRequest"
+	case namespaceDataRequest:
+		return "namespaceDataRequest"
+	case rowRequest:
+		return "rowRequest"
+	case samplesRequest:
+		return "samplesRequest"
+	case rangeRequest:
+		return "rangeRequest"
+	default:
+		panic(fmt.Sprintf("unknown shrex getter requestType: %v", int(r)))
+	}
+}
+
+func (sg *Getter) WithMetrics() error {
+	requestAttempt, err := meter.Int64Histogram(
+		"getters_shrex_attempts_per_request",
+		metric.WithDescription("Number of attempts per shrex request"),
+	)
+	if err != nil {
+		return err
+	}
+
+	sg.metrics = &metrics{
+		requestAttempt: requestAttempt,
+	}
+	return nil
+}
+
+func (m *metrics) recordAttempts(ctx context.Context, requestType requestType, attempt int, success bool) {
+	if m == nil {
+		return
+	}
+	ctx = utils.ResetContextOnError(ctx)
+
+	m.requestAttempt.Record(ctx, int64(attempt),
+		metric.WithAttributes(
+			attribute.String("request_type", requestType.String()),
+			attribute.Bool("success", success)),
+	)
+}