Turn off retries on CLI (#4918)

For the CLI it just ends up confusing people as to why it's "hanging"
before returning a 500. This can still be overridden with
VAULT_MAX_RETRIES.
This commit is contained in:
Jeff Mitchell 2018-07-12 18:38:18 -04:00 committed by GitHub
parent 8433bf26e9
commit a2b88fa239
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 12 additions and 0 deletions

View File

@ -6,6 +6,7 @@ import (
"fmt"
"io"
"io/ioutil"
"os"
"regexp"
"strings"
"sync"
@ -88,6 +89,11 @@ func (c *BaseCommand) Client() (*api.Client, error) {
return nil, errors.Wrap(err, "failed to create client")
}
// Turn off retries on the CLI
if os.Getenv(api.EnvVaultMaxRetries) == "" {
client.SetMaxRetries(0)
}
// Set the wrapping function
client.SetWrappingLookupFunc(c.DefaultWrappingLookupFunc)

View File

@ -1,6 +1,7 @@
package command
import (
"os"
"sort"
"strings"
"sync"
@ -35,6 +36,11 @@ func (p *Predict) Client() *api.Client {
client.SetToken(token)
}
// Turn off retries for prediction
if os.Getenv(api.EnvVaultMaxRetries) == "" {
client.SetMaxRetries(0)
}
p.client = client
}
})