diff --git a/cmd/config.go b/cmd/config.go index 16982ce..b1e9b77 100644 --- a/cmd/config.go +++ b/cmd/config.go @@ -150,6 +150,7 @@ func setDefaultConfig() { viper.SetDefault("randomize", false) viper.SetDefault("value", 10) viper.SetDefault("deviation", 5) + viper.SetDefault("probability", 100) viper.SetDefault("generators", []Custom{}) } diff --git a/cmd/flags.go b/cmd/flags.go index c033367..45e1df8 100644 --- a/cmd/flags.go +++ b/cmd/flags.go @@ -15,6 +15,7 @@ func commonFlags(cmd *cobra.Command) { f.Bool("randomize", viper.GetBool("randomize"), "toggle if starting point of generators should be randomized") f.Float64("value", viper.GetFloat64("value"), "starting value for generators") f.Float64("deviation", viper.GetFloat64("deviation"), "deviation for the next point in generator") + f.Uint8("probability", uint8(viper.GetUint("probability")), "probability of the points being sent, values in [1,100]") f.Uint("step", viper.GetUint("step"), "generators interval in seconds") }