mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-28 14:17:17 +00:00
92131695b1
* add new block encoding for snappy * Merge branch 'master' into fixEncoder * flip flag * Merge branch 'fixEncoder' of https://github.com/prysmaticlabs/geth-sharding into fixEncoder * not an underscore
37 lines
981 B
Go
37 lines
981 B
Go
package sync
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/gogo/protobuf/proto"
|
|
pubsub "github.com/libp2p/go-libp2p-pubsub"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/p2p"
|
|
)
|
|
|
|
func (r *Service) decodePubsubMessage(msg *pubsub.Message) (proto.Message, error) {
|
|
if msg == nil || msg.TopicIDs == nil || len(msg.TopicIDs) == 0 {
|
|
return nil, errors.New("nil pubsub message")
|
|
}
|
|
topic := msg.TopicIDs[0]
|
|
topic = strings.TrimSuffix(topic, r.p2p.Encoding().ProtocolSuffix())
|
|
topic = r.replaceForkDigest(topic)
|
|
base, ok := p2p.GossipTopicMappings[topic]
|
|
if !ok {
|
|
return nil, fmt.Errorf("no message mapped for topic %s", topic)
|
|
}
|
|
m := proto.Clone(base)
|
|
if err := r.p2p.Encoding().DecodeGossip(msg.Data, m); err != nil {
|
|
return nil, err
|
|
}
|
|
return m, nil
|
|
}
|
|
|
|
// Replaces our fork digest with the formatter.
|
|
func (r *Service) replaceForkDigest(topic string) string {
|
|
subStrings := strings.Split(topic, "/")
|
|
subStrings[2] = "%x"
|
|
return strings.Join(subStrings, "/")
|
|
}
|