SNOW-996678 Use snowflakeTransport for all clouds

azure_storage_client.go

@@ -49,6 +49,9 @@ func (util *snowflakeAzureClient) createClient(info *execResponseStageInfo, _ bo
 				MaxRetries: 60,
 				RetryDelay: 2 * time.Second,
 			},
+			Transport: &http.Client{
+				Transport: SnowflakeTransport,
+			},
 		},
 	})
 	if err != nil {

gcs_storage_client.go

@@ -69,8 +69,7 @@ func (util *snowflakeGcsClient) getFileHeader(meta *fileMetadata, filename strin
 		for k, v := range gcsHeaders {
 			req.Header.Add(k, v)
 		}
-		var client gcsAPI
-		client = &http.Client{}
+		client := newGcsClient()
 		// for testing only
 		if meta.mockGcsClient != nil {
 			client = meta.mockGcsClient
@@ -216,8 +215,7 @@ func (util *snowflakeGcsClient) uploadFile(
 	for k, v := range gcsHeaders {
 		req.Header.Add(k, v)
 	}
-	var client gcsAPI
-	client = &http.Client{}
+	client := newGcsClient()
 	// for testing only
 	if meta.mockGcsClient != nil {
 		client = meta.mockGcsClient
@@ -295,8 +293,7 @@ func (util *snowflakeGcsClient) nativeDownloadFile(
 	for k, v := range gcsHeaders {
 		req.Header.Add(k, v)
 	}
-	var client gcsAPI
-	client = &http.Client{}
+	client := newGcsClient()
 	// for testing only
 	if meta.mockGcsClient != nil {
 		client = meta.mockGcsClient
@@ -390,3 +387,9 @@ func (util *snowflakeGcsClient) generateFileURL(stageLocation string, filename s
 func (util *snowflakeGcsClient) isTokenExpired(resp *http.Response) bool {
 	return resp.StatusCode == 401
 }
+
+func newGcsClient() gcsAPI {
+	return &http.Client{
+		Transport: SnowflakeTransport,
+	}
+}

s3_storage_client.go

@@ -8,6 +8,7 @@ import (
 	"errors"
 	"fmt"
 	"io"
+	"net/http"
 	"os"
 	"strings"
 
@@ -52,6 +53,9 @@ func (util *snowflakeS3Client) createClient(info *execResponseStageInfo, useAcce
 			stageCredentials.AwsToken)),
 		EndpointResolver: resolver,
 		UseAccelerate:    useAccelerateEndpoint,
+		HTTPClient: &http.Client{
+			Transport: SnowflakeTransport,
+		},
 	}), nil
 }