mirror of
https://codeberg.org/scip/io-exporter.git
synced 2025-12-16 20:11:00 +01:00
662dc39e8f
* add grafana dashboard * test png * print help when -h is present regardless of other flags
123 lines
3.3 KiB
Go
123 lines
3.3 KiB
Go
package cmd
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"io"
|
|
"log"
|
|
"os"
|
|
"strings"
|
|
|
|
flag "github.com/spf13/pflag"
|
|
|
|
"github.com/knadh/koanf/providers/posflag"
|
|
koanf "github.com/knadh/koanf/v2"
|
|
)
|
|
|
|
const (
|
|
Version = `v0.0.8`
|
|
SLEEP = 5
|
|
Usage = `io-exporter [options] <file>
|
|
Options:
|
|
-t --timeout <int> When should the operation timeout in seconds
|
|
-s --sleeptime <int> Time to sleep between checks (default: 5s)
|
|
-l --label <label=value> Add label to exported metric
|
|
-i --internals Also add labels about resource usage
|
|
-r --read Only execute the read test
|
|
-w --write Only execute the write test
|
|
-d --debug Enable debug log level
|
|
-h --help Show help
|
|
-v --version Show program version`
|
|
|
|
O_R = iota
|
|
O_W
|
|
O_RW
|
|
)
|
|
|
|
// config via commandline flags
|
|
type Config struct {
|
|
Showversion bool `koanf:"version"` // -v
|
|
Showhelp bool `koanf:"help"` // -h
|
|
Internals bool `koanf:"internals"` // -i
|
|
Debug bool `koanf:"debug"` // -d
|
|
ReadMode bool `koanf:"read"` // -r
|
|
WriteMode bool `koanf:"write"` // -w
|
|
Label []string `koanf:"label"` // -v
|
|
Timeout int `koanf:"timeout"` // -t
|
|
Port int `koanf:"port"` // -p
|
|
Sleeptime int `koanf:"sleeptime"` // -s
|
|
|
|
File string
|
|
Labels []Label
|
|
}
|
|
|
|
func InitConfig(output io.Writer) (*Config, error) {
|
|
var kloader = koanf.New(".")
|
|
|
|
// setup custom usage
|
|
flagset := flag.NewFlagSet("config", flag.ContinueOnError)
|
|
flagset.Usage = func() {
|
|
_, err := fmt.Fprintln(output, Usage)
|
|
if err != nil {
|
|
log.Fatalf("failed to print to output: %s", err)
|
|
}
|
|
}
|
|
|
|
// parse commandline flags
|
|
flagset.BoolP("version", "v", false, "show program version")
|
|
flagset.BoolP("help", "h", false, "show help")
|
|
flagset.BoolP("debug", "d", false, "enable debug logs")
|
|
flagset.BoolP("internals", "i", false, "add internal metrics")
|
|
flagset.BoolP("read", "r", false, "only execute the read test")
|
|
flagset.BoolP("write", "w", false, "only execute the write test")
|
|
flagset.StringArrayP("label", "l", nil, "additional labels")
|
|
flagset.IntP("timeout", "t", 1, "timeout for file operation in seconds")
|
|
flagset.IntP("port", "p", 9187, "prometheus metrics port to listen to")
|
|
flagset.IntP("sleeptime", "s", 5, "time to sleep between checks (default: 5s)")
|
|
|
|
if err := flagset.Parse(os.Args[1:]); err != nil {
|
|
return nil, fmt.Errorf("failed to parse program arguments: %w", err)
|
|
}
|
|
|
|
// command line setup
|
|
if err := kloader.Load(posflag.Provider(flagset, ".", kloader), nil); err != nil {
|
|
return nil, fmt.Errorf("error loading flags: %w", err)
|
|
}
|
|
|
|
// fetch values
|
|
conf := &Config{}
|
|
if err := kloader.Unmarshal("", &conf); err != nil {
|
|
return nil, fmt.Errorf("error unmarshalling: %w", err)
|
|
}
|
|
|
|
// arg is the file under test
|
|
if len(flagset.Args()) > 0 {
|
|
conf.File = flagset.Args()[0]
|
|
} else {
|
|
if !conf.Showversion {
|
|
flagset.Usage()
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
for _, label := range conf.Label {
|
|
if len(label) == 0 {
|
|
continue
|
|
}
|
|
|
|
parts := strings.Split(label, "=")
|
|
if len(parts) != 2 {
|
|
return nil, errors.New("invalid label spec: " + label + ", expected label=value")
|
|
}
|
|
|
|
conf.Labels = append(conf.Labels, Label{Name: parts[0], Value: parts[1]})
|
|
}
|
|
|
|
if !conf.ReadMode && !conf.WriteMode {
|
|
conf.ReadMode = true
|
|
conf.WriteMode = true
|
|
}
|
|
|
|
return conf, nil
|
|
}
|