mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-05-30 02:36:42 +00:00

* [NOD-213] Add customization to txgen * [NOD-213] Add fee rate as an argument * [NOD-213] Don't delay transaction emission if there's no need * [NOD-213] enqueueTransactions -> queueTransactions * [NOD-213] reuse delay variable * [NOD-213] Add ExtractGasLimit function * [NOD-213] Use time.Ticket in sendTransactionLoop
77 lines
2.5 KiB
Go
77 lines
2.5 KiB
Go
package main
|
|
|
|
import (
|
|
"errors"
|
|
"github.com/daglabs/btcd/util"
|
|
"path/filepath"
|
|
|
|
"github.com/jessevdk/go-flags"
|
|
)
|
|
|
|
const (
|
|
defaultLogFilename = "txgen.log"
|
|
)
|
|
|
|
var (
|
|
// Default configuration options
|
|
defaultHomeDir = util.AppDataDir("txgen", false)
|
|
defaultLogFile = filepath.Join(defaultHomeDir, defaultLogFilename)
|
|
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"`
|
|
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
|
|
}
|
|
|
|
initLogRotator(defaultLogFile)
|
|
|
|
return cfg, nil
|
|
}
|