Add topology.kubernetes.io/zone if topology.bsu.outscale.com/zone not…

… found
outscale · Sep 20, 2023 · f361a69 · f361a69
1 parent 8be720e
commit f361a69
Show file tree

Hide file tree

Showing 3 changed files with 33 additions and 2 deletions.
diff --git a/pkg/driver/controller.go b/pkg/driver/controller.go
@@ -534,12 +534,20 @@ func pickAvailabilityZone(requirement *csi.TopologyRequirement) string {
 		if exists {
 			return zone
 		}
+		zone, exists = topology.GetSegments()[TopologyK8sKey]
+		if exists {
+			return zone
+		}
 	}
 	for _, topology := range requirement.GetRequisite() {
 		zone, exists := topology.GetSegments()[TopologyKey]
 		if exists {
 			return zone
 		}
+		zone, exists = topology.GetSegments()[TopologyK8sKey]
+		if exists {
+			return zone
+		}
 	}
 	return ""
 }

diff --git a/pkg/driver/controller_test.go b/pkg/driver/controller_test.go
@@ -1266,6 +1266,28 @@ func TestPickAvailabilityZone(t *testing.T) {
 			},
 			expZone: expZone,
 		},
+		{
+			name: "Pick from requisite topologyK8sKey",
+			requirement: &csi.TopologyRequirement{
+				Requisite: []*csi.Topology{
+					{
+						Segments: map[string]string{TopologyK8sKey: expZone},
+					},
+				},
+			},
+			expZone: expZone,
+		},
+		{
+			name: "Pick from multi requisites",
+			requirement: &csi.TopologyRequirement{
+				Requisite: []*csi.Topology{
+					{
+						Segments: map[string]string{TopologyKey: expZone, TopologyK8sKey: expZone},
+					},
+				},
+			},
+			expZone: expZone,
+		},
 		{
 			name: "Pick from empty topology",
 			requirement: &csi.TopologyRequirement{

diff --git a/pkg/driver/driver.go b/pkg/driver/driver.go
@@ -41,8 +41,9 @@ const (
 )
 
 const (
-	DriverName  = "bsu.csi.outscale.com"
-	TopologyKey = "topology." + DriverName + "/zone"
+	DriverName     = "bsu.csi.outscale.com"
+	TopologyKey    = "topology." + DriverName + "/zone"
+	TopologyK8sKey = "topology.kubernetes.io/zone"
 )
 
 type Driver struct {