Rongjian Lan
6 years ago
parent
4f63fe6b77
commit
4a39a7cb96
@ -0,0 +1,143 @@ |
|||||||
|
package node |
||||||
|
|
||||||
|
import ( |
||||||
|
"os" |
||||||
|
|
||||||
|
"github.com/ethereum/go-ethereum/ethdb" |
||||||
|
msg_pb "github.com/harmony-one/harmony/api/proto/message" |
||||||
|
"github.com/harmony-one/harmony/api/service" |
||||||
|
"github.com/harmony-one/harmony/api/service/blockproposal" |
||||||
|
"github.com/harmony-one/harmony/api/service/clientsupport" |
||||||
|
"github.com/harmony-one/harmony/api/service/consensus" |
||||||
|
"github.com/harmony-one/harmony/api/service/discovery" |
||||||
|
"github.com/harmony-one/harmony/api/service/explorer" |
||||||
|
"github.com/harmony-one/harmony/api/service/networkinfo" |
||||||
|
"github.com/harmony-one/harmony/api/service/randomness" |
||||||
|
"github.com/harmony-one/harmony/api/service/staking" |
||||||
|
"github.com/harmony-one/harmony/internal/utils" |
||||||
|
"github.com/harmony-one/harmony/p2p" |
||||||
|
) |
||||||
|
|
||||||
|
func (node *Node) setupForShardLeader() { |
||||||
|
nodeConfig, chanPeer := node.initNodeConfiguration() |
||||||
|
|
||||||
|
// Register peer discovery service. No need to do staking for beacon chain node.
|
||||||
|
node.serviceManager.RegisterService(service.PeerDiscovery, discovery.New(node.host, nodeConfig, chanPeer)) |
||||||
|
// Register networkinfo service. "0" is the beacon shard ID
|
||||||
|
node.serviceManager.RegisterService(service.NetworkInfo, networkinfo.New(node.host, p2p.GroupIDBeacon, chanPeer)) |
||||||
|
|
||||||
|
// Register explorer service.
|
||||||
|
node.serviceManager.RegisterService(service.SupportExplorer, explorer.New(&node.SelfPeer)) |
||||||
|
// Register consensus service.
|
||||||
|
node.serviceManager.RegisterService(service.Consensus, consensus.New(node.BlockChannel, node.Consensus, node.startConsensus)) |
||||||
|
// Register new block service.
|
||||||
|
node.serviceManager.RegisterService(service.BlockProposal, blockproposal.New(node.Consensus.ReadySignal, node.WaitForConsensusReady)) |
||||||
|
// Register client support service.
|
||||||
|
node.serviceManager.RegisterService(service.ClientSupport, clientsupport.New(node.blockchain.State, node.CallFaucetContract, node.getDeployedStakingContract, node.SelfPeer.IP, node.SelfPeer.Port)) |
||||||
|
// Register randomness service
|
||||||
|
node.serviceManager.RegisterService(service.Randomness, randomness.New(node.DRand)) |
||||||
|
} |
||||||
|
|
||||||
|
func (node *Node) setupForShardValidator() { |
||||||
|
nodeConfig, chanPeer := node.initNodeConfiguration() |
||||||
|
|
||||||
|
// Register peer discovery service. "0" is the beacon shard ID. No need to do staking for beacon chain node.
|
||||||
|
node.serviceManager.RegisterService(service.PeerDiscovery, discovery.New(node.host, nodeConfig, chanPeer)) |
||||||
|
// Register networkinfo service. "0" is the beacon shard ID
|
||||||
|
node.serviceManager.RegisterService(service.NetworkInfo, networkinfo.New(node.host, p2p.GroupIDBeacon, chanPeer)) |
||||||
|
} |
||||||
|
|
||||||
|
func (node *Node) setupForBeaconLeader() { |
||||||
|
nodeConfig, chanPeer := node.initBeaconNodeConfiguration() |
||||||
|
|
||||||
|
// Register peer discovery service. No need to do staking for beacon chain node.
|
||||||
|
node.serviceManager.RegisterService(service.PeerDiscovery, discovery.New(node.host, nodeConfig, chanPeer)) |
||||||
|
// Register networkinfo service.
|
||||||
|
node.serviceManager.RegisterService(service.NetworkInfo, networkinfo.New(node.host, p2p.GroupIDBeacon, chanPeer)) |
||||||
|
// Register consensus service.
|
||||||
|
node.serviceManager.RegisterService(service.Consensus, consensus.New(node.BlockChannel, node.Consensus, node.startConsensus)) |
||||||
|
// Register new block service.
|
||||||
|
node.serviceManager.RegisterService(service.BlockProposal, blockproposal.New(node.Consensus.ReadySignal, node.WaitForConsensusReady)) |
||||||
|
// Register client support service.
|
||||||
|
node.serviceManager.RegisterService(service.ClientSupport, clientsupport.New(node.blockchain.State, node.CallFaucetContract, node.getDeployedStakingContract, node.SelfPeer.IP, node.SelfPeer.Port)) |
||||||
|
// Register randomness service
|
||||||
|
node.serviceManager.RegisterService(service.Randomness, randomness.New(node.DRand)) |
||||||
|
} |
||||||
|
|
||||||
|
func (node *Node) setupForBeaconValidator() { |
||||||
|
nodeConfig, chanPeer := node.initBeaconNodeConfiguration() |
||||||
|
|
||||||
|
// Register peer discovery service. No need to do staking for beacon chain node.
|
||||||
|
node.serviceManager.RegisterService(service.PeerDiscovery, discovery.New(node.host, nodeConfig, chanPeer)) |
||||||
|
// Register networkinfo service.
|
||||||
|
node.serviceManager.RegisterService(service.NetworkInfo, networkinfo.New(node.host, p2p.GroupIDBeacon, chanPeer)) |
||||||
|
// Register randomness service
|
||||||
|
node.serviceManager.RegisterService(service.Randomness, randomness.New(node.DRand)) |
||||||
|
} |
||||||
|
|
||||||
|
func (node *Node) setupForNewNode() { |
||||||
|
nodeConfig, chanPeer := node.initNodeConfiguration() |
||||||
|
|
||||||
|
// Register staking service.
|
||||||
|
node.serviceManager.RegisterService(service.Staking, staking.New(node.host, node.AccountKey, 0, node.beaconChain)) |
||||||
|
// Register peer discovery service. "0" is the beacon shard ID
|
||||||
|
node.serviceManager.RegisterService(service.PeerDiscovery, discovery.New(node.host, nodeConfig, chanPeer)) |
||||||
|
// Register networkinfo service. "0" is the beacon shard ID
|
||||||
|
node.serviceManager.RegisterService(service.NetworkInfo, networkinfo.New(node.host, p2p.GroupIDBeacon, chanPeer)) |
||||||
|
|
||||||
|
// TODO: how to restart networkinfo and discovery service after receiving shard id info from beacon chain?
|
||||||
|
} |
||||||
|
|
||||||
|
func (node *Node) setupForClientNode() { |
||||||
|
nodeConfig, chanPeer := node.initNodeConfiguration() |
||||||
|
|
||||||
|
// Register peer discovery service.
|
||||||
|
node.serviceManager.RegisterService(service.PeerDiscovery, discovery.New(node.host, nodeConfig, chanPeer)) |
||||||
|
// Register networkinfo service. "0" is the beacon shard ID
|
||||||
|
node.serviceManager.RegisterService(service.NetworkInfo, networkinfo.New(node.host, p2p.GroupIDBeacon, chanPeer)) |
||||||
|
} |
||||||
|
|
||||||
|
// AddBeaconChainDatabase adds database support for beaconchain blocks on normal sharding nodes (not BeaconChain node)
|
||||||
|
func (node *Node) AddBeaconChainDatabase(db ethdb.Database) { |
||||||
|
database := db |
||||||
|
if database == nil { |
||||||
|
database = ethdb.NewMemDatabase() |
||||||
|
} |
||||||
|
// TODO (chao) currently we use the same genesis block as normal shard
|
||||||
|
chain, err := node.GenesisBlockSetup(database) |
||||||
|
if err != nil { |
||||||
|
utils.GetLogInstance().Error("Error when doing genesis setup") |
||||||
|
os.Exit(1) |
||||||
|
} |
||||||
|
node.beaconChain = chain |
||||||
|
} |
||||||
|
|
||||||
|
// ServiceManagerSetup setups service store.
|
||||||
|
func (node *Node) ServiceManagerSetup() { |
||||||
|
node.serviceManager = &service.Manager{} |
||||||
|
node.serviceMessageChan = make(map[service.Type]chan *msg_pb.Message) |
||||||
|
switch node.Role { |
||||||
|
case ShardLeader: |
||||||
|
node.setupForShardLeader() |
||||||
|
case ShardValidator: |
||||||
|
node.setupForShardValidator() |
||||||
|
case BeaconLeader: |
||||||
|
node.setupForBeaconLeader() |
||||||
|
case BeaconValidator: |
||||||
|
node.setupForBeaconValidator() |
||||||
|
case NewNode: |
||||||
|
node.setupForNewNode() |
||||||
|
case ClientNode: |
||||||
|
node.setupForClientNode() |
||||||
|
} |
||||||
|
node.serviceManager.SetupServiceMessageChan(node.serviceMessageChan) |
||||||
|
} |
||||||
|
|
||||||
|
// RunServices runs registered services.
|
||||||
|
func (node *Node) RunServices() { |
||||||
|
if node.serviceManager == nil { |
||||||
|
utils.GetLogInstance().Info("Service manager is not set up yet.") |
||||||
|
return |
||||||
|
} |
||||||
|
node.serviceManager.RunServices() |
||||||
|
} |
||||||
Loading…
Reference in new issue