/
service_setup.go
89 lines (80 loc) · 2.66 KB
/
service_setup.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
package node
import (
"fmt"
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/consensus"
"github.com/harmony-one/harmony/api/service/explorer"
"github.com/harmony-one/harmony/api/service/networkinfo"
nodeconfig "github.com/harmony-one/harmony/internal/configs/node"
"github.com/harmony-one/harmony/internal/utils"
)
func (node *Node) setupForValidator() {
_, chanPeer, _ := node.initNodeConfiguration()
// Register networkinfo service. "0" is the beacon shard ID
node.serviceManager.RegisterService(
service.NetworkInfo,
networkinfo.MustNew(
node.host, node.NodeConfig.GetShardGroupID(), chanPeer, nil, node.networkInfoDHTPath(),
),
)
// 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.Consensus.CommitSigChannel, node.WaitForConsensusReadyV2),
)
}
func (node *Node) setupForExplorerNode() {
_, chanPeer, _ := node.initNodeConfiguration()
// Register networkinfo service.
node.serviceManager.RegisterService(
service.NetworkInfo,
networkinfo.MustNew(
node.host, node.NodeConfig.GetShardGroupID(), chanPeer, nil, node.networkInfoDHTPath()),
)
// Register explorer service.
node.serviceManager.RegisterService(
service.SupportExplorer, explorer.New(&node.SelfPeer, node.stateSync, node.Blockchain()),
)
}
// ServiceManagerSetup setups service store.
func (node *Node) ServiceManagerSetup() {
node.serviceManager = &service.Manager{}
node.serviceMessageChan = make(map[service.Type]chan *msg_pb.Message)
switch node.NodeConfig.Role() {
case nodeconfig.Validator:
node.setupForValidator()
case nodeconfig.ExplorerNode:
node.setupForExplorerNode()
}
node.serviceManager.SetupServiceMessageChan(node.serviceMessageChan)
}
// RunServices runs registered services.
func (node *Node) RunServices() {
if node.serviceManager == nil {
utils.Logger().Info().Msg("Service manager is not set up yet.")
return
}
node.serviceManager.RunServices()
}
// StopServices runs registered services.
func (node *Node) StopServices() {
if node.serviceManager == nil {
utils.Logger().Info().Msg("Service manager is not set up yet.")
return
}
node.serviceManager.StopServicesByRole([]service.Type{})
}
func (node *Node) networkInfoDHTPath() string {
return fmt.Sprintf(".dht-%s-%s-c%s",
node.SelfPeer.IP,
node.SelfPeer.Port,
node.chainConfig.ChainID,
)
}