From b0cd68106e8bff900da79afecfb980da81006008 Mon Sep 17 00:00:00 2001
From: Piotr Fus <piotr.fus@snowflake.com>
Date: Thu, 16 Jan 2025 07:52:18 +0100
Subject: [PATCH] SNOW-1856499 Retry HEAD requests to GCP (#1274)

---
 gcs_storage_client.go | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/gcs_storage_client.go b/gcs_storage_client.go
index 8558094ba..cee14e395 100644
--- a/gcs_storage_client.go
+++ b/gcs_storage_client.go
@@ -78,7 +78,12 @@ func (util *snowflakeGcsClient) getFileHeader(meta *fileMetadata, filename strin
 			if meta.mockGcsClient != nil {
 				client = meta.mockGcsClient
 			}
-			return client.Do(req)
+			resp, err := client.Do(req)
+			if err != nil && strings.HasSuffix(err.Error(), "EOF") {
+				logger.Debug("Retrying HEAD request because of EOF")
+				resp, err = client.Do(req)
+			}
+			return resp, err
 		})
 		if err != nil {
 			return nil, err