kaspad/cmd/txgen/config.go
Ori Newman 6bb53eaae3 [NOD-256] add error log (#369)
* [NOD-256] Add error log

* [NOD-256] Add error log

* [NOD-256] Fix typo and comment

* [NOD-256] Remove btclog dir

* [NOD-256] Format project

* [NOD-256] Add error log files

* [NOD-256] Add an option to add a log file to write into to an existing backend logger

* [NOD-256] Get rid of redundant logs initialization

* [NOD-256] rename initLogRotators to initLog

* [NOD-256] Get rid ExampleSignTxOutput and convert ExampleBlockDAG_ProcessBlock to a regular test

* [NOD-256] Show error message if os.Exiting from initLog
2019-08-21 11:26:21 +03:00

80 lines
2.8 KiB
Go

package main
import (
"errors"
"github.com/daglabs/btcd/util"
"path/filepath"
"github.com/jessevdk/go-flags"
)
const (
defaultLogFilename = "txgen.log"
defaultErrLogFilename = "txgen_err.log"
)
var (
// Default configuration options
defaultHomeDir = util.AppDataDir("txgen", false)
defaultLogFile = filepath.Join(defaultHomeDir, defaultLogFilename)
defaultErrLogFile = filepath.Join(defaultHomeDir, defaultErrLogFilename)
defaultTargetNumberOfOutputs uint64 = 1
defaultTargetNumberOfInputs uint64 = 1
)
type config struct {
Address string `long:"address" description:"An address to a JSON-RPC endpoints" required:"true"`
PrivateKey string `long:"private-key" description:"Private key" required:"true"`
SecondaryAddress string `long:"secondary-address" description:"An address that gets paid once per txgen run"`
CertificatePath string `long:"cert" description:"Path to certificate accepted by JSON-RPC endpoint"`
DisableTLS bool `long:"notls" description:"Disable TLS"`
TxInterval uint64 `long:"tx-interval" description:"Transaction emission interval (in milliseconds)"`
TargetNumberOfOutputs uint64 `long:"num-outputs" description:"Target number of transaction outputs (with some randomization)"`
TargetNumberOfInputs uint64 `long:"num-inputs" description:"Target number of transaction inputs (with some randomization)"`
AveragePayloadSize uint64 `long:"payload-size" description:"Average size of transaction payload"`
AverageGasFraction float64 `long:"gas-fraction" description:"The average portion of gas from the gas limit"`
AverageFeeRate uint64 `long:"fee-rate" description:"Average coins per byte fee rate"`
}
func parseConfig() (*config, error) {
cfg := &config{}
parser := flags.NewParser(cfg, flags.PrintErrors|flags.HelpFlag)
_, err := parser.Parse()
if err != nil {
return nil, err
}
if cfg.CertificatePath == "" && !cfg.DisableTLS {
return nil, errors.New("--notls has to be disabled if --cert is used")
}
if cfg.CertificatePath != "" && cfg.DisableTLS {
return nil, errors.New("--cert should be omitted if --notls is used")
}
if cfg.AverageGasFraction >= 1 || cfg.AverageGasFraction < 0 {
return nil, errors.New("--gas-fraction should be between 0 and 1")
}
if cfg.TargetNumberOfOutputs < 0 {
return nil, errors.New("--num-outputs should be positive")
}
if cfg.TargetNumberOfInputs < 0 {
return nil, errors.New("--num-inputs should be positive")
}
if cfg.TargetNumberOfOutputs == 0 {
cfg.TargetNumberOfOutputs = defaultTargetNumberOfOutputs
}
if cfg.TargetNumberOfInputs == 0 {
cfg.TargetNumberOfInputs = defaultTargetNumberOfInputs
}
initLog(defaultLogFile, defaultErrLogFile)
return cfg, nil
}