From 5209a01fe6f8dde94a8074a76acb40684da46286 Mon Sep 17 00:00:00 2001
From: diptiranjanpx <51692470+diptiranjanpx@users.noreply.github.com>
Date: Fri, 20 Sep 2024 14:00:47 +0530
Subject: [PATCH] PWX-39128: Adding env variable SKIP_AWS_DRIVER_INIT support
 to skip the aws driver init in stork. (#1852)

(cherry picked from commit 4580c07da504373d1a0813d7b325812990465037)
---
 drivers/volume/aws/aws.go | 19 ++++++++++++-------
 1 file changed, 12 insertions(+), 7 deletions(-)

diff --git a/drivers/volume/aws/aws.go b/drivers/volume/aws/aws.go
index c3db9f7810..6785b544e7 100644
--- a/drivers/volume/aws/aws.go
+++ b/drivers/volume/aws/aws.go
@@ -2,6 +2,7 @@ package aws
 
 import (
 	"fmt"
+	"os"
 	"time"
 
 	aws_sdk "github.com/aws/aws-sdk-go/aws"
@@ -672,13 +673,17 @@ func (a *aws) IsVirtualMachineSupported() bool {
 }
 
 func init() {
-	a := &aws{}
-	err := a.Init(nil)
-	if err != nil {
-		logrus.Debugf("Error init'ing aws driver: %v", err)
-	}
-	if err := storkvolume.Register(storkvolume.AWSDriverName, a); err != nil {
-		logrus.Panicf("Error registering aws volume driver: %v", err)
+	if os.Getenv("SKIP_AWS_DRIVER_INIT") == "true" {
+		logrus.Infof("Skipping aws driver init")
+	} else {
+		a := &aws{}
+		err := a.Init(nil)
+		if err != nil {
+			logrus.Debugf("Error init'ing aws driver: %v", err)
+		}
+		if err := storkvolume.Register(storkvolume.AWSDriverName, a); err != nil {
+			logrus.Panicf("Error registering aws volume driver: %v", err)
+		}
 	}
 }