fix after review

snowflakedb · Dec 16, 2024 · 4c31de5 · 4c31de5
1 parent cfb818b
commit 4c31de5
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/azure_storage_client.go b/azure_storage_client.go
@@ -287,7 +287,7 @@ func (util *snowflakeAzureClient) nativeDownloadFile(
 	}
 	if meta.options.GetFileToStream {
 		blobDownloadResponse, err := withCloudStorageTimeout(util.cfg, func(ctx context.Context) (azblob.DownloadStreamResponse, error) {
-			return blobClient.DownloadStream(context.Background(), &azblob.DownloadStreamOptions{})
+			return blobClient.DownloadStream(ctx, &azblob.DownloadStreamOptions{})
 		})
 		if err != nil {
 			return err

diff --git a/s3_storage_client.go b/s3_storage_client.go
@@ -199,7 +199,7 @@ func (util *snowflakeS3Client) uploadFile(
 			if meta.realSrcStream != nil {
 				uploadStream = meta.realSrcStream
 			}
-			return uploader.Upload(context.Background(), &s3.PutObjectInput{
+			return uploader.Upload(ctx, &s3.PutObjectInput{
 				Bucket:   &s3loc.bucketName,
 				Key:      &s3path,
 				Body:     bytes.NewBuffer(uploadStream.Bytes()),
@@ -211,13 +211,13 @@ func (util *snowflakeS3Client) uploadFile(
 		if err != nil {
 			return nil, err
 		}
-		return uploader.Upload(context.Background(), &s3.PutObjectInput{
+		return uploader.Upload(ctx, &s3.PutObjectInput{
 			Bucket:   &s3loc.bucketName,
 			Key:      &s3path,
 			Body:     file,
 			Metadata: s3Meta,
 		})
-		
+
 	})
 
 	if err != nil {