Add custom http client to helpers aws session

Author: robellison-je

pkg/common/helpers.go

@@ -18,6 +18,7 @@ import (
 	"crypto/x509"
 	"fmt"
 	"hash"
+	"net/http"
 	"os"
 	"time"
 
@@ -33,11 +34,18 @@ import (
 // using the standard auth flow.  We also have the ability to pass a role ARN
 // to allow for roles to be assumed in cross-account access flows.
 func GetAWSSession(region string, roleARN string, endpoint string) (sess *session.Session, cfg *aws.Config, accountID *string, err error) {
+	transport := http.DefaultTransport.(*http.Transport).Clone()
+	transport.MaxIdleConnsPerHost = transport.MaxIdleConns
+	httpClient := &http.Client{
+		Transport: transport,
+	}
+
 	sess = session.Must(session.NewSessionWithOptions(session.Options{
 		SharedConfigState: session.SharedConfigEnable,
 		Config: aws.Config{
-			Region:   aws.String(region),
-			Endpoint: aws.String(endpoint),
+			Region:     aws.String(region),
+			Endpoint:   aws.String(endpoint),
+			HTTPClient: httpClient,
 		},
 	}))
 
@@ -46,6 +54,7 @@ func GetAWSSession(region string, roleARN string, endpoint string) (sess *sessio
 		cfg = &aws.Config{
 			Credentials: creds,
 			Region:      aws.String(region),
+			HTTPClient:  httpClient,
 		}
 	}