From ac2090973e4e3e4fc57bdf1e8da06de4003795d7 Mon Sep 17 00:00:00 2001
From: Arnon Gilboa <agilboa@redhat.com>
Date: Mon, 18 Nov 2024 14:56:34 +0200
Subject: [PATCH] Add RWX to AWS EBS io2 capabilities

Reference:
https://github.com/kubernetes-sigs/aws-ebs-csi-driver/pull/1799

Signed-off-by: Arnon Gilboa <agilboa@redhat.com>
---
 pkg/storagecapabilities/storagecapabilities.go | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/pkg/storagecapabilities/storagecapabilities.go b/pkg/storagecapabilities/storagecapabilities.go
index f712c64b59..534d35a161 100644
--- a/pkg/storagecapabilities/storagecapabilities.go
+++ b/pkg/storagecapabilities/storagecapabilities.go
@@ -70,6 +70,7 @@ var CapabilitiesByProvisionerKey = map[string][]StorageCapabilities{
 	// AWSElasticBlockStore
 	"kubernetes.io/aws-ebs": {{rwo, block}},
 	"ebs.csi.aws.com":       {{rwo, block}},
+	"ebs.csi.aws.com/io2":   {{rwx, block}, {rwo, block}, {rwo, file}},
 	// AWSElasticFileSystem
 	"efs.csi.aws.com": {{rwx, file}, {rwo, file}},
 	// Azure disk
@@ -350,6 +351,13 @@ var storageClassToProvisionerKeyMapper = map[string]func(sc *storagev1.StorageCl
 		}
 		return "driver.longhorn.io"
 	},
+	"ebs.csi.aws.com": func(sc *storagev1.StorageClass) string {
+		val := sc.Parameters["type"]
+		if val == "io2" {
+			return "ebs.csi.aws.com/io2"
+		}
+		return "ebs.csi.aws.com"
+	},
 }
 
 func getFSType(sc *storagev1.StorageClass) string {