prysm-pulse/validator/node/p2p_config.go
terence tsao e1137e5801 Allow Beacon Node to Connect to Bootstrap Node for Peer Discovery (#740)
* start sequence for node to connect to bootstrap nodes

* lint

* maybe warning is more appropiate?

* fix gazelle

* preston's feedback

* add go-ipfs-addr to workspace

* move flag to shared

* removed routed host
2018-11-14 21:25:06 -05:00

40 lines
1.1 KiB
Go

package node
import (
"github.com/golang/protobuf/proto"
pb "github.com/prysmaticlabs/prysm/proto/sharding/p2p/v1"
"github.com/prysmaticlabs/prysm/shared/cmd"
"github.com/prysmaticlabs/prysm/shared/p2p"
"github.com/prysmaticlabs/prysm/shared/p2p/adapter/tracer"
"github.com/urfave/cli"
)
var topicMappings = map[pb.Topic]proto.Message{
pb.Topic_COLLATION_BODY_REQUEST: &pb.CollationBodyRequest{},
pb.Topic_COLLATION_BODY_RESPONSE: &pb.CollationBodyResponse{},
pb.Topic_TRANSACTIONS: &pb.Transaction{},
}
func configureP2P(ctx *cli.Context) (*p2p.Server, error) {
s, err := p2p.NewServer(ctx.GlobalString(cmd.BootstrapNode.Name))
if err != nil {
return nil, err
}
traceAdapter, err := tracer.New("validator",
ctx.GlobalString(cmd.TracingEndpointFlag.Name),
ctx.GlobalFloat64(cmd.TraceSampleFractionFlag.Name),
ctx.GlobalBool(cmd.EnableTracingFlag.Name))
if err != nil {
return nil, err
}
// TODO(437): Define default adapters for logging, monitoring, etc.
adapters := []p2p.Adapter{traceAdapter}
for k, v := range topicMappings {
s.RegisterTopic(k.String(), v, adapters...)
}
return s, nil
}