mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2025-01-09 03:01:19 +00:00
55f311eb73
* WIP * Add REST implementation for Validator's SubscribeCommitteeSubnets * Remove redundant test * Initialize dutiesProvider * Remove duplicate import * Fix build break * Address PR comments * Remove file committed by mistake * Fix broken test Co-authored-by: james-prysm <90280386+james-prysm@users.noreply.github.com> Co-authored-by: Radosław Kapka <rkapka@wp.pl>
82 lines
3.2 KiB
Go
82 lines
3.2 KiB
Go
package beacon_api
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"encoding/json"
|
|
"strconv"
|
|
|
|
"github.com/pkg/errors"
|
|
"github.com/prysmaticlabs/prysm/v3/beacon-chain/rpc/apimiddleware"
|
|
types "github.com/prysmaticlabs/prysm/v3/consensus-types/primitives"
|
|
ethpb "github.com/prysmaticlabs/prysm/v3/proto/prysm/v1alpha1"
|
|
"github.com/prysmaticlabs/prysm/v3/time/slots"
|
|
)
|
|
|
|
func (c beaconApiValidatorClient) subscribeCommitteeSubnets(ctx context.Context, in *ethpb.CommitteeSubnetsSubscribeRequest, validatorIndices []types.ValidatorIndex) error {
|
|
if in == nil {
|
|
return errors.New("committee subnets subscribe request is nil")
|
|
}
|
|
|
|
if len(in.CommitteeIds) != len(in.Slots) || len(in.CommitteeIds) != len(in.IsAggregator) || len(in.CommitteeIds) != len(validatorIndices) {
|
|
return errors.New("arrays `in.CommitteeIds`, `in.Slots`, `in.IsAggregator` and `validatorIndices` don't have the same length")
|
|
}
|
|
|
|
slotToCommitteesAtSlotMap := make(map[types.Slot]uint64)
|
|
jsonCommitteeSubscriptions := make([]*apimiddleware.BeaconCommitteeSubscribeJson, len(in.CommitteeIds))
|
|
for index := range in.CommitteeIds {
|
|
subscribeSlot := in.Slots[index]
|
|
subscribeCommitteeId := in.CommitteeIds[index]
|
|
subscribeIsAggregator := in.IsAggregator[index]
|
|
subscribeValidatorIndex := validatorIndices[index]
|
|
|
|
committeesAtSlot, foundSlot := slotToCommitteesAtSlotMap[subscribeSlot]
|
|
if !foundSlot {
|
|
// Lazily fetch the committeesAtSlot from the beacon node if they are not already in the map
|
|
epoch := slots.ToEpoch(subscribeSlot)
|
|
duties, err := c.dutiesProvider.GetAttesterDuties(ctx, epoch, validatorIndices)
|
|
if err != nil {
|
|
return errors.Wrapf(err, "failed to get duties for epoch `%d`", epoch)
|
|
}
|
|
|
|
for _, duty := range duties {
|
|
dutySlot, err := strconv.ParseUint(duty.Slot, 10, 64)
|
|
if err != nil {
|
|
return errors.Wrapf(err, "failed to parse slot `%s`", duty.Slot)
|
|
}
|
|
|
|
committees, err := strconv.ParseUint(duty.CommitteesAtSlot, 10, 64)
|
|
if err != nil {
|
|
return errors.Wrapf(err, "failed to parse CommitteesAtSlot `%s`", duty.CommitteesAtSlot)
|
|
}
|
|
|
|
slotToCommitteesAtSlotMap[types.Slot(dutySlot)] = committees
|
|
}
|
|
|
|
// If the slot still isn't in the map, we either received bad data from the beacon node or the caller of this function gave us bad data
|
|
if committeesAtSlot, foundSlot = slotToCommitteesAtSlotMap[subscribeSlot]; !foundSlot {
|
|
return errors.Errorf("failed to get committees for slot `%d`", subscribeSlot)
|
|
}
|
|
}
|
|
|
|
jsonCommitteeSubscriptions[index] = &apimiddleware.BeaconCommitteeSubscribeJson{
|
|
CommitteeIndex: strconv.FormatUint(uint64(subscribeCommitteeId), 10),
|
|
CommitteesAtSlot: strconv.FormatUint(committeesAtSlot, 10),
|
|
Slot: strconv.FormatUint(uint64(subscribeSlot), 10),
|
|
IsAggregator: subscribeIsAggregator,
|
|
ValidatorIndex: strconv.FormatUint(uint64(subscribeValidatorIndex), 10),
|
|
}
|
|
}
|
|
|
|
committeeSubscriptionsBytes, err := json.Marshal(jsonCommitteeSubscriptions)
|
|
if err != nil {
|
|
return errors.Wrap(err, "failed to marshal committees subscriptions")
|
|
}
|
|
|
|
if _, err := c.jsonRestHandler.PostRestJson(ctx, "/eth/v1/validator/beacon_committee_subscriptions", nil, bytes.NewBuffer(committeeSubscriptionsBytes), nil); err != nil {
|
|
return errors.Wrap(err, "failed to send POST data to REST endpoint")
|
|
}
|
|
|
|
return nil
|
|
}
|