kaspad/cmd/txgen/config.go
Dan Aharoni 1f35378a4d [NOD-386] Extract net parsing functionality to a shared place. (#453)
* [NOD-386] Extract net parsing functionality to a shared place.

* [NOD-386] Add extract ActiveNetParams to cmdconfig

* [NOD-386] Adding comments so go-vet won't shout at me

* [NOD-386] Rename package name to config

* [NOD-386] Rename commandConfig to configFlags

* [NOD-386] Rename function to ResolveNetwork

* [NOD-386] Fix renaming errors

* [NOD-386] Refactor network config to btcd level so APIserver and btcd could use it

* [NOD-386] Refactor network config to config package

* [NOD-386] Move ActiveNetParams to network section

* [NOD-386] Explictly return nil

* [NOD-386] Reuse activeNetParams from netwrok config

* [NOD-386] Set ActiveNetworkFlags instance to be global

* [NOD-386] Remove redundant newline

* [NOD-386] Init ActiveNetParams in address manager test

* [NOD-386] Add dnsseeder network config

* [NOD-386] Use ActiveConfig() method to access configuration
2019-11-12 10:51:36 +02:00

80 lines
2.8 KiB
Go

package main
import (
"github.com/daglabs/btcd/util"
"github.com/pkg/errors"
"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 configFlags 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 float64 `long:"fee-rate" description:"Average coins per gram fee rate"`
}
func parseConfig() (*configFlags, error) {
cfg := &configFlags{}
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
}