mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-25 21:07:18 +00:00
d17996f8b0
* Update V3 from V4 * Fix build v3 -> v4 * Update ssz * Update beacon_chain.pb.go * Fix formatter import * Update update-mockgen.sh comment to v4 * Fix conflicts. Pass build and tests * Fix test
62 lines
1.9 KiB
Go
62 lines
1.9 KiB
Go
package sync
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/pkg/errors"
|
|
"github.com/prysmaticlabs/prysm/v4/beacon-chain/cache"
|
|
"github.com/prysmaticlabs/prysm/v4/config/params"
|
|
"github.com/prysmaticlabs/prysm/v4/consensus-types/primitives"
|
|
"github.com/prysmaticlabs/prysm/v4/container/slice"
|
|
eth "github.com/prysmaticlabs/prysm/v4/proto/prysm/v1alpha1"
|
|
"github.com/prysmaticlabs/prysm/v4/time/slots"
|
|
"google.golang.org/protobuf/proto"
|
|
)
|
|
|
|
func (s *Service) committeeIndexBeaconAttestationSubscriber(_ context.Context, msg proto.Message) error {
|
|
a, ok := msg.(*eth.Attestation)
|
|
if !ok {
|
|
return fmt.Errorf("message was not type *eth.Attestation, type=%T", msg)
|
|
}
|
|
|
|
if a.Data == nil {
|
|
return errors.New("nil attestation")
|
|
}
|
|
s.setSeenCommitteeIndicesSlot(a.Data.Slot, a.Data.CommitteeIndex, a.AggregationBits)
|
|
|
|
exists, err := s.cfg.attPool.HasAggregatedAttestation(a)
|
|
if err != nil {
|
|
return errors.Wrap(err, "Could not determine if attestation pool has this atttestation")
|
|
}
|
|
if exists {
|
|
return nil
|
|
}
|
|
|
|
return s.cfg.attPool.SaveUnaggregatedAttestation(a)
|
|
}
|
|
|
|
func (_ *Service) persistentSubnetIndices() []uint64 {
|
|
return cache.SubnetIDs.GetAllSubnets()
|
|
}
|
|
|
|
func (_ *Service) aggregatorSubnetIndices(currentSlot primitives.Slot) []uint64 {
|
|
endEpoch := slots.ToEpoch(currentSlot) + 1
|
|
endSlot := params.BeaconConfig().SlotsPerEpoch.Mul(uint64(endEpoch))
|
|
var commIds []uint64
|
|
for i := currentSlot; i <= endSlot; i++ {
|
|
commIds = append(commIds, cache.SubnetIDs.GetAggregatorSubnetIDs(i)...)
|
|
}
|
|
return slice.SetUint64(commIds)
|
|
}
|
|
|
|
func (_ *Service) attesterSubnetIndices(currentSlot primitives.Slot) []uint64 {
|
|
endEpoch := slots.ToEpoch(currentSlot) + 1
|
|
endSlot := params.BeaconConfig().SlotsPerEpoch.Mul(uint64(endEpoch))
|
|
var commIds []uint64
|
|
for i := currentSlot; i <= endSlot; i++ {
|
|
commIds = append(commIds, cache.SubnetIDs.GetAttesterSubnetIDs(i)...)
|
|
}
|
|
return slice.SetUint64(commIds)
|
|
}
|