You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
48 lines
1.8 KiB
48 lines
1.8 KiB
package service
|
|
|
|
import (
|
|
"strconv"
|
|
|
|
nodeconfig "github.com/harmony-one/harmony/internal/configs/node"
|
|
"github.com/harmony-one/harmony/p2p"
|
|
)
|
|
|
|
// NodeConfig defines a structure of node configuration
|
|
// that can be used in services.
|
|
// This is to pass node configuration to services and prvent
|
|
// cyclic imports
|
|
type NodeConfig struct {
|
|
// The three groupID design, please refer to https://github.com/harmony-one/harmony/blob/master/node/node.md#libp2p-integration
|
|
Beacon p2p.GroupID // the beacon group ID
|
|
Group p2p.GroupID // the group ID of the shard
|
|
Client p2p.GroupID // the client group ID of the shard
|
|
IsClient bool // whether this node is a client node, such as wallet/txgen
|
|
IsBeacon bool // whether this node is a beacon node or not
|
|
IsWallet bool // whether this node is a wallet or not
|
|
IsLeader bool // whether this node is a leader or not
|
|
ShardID uint32 // shardID of this node
|
|
Actions map[p2p.GroupID]p2p.ActionType // actions on the groups
|
|
}
|
|
|
|
// GroupIDShards is a map of Group ID
|
|
// key is the shard ID
|
|
// value is the corresponding group ID
|
|
var (
|
|
GroupIDShards map[p2p.ShardIDType]p2p.GroupID
|
|
GroupIDShardClients map[p2p.ShardIDType]p2p.GroupID
|
|
)
|
|
|
|
func init() {
|
|
GroupIDShards = make(map[p2p.ShardIDType]p2p.GroupID)
|
|
GroupIDShardClients = make(map[p2p.ShardIDType]p2p.GroupID)
|
|
|
|
// init beacon chain group IDs
|
|
GroupIDShards["0"] = p2p.GroupIDBeacon
|
|
GroupIDShardClients["0"] = p2p.GroupIDBeaconClient
|
|
|
|
for i := 1; i < nodeconfig.MaxShards; i++ {
|
|
sid := p2p.ShardIDType(strconv.Itoa(i))
|
|
GroupIDShards[sid] = p2p.NewGroupIDShard(sid)
|
|
GroupIDShardClients[sid] = p2p.NewGroupIDShardClient(sid)
|
|
}
|
|
}
|
|
|