cerc-io/eth-statediff-service
16
0
Fork 0
Code Issues 2 Pull Requests Actions Packages Projects Releases Wiki Activity

generate rand node id if none is provided

Browse Source
This commit is contained in:
i-norden 2021-10-28 10:15:38 -05:00
parent 728642afec
commit c83e8835ab
2 changed files with 52 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

53
cmd/root.go
View File

@ -18,8 +18,10 @@ package cmd
import ( import (
"fmt" "fmt"
"math/rand"
"os" "os"
"strings" "strings"
"time"
"github.com/ethereum/go-ethereum/statediff/indexer/node" "github.com/ethereum/go-ethereum/statediff/indexer/node"
"github.com/ethereum/go-ethereum/statediff/indexer/postgres" "github.com/ethereum/go-ethereum/statediff/indexer/postgres"
@ -172,6 +174,8 @@ func init() {
viper.BindPFlag("prerun.only", rootCmd.PersistentFlags().Lookup("prerun-only")) viper.BindPFlag("prerun.only", rootCmd.PersistentFlags().Lookup("prerun-only"))
viper.BindPFlag("prerun.start", rootCmd.PersistentFlags().Lookup("prerun-start")) viper.BindPFlag("prerun.start", rootCmd.PersistentFlags().Lookup("prerun-start"))
viper.BindPFlag("prerun.stop", rootCmd.PersistentFlags().Lookup("prerun-stop")) viper.BindPFlag("prerun.stop", rootCmd.PersistentFlags().Lookup("prerun-stop"))
rand.Seed(time.Now().UnixNano())
} }
func initConfig() { func initConfig() {
@ -188,13 +192,50 @@ func initConfig() {
} }
func GetEthNodeInfo() node.Info { func GetEthNodeInfo() node.Info {
return node.Info{ var nodeID, genesisBlock, networkID, clientName string
ID: viper.GetString("ethereum.nodeID"), var chainID uint64
ClientName: viper.GetString("ethereum.clientName"), if !viper.IsSet("ethereum.nodeID") {
GenesisBlock: viper.GetString("ethereum.genesisBlock"), nodeID = randSeq(12)
NetworkID: viper.GetString("ethereum.networkID"), } else {
ChainID: viper.GetUint64("ethereum.chainID"), nodeID = viper.GetString("ethereum.nodeID")
} }
if !viper.IsSet("ethereum.genesisBlock") {
genesisBlock = "0xd4e56740f876aef8c010b86a40d5f56745a118d0906a34e69aec8c0db1cb8fa3"
} else {
genesisBlock = viper.GetString("ethereum.genesisBlock")
}
if !viper.IsSet("ethereum.chainID") {
chainID = 1
} else {
chainID = viper.GetUint64("ethereum.chainID")
}
if !viper.IsSet("ethereum.networkID") {
networkID = "1"
} else {
networkID = viper.GetString("ethereum.networkID")
}
if !viper.IsSet("ethereum.clientName") {
clientName = "eth-statediff-service"
} else {
clientName = viper.GetString("ethereum.clientName")
}
return node.Info{
ID: nodeID,
ClientName: clientName,
GenesisBlock: genesisBlock,
NetworkID: networkID,
ChainID: chainID,
}
}
var characters = []rune("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789")
func randSeq(n int) string {
b := make([]rune, n)
for i := range b {
b[i] = characters[rand.Intn(len(characters))]
}
return string(b)
} }
func GetDBParams() postgres.ConnectionParams { func GetDBParams() postgres.ConnectionParams {