Skip to content

Commit

Permalink
Merge pull request #1787 from koendc/f-aws-max-retries
Browse files Browse the repository at this point in the history
aws: make MaxRetries for API calls configurable
  • Loading branch information
mitchellh committed May 4, 2015
2 parents f80b7b6 + 398c22a commit c8c0b02
Show file tree
Hide file tree
Showing 3 changed files with 27 additions and 8 deletions.
11 changes: 7 additions & 4 deletions builtin/providers/aws/config.go
Original file line number Diff line number Diff line change
Expand Up @@ -19,10 +19,11 @@ import (
)

type Config struct {
AccessKey string
SecretKey string
Token string
Region string
AccessKey string
SecretKey string
Token string
Region string
MaxRetries int

AllowedAccountIds []interface{}
ForbiddenAccountIds []interface{}
Expand Down Expand Up @@ -64,6 +65,7 @@ func (c *Config) Client() (interface{}, error) {
awsConfig := &aws.Config{
Credentials: creds,
Region: c.Region,
MaxRetries: c.MaxRetries,
}

log.Println("[INFO] Initializing ELB connection")
Expand Down Expand Up @@ -96,6 +98,7 @@ func (c *Config) Client() (interface{}, error) {
client.r53conn = route53.New(&aws.Config{
Credentials: creds,
Region: "us-east-1",
MaxRetries: c.MaxRetries,
})

log.Println("[INFO] Initializing Elasticache Connection")
Expand Down
20 changes: 16 additions & 4 deletions builtin/providers/aws/provider.go
Original file line number Diff line number Diff line change
Expand Up @@ -54,6 +54,13 @@ func Provider() terraform.ResourceProvider {
InputDefault: "us-east-1",
},

"max_retries": &schema.Schema{
Type: schema.TypeInt,
Optional: true,
Default: 11,
Description: descriptions["max_retries"],
},

"allowed_account_ids": &schema.Schema{
Type: schema.TypeSet,
Elem: &schema.Schema{Type: schema.TypeString},
Expand Down Expand Up @@ -132,15 +139,20 @@ func init() {

"token": "session token. A session token is only required if you are\n" +
"using temporary security credentials.",

"max_retries": "The maximum number of times an AWS API request is\n" +
"being executed. If the API request still fails, an error is\n" +
"thrown.",
}
}

func providerConfigure(d *schema.ResourceData) (interface{}, error) {
config := Config{
AccessKey: d.Get("access_key").(string),
SecretKey: d.Get("secret_key").(string),
Token: d.Get("token").(string),
Region: d.Get("region").(string),
AccessKey: d.Get("access_key").(string),
SecretKey: d.Get("secret_key").(string),
Token: d.Get("token").(string),
Region: d.Get("region").(string),
MaxRetries: d.Get("max_retries").(int),
}

if v, ok := d.GetOk("allowed_account_ids"); ok {
Expand Down
4 changes: 4 additions & 0 deletions website/source/docs/providers/aws/index.html.markdown
Original file line number Diff line number Diff line change
Expand Up @@ -43,6 +43,10 @@ The following arguments are supported in the `provider` block:
* `region` - (Required) This is the AWS region. It must be provided, but
it can also be sourced from the `AWS_DEFAULT_REGION` environment variables.

* `max_retries` - (Optional) This is the maximum number of times an API call is
being retried in case requests are being throttled or experience transient failures.
The delay between the subsequent API calls increases exponentially.

* `allowed_account_ids` - (Optional) List of allowed AWS account IDs (whitelist)
to prevent you mistakenly using a wrong one (and end up destroying live environment).
Conflicts with `forbidden_account_ids`.
Expand Down

0 comments on commit c8c0b02

Please sign in to comment.