From 2f86292104ed7c61e8cff02b9596318427fd1992 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 44798c4023..7456d3251b 100644
--- a/pkg/storagecapabilities/storagecapabilities.go
+++ b/pkg/storagecapabilities/storagecapabilities.go
@@ -68,6 +68,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
@@ -347,6 +348,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 {