github/prysm
1
1
Fork 0
mirror of https://github.com/OffchainLabs/prysm.git synced 2026-01-09 15:37:56 -05:00
Code Issues Packages Projects Releases Wiki Activity

beacon: Beacon Node gRPC Server and Client Services (#364)

Browse Source
This commit is contained in:
Raul Jordan
2018-08-01 17:08:44 -05:00
committed by GitHub
parent c9f1bfc19c
commit 29596bf862
14 changed files with 295 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
beacon-chain/node/node.go
View File

@@ -11,6 +11,7 @@ import (
"github.com/ethereum/go-ethereum/common"
"github.com/prysmaticlabs/prysm/beacon-chain/blockchain"
"github.com/prysmaticlabs/prysm/beacon-chain/powchain"
"github.com/prysmaticlabs/prysm/beacon-chain/rpc"
"github.com/prysmaticlabs/prysm/beacon-chain/simulator"
rbcsync "github.com/prysmaticlabs/prysm/beacon-chain/sync"
"github.com/prysmaticlabs/prysm/beacon-chain/utils"
@@ -64,16 +65,18 @@ func New(ctx *cli.Context) (*BeaconNode, error) {
return nil, err
}
if ctx.GlobalBool(utils.SimulatorFlag.Name) {
if err := beacon.registerSimulatorService(); err != nil {
return nil, err
}
if err := beacon.registerSimulatorService(ctx); err != nil {
return nil, err
}
if err := beacon.registerSyncService(); err != nil {
return nil, err
}
if err := beacon.registerRPCService(ctx); err != nil {
return nil, err
}
return beacon, nil
}
@@ -180,7 +183,10 @@ func (b *BeaconNode) registerSyncService() error {
return b.services.RegisterService(syncService)
}
func (b *BeaconNode) registerSimulatorService() error {
func (b *BeaconNode) registerSimulatorService(ctx *cli.Context) error {
if !ctx.GlobalBool(utils.SimulatorFlag.Name) {
return nil
}
var p2pService *p2p.Server
if err := b.services.FetchService(&p2pService); err != nil {
return err
@@ -200,3 +206,11 @@ func (b *BeaconNode) registerSimulatorService() error {
simulatorService := simulator.NewSimulator(context.TODO(), cfg, p2pService, web3Service, chainService)
return b.services.RegisterService(simulatorService)
}
func (b *BeaconNode) registerRPCService(ctx *cli.Context) error {
port := ctx.GlobalString(utils.RPCPort.Name)
rpcService := rpc.NewRPCService(context.TODO(), &rpc.Config{
Port: port,
})
return b.services.RegisterService(rpcService)
}