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

Part 1 of Attestation Service - [sync -> blockchain -> rpc] (#548)

Browse Source 
* Revert "Add Skip Coverage Condition in Coverage.sh (#416)"

This reverts commit 72a5dd1cf4.

* add coverage.sh back

* beginning sig aggregation routine

* fixed errors

* fixed existing tests

* merged master

* add attestation build.bazel

* private visibility

* tests for attestation service

* lint

* bazel gazelle

* add validator beacon file

* lint

* tests for sync service

* removed tmp.go

* bazel gazelle
This commit is contained in:
terence tsao
2018-09-23 18:22:09 -07:00
committed by Nishant Das
parent f3774edeb1
commit 1aeb0e6b10
16 changed files with 830 additions and 442 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -12,12 +12,13 @@ import (
"github.com/ethereum/go-ethereum/common"
"github.com/ethereum/go-ethereum/ethclient"
gethRPC "github.com/ethereum/go-ethereum/rpc"
"github.com/prysmaticlabs/prysm/beacon-chain/attestation"
"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"
initialsync "github.com/prysmaticlabs/prysm/beacon-chain/sync/initial-sync"
"github.com/prysmaticlabs/prysm/beacon-chain/sync/initial-sync"
"github.com/prysmaticlabs/prysm/beacon-chain/utils"
"github.com/prysmaticlabs/prysm/shared"
"github.com/prysmaticlabs/prysm/shared/cmd"
@@ -69,6 +70,10 @@ func NewBeaconNode(ctx *cli.Context) (*BeaconNode, error) {
return nil, err
}
if err := beacon.registerService(); err != nil {
return nil, err
}
if err := beacon.registerSyncService(); err != nil {
return nil, err
}
@@ -180,6 +185,19 @@ func (b *BeaconNode) registerBlockchainService(ctx *cli.Context) error {
return b.services.RegisterService(blockchainService)
}
func (b *BeaconNode) registerService() error {
handler, err := attestation.NewHandler(b.db.DB())
if err != nil {
return fmt.Errorf("could not register attestation service: %v", err)
}
attestationService := attestation.NewAttestService(context.TODO(), &attestation.Config{
Handler: handler,
})
return b.services.RegisterService(attestationService)
}
func (b *BeaconNode) registerPOWChainService(ctx *cli.Context) error {
if ctx.GlobalBool(utils.DevFlag.Name) {
return nil
@@ -213,7 +231,16 @@ func (b *BeaconNode) registerSyncService() error {
return err
}
syncService := rbcsync.NewSyncService(context.Background(), rbcsync.DefaultConfig(), p2pService, chainService)
var attestationService *attestation.Service
if err := b.services.FetchService(&attestationService); err != nil {
return err
}
cfg := rbcsync.DefaultConfig()
cfg.ChainService = chainService
cfg.AttestService = attestationService
syncService := rbcsync.NewSyncService(context.Background(), cfg, p2pService)
return b.services.RegisterService(syncService)
}
@@ -279,6 +306,11 @@ func (b *BeaconNode) registerRPCService(ctx *cli.Context) error {
return err
}
var attestationService *attestation.Service
if err := b.services.FetchService(&attestationService); err != nil {
return err
}
var web3Service *powchain.Web3Service
var devMode = ctx.GlobalBool(utils.DevFlag.Name)
if !devMode {
@@ -291,14 +323,15 @@ func (b *BeaconNode) registerRPCService(ctx *cli.Context) error {
cert := ctx.GlobalString(utils.CertFlag.Name)
key := ctx.GlobalString(utils.KeyFlag.Name)
rpcService := rpc.NewRPCService(context.TODO(), &rpc.Config{
Port: port,
CertFlag: cert,
KeyFlag: key,
SubscriptionBuf: 100,
CanonicalFetcher: chainService,
ChainService: chainService,
POWChainService: web3Service,
DevMode: devMode,
Port: port,
CertFlag: cert,
KeyFlag: key,
SubscriptionBuf: 100,
CanonicalFetcher: chainService,
ChainService: chainService,
AttestationService: attestationService,
POWChainService: web3Service,
DevMode: devMode,
})
return b.services.RegisterService(rpcService)