ResiLien
/
terraform
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

state/remote/atlas: switch to retryablehttp 

The retryablehttp package implements basic retries w/ exponential
backoff, which helps the remote state push recover in cases of
connectivity blips or transient errors.
This commit is contained in:
Paul Hinze 2015-12-17 19:15:45 -06:00
parent 08ad2d2d7e
commit 1100243536
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
state/remote/atlas.go
View File

@ -13,7 +13,7 @@ import (
"path" "path"
"strings" "strings"
"github.com/hashicorp/go-cleanhttp" "github.com/hashicorp/go-retryablehttp"
"github.com/hashicorp/terraform/terraform" "github.com/hashicorp/terraform/terraform"
) )
@ -77,14 +77,14 @@ type AtlasClient struct {
Name string Name string
AccessToken string AccessToken string
RunId string RunId string
HTTPClient *http.Client HTTPClient *retryablehttp.Client
conflictHandlingAttempted bool conflictHandlingAttempted bool
} }
func (c *AtlasClient) Get() (*Payload, error) { func (c *AtlasClient) Get() (*Payload, error) {
// Make the HTTP request // Make the HTTP request
req, err := http.NewRequest("GET", c.url().String(), nil) req, err := retryablehttp.NewRequest("GET", c.url().String(), nil)
if err != nil { if err != nil {
return nil, fmt.Errorf("Failed to make HTTP request: %v", err) return nil, fmt.Errorf("Failed to make HTTP request: %v", err)
} }
@ -158,7 +158,7 @@ func (c *AtlasClient) Put(state []byte) error {
b64 := base64.StdEncoding.EncodeToString(hash[:]) b64 := base64.StdEncoding.EncodeToString(hash[:])
// Make the HTTP client and request // Make the HTTP client and request
req, err := http.NewRequest("PUT", base.String(), bytes.NewReader(state)) req, err := retryablehttp.NewRequest("PUT", base.String(), bytes.NewReader(state))
if err != nil { if err != nil {
return fmt.Errorf("Failed to make HTTP request: %v", err) return fmt.Errorf("Failed to make HTTP request: %v", err)
} }
@ -191,7 +191,7 @@ func (c *AtlasClient) Put(state []byte) error {
func (c *AtlasClient) Delete() error { func (c *AtlasClient) Delete() error {
// Make the HTTP request // Make the HTTP request
req, err := http.NewRequest("DELETE", c.url().String(), nil) req, err := retryablehttp.NewRequest("DELETE", c.url().String(), nil)
if err != nil { if err != nil {
return fmt.Errorf("Failed to make HTTP request: %v", err) return fmt.Errorf("Failed to make HTTP request: %v", err)
} }
@ -249,11 +249,11 @@ func (c *AtlasClient) url() *url.URL {
} }
} }
func (c *AtlasClient) http() *http.Client { func (c *AtlasClient) http() *retryablehttp.Client {
if c.HTTPClient != nil { if c.HTTPClient != nil {
return c.HTTPClient return c.HTTPClient
} }
return cleanhttp.DefaultClient() return retryablehttp.NewClient()
} }
// Atlas returns an HTTP 409 - Conflict if the pushed state reports the same // Atlas returns an HTTP 409 - Conflict if the pushed state reports the same