adds retryablehttp client in place of standard (#168)

segmentio · Jan 23, 2025 · 18bbd80 · 18bbd80
1 parent f1f4434
commit 18bbd80
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/internal/provider/provider.go b/internal/provider/provider.go
@@ -7,6 +7,7 @@ import (
 	"context"
 	"os"
 
+	"github.com/hashicorp/go-retryablehttp"
 	"github.com/hashicorp/terraform-plugin-framework/datasource"
 	"github.com/hashicorp/terraform-plugin-framework/path"
 	"github.com/hashicorp/terraform-plugin-framework/provider"
@@ -128,6 +129,9 @@ func (p *segmentProvider) Configure(ctx context.Context, req provider.ConfigureR
 			URL: url,
 		},
 	}
+	retryClient := retryablehttp.NewClient()
+	retryClient.RetryMax = 10
+	configuration.HTTPClient = retryClient.StandardClient()
 
 	client := api.NewAPIClient(configuration)