Add gorealis timeout flag (#14)

Flag was added to control gorealis timeout
This commit is contained in:
lenhattan86 2020-10-08 09:01:56 -07:00 committed by GitHub
parent 2a76e3e2b4
commit bb370151e6
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -47,6 +47,7 @@ var filename string
var message = new(string)
var updateID string
var monitor bool
var timeout time.Duration // seconds
var log = logrus.New()
const australisVer = "v1.0.1"
@ -68,6 +69,7 @@ func init() {
rootCmd.PersistentFlags().StringVar(&configFile, "config", "/etc/aurora/australis.yml", "Config file to use.")
rootCmd.PersistentFlags().BoolVar(&toJson, "toJSON", false, "Print output in JSON format.")
rootCmd.PersistentFlags().StringVarP(&logLevel, "logLevel", "l", "info", "Set logging level ["+internal.GetLoggingLevels()+"].")
rootCmd.PersistentFlags().DurationVarP(&timeout, "timeout", "t", 20, "Gorealis timeout (default: 20 seconds)")
}
var rootCmd = &cobra.Command{
@ -142,7 +144,7 @@ func connect(cmd *cobra.Command, args []string) {
realisOptions := []realis.ClientOption{realis.BasicAuth(username, password),
realis.ThriftJSON(),
realis.Timeout(20 * time.Second),
realis.Timeout(timeout * time.Second),
realis.BackOff(realis.Backoff{
Steps: 2,
Duration: 10 * time.Second,