feat(beacon): add FetchBeaconBlockBlobs method

ethpandaops · Oct 10, 2023 · 0964cd7 · 0964cd7
1 parent 1019d16
commit 0964cd7
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/pkg/beacon/beacon.go b/pkg/beacon/beacon.go
@@ -11,6 +11,7 @@ import (
 	v1 "github.com/attestantio/go-eth2-client/api/v1"
 	"github.com/attestantio/go-eth2-client/spec"
 	"github.com/attestantio/go-eth2-client/spec/altair"
+	"github.com/attestantio/go-eth2-client/spec/deneb"
 	"github.com/attestantio/go-eth2-client/spec/phase0"
 	"github.com/chuckpreslar/emission"
 	"github.com/ethpandaops/beacon/pkg/beacon/api"
@@ -82,6 +83,8 @@ type Node interface {
 	FetchBeaconCommittees(ctx context.Context, state string, epoch phase0.Epoch) ([]*v1.BeaconCommittee, error)
 	// FetchAttestationData fetches the attestation data for the given slot and committee index.
 	FetchAttestationData(ctx context.Context, slot phase0.Slot, committeeIndex phase0.CommitteeIndex) (*phase0.AttestationData, error)
+	// FetchBeaconBlockBlobs fetches blob sidecars for the given block id.
+	FetchBeaconBlockBlobs(ctx context.Context, blockID string) ([]*deneb.BlobSidecar, error)
 
 	// Subscriptions
 	// - Proxied Beacon events

diff --git a/pkg/beacon/fetch.go b/pkg/beacon/fetch.go
@@ -7,6 +7,7 @@ import (
 	eth2client "github.com/attestantio/go-eth2-client"
 	v1 "github.com/attestantio/go-eth2-client/api/v1"
 	"github.com/attestantio/go-eth2-client/spec"
+	"github.com/attestantio/go-eth2-client/spec/deneb"
 	"github.com/attestantio/go-eth2-client/spec/phase0"
 	"github.com/ethpandaops/beacon/pkg/beacon/api/types"
 	"github.com/ethpandaops/beacon/pkg/beacon/state"
@@ -136,6 +137,20 @@ func (n *node) FetchSpec(ctx context.Context) (*state.Spec, error) {
 	return &sp, nil
 }
 
+func (n *node) FetchBeaconBlockBlobs(ctx context.Context, blockID string) ([]*deneb.BlobSidecar, error) {
+	provider, isProvider := n.client.(eth2client.BeaconBlockBlobsProvider)
+	if !isProvider {
+		return nil, errors.New("client does not implement eth2client.BeaconBlockBlobsProvider")
+	}
+
+	data, err := provider.BeaconBlockBlobs(ctx, blockID)
+	if err != nil {
+		return nil, err
+	}
+
+	return data, nil
+}
+
 func (n *node) FetchProposerDuties(ctx context.Context, epoch phase0.Epoch) ([]*v1.ProposerDuty, error) {
 	n.log.WithField("epoch", epoch).Debug("Fetching proposer duties")