SNOW-1856499 Retry HEAD requests to GCP (#1274)

snowflakedb · Jan 16, 2025 · b0cd681 · b0cd681
1 parent 80c18ea
commit b0cd681
Showing 1 changed file with 6 additions and 1 deletion.
diff --git 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