Merge pull request #197 from harmony-one/open_source_cleanup
move client/txgen to cmd/client/txgenpull/200/head
Leo Chen
6 years ago
committed by
GitHub
commit
a89059ce2b
@ -1,101 +0,0 @@ |
|||||||
package config |
|
||||||
|
|
||||||
import ( |
|
||||||
"bufio" |
|
||||||
"log" |
|
||||||
"os" |
|
||||||
"strconv" |
|
||||||
"strings" |
|
||||||
|
|
||||||
"github.com/harmony-one/harmony/p2p" |
|
||||||
) |
|
||||||
|
|
||||||
// Entry is a single config of a node.
|
|
||||||
type Entry struct { |
|
||||||
IP string |
|
||||||
Port string |
|
||||||
Role string |
|
||||||
ShardID string |
|
||||||
} |
|
||||||
|
|
||||||
// Config is a struct containing multiple Entry of all nodes.
|
|
||||||
type Config struct { |
|
||||||
config []Entry |
|
||||||
} |
|
||||||
|
|
||||||
// NewConfig returns a pointer to a Config.
|
|
||||||
func NewConfig() *Config { |
|
||||||
config := Config{} |
|
||||||
return &config |
|
||||||
} |
|
||||||
|
|
||||||
// GetValidators returns all the validator peers
|
|
||||||
func (config *Config) GetValidators() []p2p.Peer { |
|
||||||
var peerList []p2p.Peer |
|
||||||
for _, entry := range config.config { |
|
||||||
if entry.Role != "validator" { |
|
||||||
continue |
|
||||||
} |
|
||||||
peer := p2p.Peer{Port: entry.Port, IP: entry.IP} |
|
||||||
peerList = append(peerList, peer) |
|
||||||
} |
|
||||||
return peerList |
|
||||||
} |
|
||||||
|
|
||||||
// GetShardIDToLeaderMap returns all the leader peers and corresponding shard Ids
|
|
||||||
func (config *Config) GetShardIDToLeaderMap() map[uint32]p2p.Peer { |
|
||||||
shardIDLeaderMap := map[uint32]p2p.Peer{} |
|
||||||
for _, entry := range config.config { |
|
||||||
if entry.Role == "leader" { |
|
||||||
val, err := strconv.Atoi(entry.ShardID) |
|
||||||
if err == nil { |
|
||||||
shardIDLeaderMap[uint32(val)] = p2p.Peer{IP: entry.IP, Port: entry.Port} |
|
||||||
} else { |
|
||||||
log.Print("[Generator] Error parsing the shard Id ", entry.ShardID) |
|
||||||
} |
|
||||||
} |
|
||||||
} |
|
||||||
return shardIDLeaderMap |
|
||||||
} |
|
||||||
|
|
||||||
// GetClientPeer returns the client peer.
|
|
||||||
func (config *Config) GetClientPeer() *p2p.Peer { |
|
||||||
for _, entry := range config.config { |
|
||||||
if entry.Role != "client" { |
|
||||||
continue |
|
||||||
} |
|
||||||
peer := p2p.Peer{Port: entry.Port, IP: entry.IP} |
|
||||||
return &peer |
|
||||||
} |
|
||||||
return nil |
|
||||||
} |
|
||||||
|
|
||||||
// GetClientPort returns the port of the client node in the config
|
|
||||||
func (config *Config) GetClientPort() string { |
|
||||||
for _, entry := range config.config { |
|
||||||
if entry.Role == "client" { |
|
||||||
return entry.Port |
|
||||||
} |
|
||||||
} |
|
||||||
return "" |
|
||||||
} |
|
||||||
|
|
||||||
// ReadConfigFile parses the config file and return a 2d array containing the file data
|
|
||||||
func (config *Config) ReadConfigFile(filename string) error { |
|
||||||
file, err := os.Open(filename) |
|
||||||
if err != nil { |
|
||||||
log.Fatal("Failed to read config file ", filename) |
|
||||||
return err |
|
||||||
} |
|
||||||
defer file.Close() |
|
||||||
fscanner := bufio.NewScanner(file) |
|
||||||
|
|
||||||
result := []Entry{} |
|
||||||
for fscanner.Scan() { |
|
||||||
p := strings.Split(fscanner.Text(), " ") |
|
||||||
entry := Entry{p[0], p[1], p[2], p[3]} |
|
||||||
result = append(result, entry) |
|
||||||
} |
|
||||||
config.config = result |
|
||||||
return nil |
|
||||||
} |
|
||||||
Loading…
Reference in new issue