Merge pull request #10 from getamis/feature/add-host-and-port-in-static-file

core: add host and port in static node file
This commit is contained in:
Alan Chen 2017-08-11 16:19:29 +08:00 committed by GitHub
commit 62e9789120
1 changed files with 8 additions and 5 deletions

View File

@ -38,13 +38,16 @@ const (
defaultLocalDir = "/tmp/gdata"
datadirPrivateKey = "nodekey"
defaultIP = "localhost"
clientIdentifier = "geth"
staticNodeJson = "static-nodes.json"
genesisJson = "genesis.json"
)
var (
defaultIP = net.IPv4(127, 0, 0, 1)
)
func GenerateClusterKeys(numbers int) []*ecdsa.PrivateKey {
keys := make([]*ecdsa.PrivateKey, numbers)
for i := 0; i < len(keys); i++ {
@ -134,7 +137,7 @@ func saveNodeKey(key *ecdsa.PrivateKey) (string, error) {
return instanceDir, nil
}
func saveStaticNode(dataDir string, nodes []*discover.Node) error {
func saveStaticNode(dataDir string, nodes []string) error {
filePath := filepath.Join(dataDir, clientIdentifier)
keyPath := filepath.Join(filePath, staticNodeJson)
@ -146,12 +149,12 @@ func saveStaticNode(dataDir string, nodes []*discover.Node) error {
return ioutil.WriteFile(keyPath, raw, 0600)
}
func transformToStaticNodes(envs []*Env) []*discover.Node {
nodes := make([]*discover.Node, len(envs))
func transformToStaticNodes(envs []*Env) []string {
nodes := make([]string, len(envs))
for i, env := range envs {
nodeID := discover.PubkeyID(&env.Key.PublicKey)
nodes[i] = discover.NewNode(nodeID, net.ParseIP(defaultIP), 0, env.HttpPort)
nodes[i] = discover.NewNode(nodeID, defaultIP, 0, env.HttpPort).String()
}
return nodes
}