add use wyvern storage class attrib

avoltz · Sep 19, 2023 · 03564e4 · 03564e4
1 parent 0caf6fa
commit 03564e4
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 0 deletions.
diff --git a/pkg/blob/blob.go b/pkg/blob/blob.go
@@ -106,6 +106,7 @@ const (
 	mountPermissionsField        = "mountpermissions"
 	useDataPlaneAPIField         = "usedataplaneapi"
 	EcStrgAuthenticationField    = "edgecache-storage-auth"
+	UseWyvernField               = "wyvern-in-use"
 
 	// See https://docs.microsoft.com/en-us/rest/api/storageservices/naming-and-referencing-containers--blobs--and-metadata#container-names
 	containerNameMinLength = 3

diff --git a/pkg/blob/controllerserver.go b/pkg/blob/controllerserver.go
@@ -168,6 +168,8 @@ func (d *Driver) CreateVolume(ctx context.Context, req *csi.CreateVolumeRequest)
 			networkEndpointType = v
 		case EcStrgAuthenticationField:
 			containerNameReplaceMap[EcStrgAuthenticationField] = v
+		case usewyvern:
+			containerNameReplaceMap[usewyvern] = v
 		case mountPermissionsField:
 			// only do validations here, used in NodeStageVolume, NodePublishVolume
 			if v != "" {

diff --git a/pkg/blob/nodeserver.go b/pkg/blob/nodeserver.go
@@ -336,6 +336,11 @@ func (d *Driver) NodeStageVolume(ctx context.Context, req *csi.NodeStageVolumeRe
 	}
 
 	if protocol == EcProtocol {
+		useWyvern, useWyvernOk := attrib[UseWyvernField]
+		if useWyvernOk {
+			klog.Infof("this is the use wyvern field %s", useWyvern)
+		}
+
 		// get authentication method
 		storageAuthType, storageAuthTypeOk := attrib[EcStrgAuthenticationField]
 		if !storageAuthTypeOk {