diff --git a/cmd/root.go b/cmd/root.go index 6915d52..ea2cc4d 100644 --- a/cmd/root.go +++ b/cmd/root.go @@ -44,6 +44,7 @@ func RootCmd(version string) *cobra.Command { cmd.PersistentFlags().BoolVarP(&imsConfig.ProxyIgnoreTLS, "proxyIgnoreTLS", "T", false, "Ignore TLS certificate verification (only valid when connecting through a proxy).") cmd.PersistentFlags().StringVarP(&configFile, "configFile", "f", "", "Configuration file.") + cmd.PersistentFlags().IntVar(&imsConfig.Timeout, "timeout", 30, "HTTP client timeout in seconds.") cmd.AddCommand( authzCmd(imsConfig), diff --git a/ims/client.go b/ims/client.go index cf77d03..5ab17f8 100644 --- a/ims/client.go +++ b/ims/client.go @@ -22,13 +22,13 @@ func (i Config) httpClient() (*http.Client, error) { t.TLSClientConfig = &tls.Config{InsecureSkipVerify: true} } client := &http.Client{ - Timeout: 30 * time.Second, + Timeout: time.Duration(i.Timeout) * time.Second, Transport: t, } return client, nil } client := &http.Client{ - Timeout: 30 * time.Second, + Timeout: time.Duration(i.Timeout) * time.Second, } return client, nil }