mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-25 04:47:18 +00:00
84916672c6
* replace eth2 types * replace protos * regen proto * replace * gaz * deps * amend * regen proto * mod * gaz * gaz * ensure build * ssz * add dep * no more eth2 types * no more eth2 * remg * all builds * buidl * tidy * clean * fmt * val serv * gaz Co-authored-by: Preston Van Loon <preston@prysmaticlabs.com>
61 lines
2.2 KiB
Go
61 lines
2.2 KiB
Go
package p2p
|
|
|
|
import (
|
|
"reflect"
|
|
|
|
"github.com/prysmaticlabs/prysm/config/params"
|
|
types "github.com/prysmaticlabs/prysm/consensus-types/primitives"
|
|
ethpb "github.com/prysmaticlabs/prysm/proto/prysm/v1alpha1"
|
|
"google.golang.org/protobuf/proto"
|
|
)
|
|
|
|
// gossipTopicMappings represent the protocol ID to protobuf message type map for easy
|
|
// lookup.
|
|
var gossipTopicMappings = map[string]proto.Message{
|
|
BlockSubnetTopicFormat: ðpb.SignedBeaconBlock{},
|
|
AttestationSubnetTopicFormat: ðpb.Attestation{},
|
|
ExitSubnetTopicFormat: ðpb.SignedVoluntaryExit{},
|
|
ProposerSlashingSubnetTopicFormat: ðpb.ProposerSlashing{},
|
|
AttesterSlashingSubnetTopicFormat: ðpb.AttesterSlashing{},
|
|
AggregateAndProofSubnetTopicFormat: ðpb.SignedAggregateAttestationAndProof{},
|
|
SyncContributionAndProofSubnetTopicFormat: ðpb.SignedContributionAndProof{},
|
|
SyncCommitteeSubnetTopicFormat: ðpb.SyncCommitteeMessage{},
|
|
}
|
|
|
|
// GossipTopicMappings is a function to return the assigned data type
|
|
// versioned by epoch.
|
|
func GossipTopicMappings(topic string, epoch types.Epoch) proto.Message {
|
|
if topic == BlockSubnetTopicFormat {
|
|
if epoch >= params.BeaconConfig().BellatrixForkEpoch {
|
|
return ðpb.SignedBeaconBlockBellatrix{}
|
|
}
|
|
if epoch >= params.BeaconConfig().AltairForkEpoch {
|
|
return ðpb.SignedBeaconBlockAltair{}
|
|
}
|
|
}
|
|
return gossipTopicMappings[topic]
|
|
}
|
|
|
|
// AllTopics returns all topics stored in our
|
|
// gossip mapping.
|
|
func AllTopics() []string {
|
|
var topics []string
|
|
for k := range gossipTopicMappings {
|
|
topics = append(topics, k)
|
|
}
|
|
return topics
|
|
}
|
|
|
|
// GossipTypeMapping is the inverse of GossipTopicMappings so that an arbitrary protobuf message
|
|
// can be mapped to a protocol ID string.
|
|
var GossipTypeMapping = make(map[reflect.Type]string, len(gossipTopicMappings))
|
|
|
|
func init() {
|
|
for k, v := range gossipTopicMappings {
|
|
GossipTypeMapping[reflect.TypeOf(v)] = k
|
|
}
|
|
// Specially handle Altair Objects.
|
|
GossipTypeMapping[reflect.TypeOf(ðpb.SignedBeaconBlockAltair{})] = BlockSubnetTopicFormat
|
|
GossipTypeMapping[reflect.TypeOf(ðpb.SignedBeaconBlockBellatrix{})] = BlockSubnetTopicFormat
|
|
}
|