diff --git a/cmd/root.go b/cmd/root.go index 671cd30..b39a7e5 100644 --- a/cmd/root.go +++ b/cmd/root.go @@ -47,6 +47,7 @@ var filename string var message = new(string) var updateID string var monitor bool +var timeout int64 // seconds var log = logrus.New() const australisVer = "v0.22.0" @@ -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().Int64VarP(&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(time.Duration(timeout) * time.Second), realis.BackOff(realis.Backoff{ Steps: 2, Duration: 10 * time.Second,