x-check-mailserver: add --datadir flag

Signed-off-by: Jakub Sokołowski <jakub@status.im>
This commit is contained in:
Jakub Sokołowski 2020-07-21 12:27:55 +02:00
parent e4fac6e106
commit 4681e09cfe
No known key found for this signature in database
GPG Key ID: 4EF064D0E6D63020
2 changed files with 6 additions and 3 deletions

View File

@ -9,6 +9,7 @@ import (
var ( var (
fleet = pflag.StringP("fleet", "f", params.FleetProd, "cluster fleet") fleet = pflag.StringP("fleet", "f", params.FleetProd, "cluster fleet")
datadir = pflag.StringP("datadir", "d", "/tmp", "home directory for node data")
mailservers = pflag.StringArrayP("mailservers", "m", nil, "a list of mail servers") mailservers = pflag.StringArrayP("mailservers", "m", nil, "a list of mail servers")
duration = pflag.DurationP("duration", "l", time.Hour*24, "length of time span from now") duration = pflag.DurationP("duration", "l", time.Hour*24, "length of time span from now")
channels = pflag.StringArrayP("channels", "c", []string{"status"}, "name of one or more channels") channels = pflag.StringArrayP("channels", "c", []string{"status"}, "name of one or more channels")

View File

@ -12,6 +12,7 @@ import (
"time" "time"
"github.com/ethereum/go-ethereum/log" "github.com/ethereum/go-ethereum/log"
"github.com/ethereum/go-ethereum/p2p/enode"
"github.com/status-im/status-go/logutils" "github.com/status-im/status-go/logutils"
"github.com/status-im/status-go/params" "github.com/status-im/status-go/params"
) )
@ -65,8 +66,9 @@ func main() {
wg.Add(len(mailserversToCheck)) wg.Add(len(mailserversToCheck))
for _, enode := range mailserversToCheck { for _, msEnode := range mailserversToCheck {
config.DataDir, err = ioutil.TempDir("", "") var nodeId = enode.MustParse(msEnode).ID().String()
config.DataDir, err = ioutil.TempDir(*datadir, nodeId)
if err != nil { if err != nil {
log.Crit("failed to create temp dir", "err", err) log.Crit("failed to create temp dir", "err", err)
} }
@ -74,7 +76,7 @@ func main() {
nodeConfig := *config nodeConfig := *config
log.Debug("using node config", "config", nodeConfig) log.Debug("using node config", "config", nodeConfig)
work := NewWorkUnit(enode, &nodeConfig) work := NewWorkUnit(msEnode, &nodeConfig)
go func(work *WorkUnit) { go func(work *WorkUnit) {
if err := work.Execute(workConfig); err != nil { if err := work.Execute(workConfig); err != nil {
log.Crit("failed to execute work", "err", err, "enode", work.MailServerEnode) log.Crit("failed to execute work", "err", err, "enode", work.MailServerEnode)