Add support for passing Blob Query Params

openshift-pipelines · Oct 7, 2024 · 4cf0e53 · 4cf0e53
1 parent ad8fdc9
commit 4cf0e53
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/internal/provider/blob/blob.go b/internal/provider/blob/blob.go
@@ -19,6 +19,14 @@ const (
 	cacheFile = "cache.tar.gz"
 )
 
+var (
+	queryParams string
+)
+
+func init() {
+	queryParams = os.Getenv("BLOB_QUERY_PARAMS")
+}
+
 func Fetch(ctx context.Context, url url.URL, folder string) error {
 	log.Printf("Downloading cache from %s to %s", url.String(), folder)
 	file, err := os.CreateTemp("", cacheFile)
@@ -28,7 +36,7 @@ func Fetch(ctx context.Context, url url.URL, folder string) error {
 	}
 	defer os.Remove(file.Name())
 
-	bucket, err := blob.OpenBucket(ctx, url.String()+"?usePathStyle=true")
+	bucket, err := blob.OpenBucket(ctx, url.String()+queryParams)
 	if err != nil {
 		log.Printf("error opening bucket: %s", err)
 		return err
@@ -68,7 +76,7 @@ func Upload(ctx context.Context, url url.URL, folder string) error {
 		return err
 	}
 
-	bucket, err := blob.OpenBucket(ctx, url.String()+"?usePathStyle=true")
+	bucket, err := blob.OpenBucket(ctx, url.String()+queryParams)
 	if err != nil {
 		return err
 	}