diff --git a/go.mod b/go.mod index f862c91c0..94de91860 100644 --- a/go.mod +++ b/go.mod @@ -14,6 +14,7 @@ require ( github.com/hashicorp/go-cleanhttp v0.5.2 github.com/hashicorp/go-hclog v1.0.0 github.com/hashicorp/go-multierror v1.1.1 + github.com/hashicorp/go-retryablehttp v0.7.0 github.com/hashicorp/go-secure-stdlib/awsutil v0.1.5 github.com/hashicorp/go-secure-stdlib/parseutil v0.1.2 github.com/hashicorp/terraform-plugin-sdk/v2 v2.10.0 diff --git a/util/util.go b/util/util.go index b44c1499f..20ca23406 100644 --- a/util/util.go +++ b/util/util.go @@ -355,7 +355,10 @@ func StatusCheckRetry(statusCodes ...int) retryablehttp.CheckRetry { // SetupCCCRetryClient for handling Client Controlled Consistency related // requests. func SetupCCCRetryClient(client *api.Client, maxRetry int) { - client.SetReadYourWrites(true) + if !client.ReadYourWrites() { + client.SetReadYourWrites(true) + } + client.SetMaxRetries(maxRetry) client.SetCheckRetry(StatusCheckRetry(http.StatusNotFound))