Adjust Node Label Conditions Based on Full Label Name on strechcluster

Signed-off-by: Oded Viner <[email protected]>
red-hat-storage · Jan 12, 2025 · 9f8e66d · 9f8e66d
1 parent 0fbee25
commit 9f8e66d
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 1 deletion.
diff --git a/api/v1/topologymap.go b/api/v1/topologymap.go
@@ -17,9 +17,12 @@ limitations under the License.
 package v1
 
 import (
+	corev1 "k8s.io/api/core/v1"
 	"strings"
 )
 
+const labelZoneFailureDomainWithoutBeta = "failure-domain.kubernetes.io/zone"
+
 // NewNodeTopologyMap returns an initialized NodeTopologyMap
 func NewNodeTopologyMap() *NodeTopologyMap {
 	return &NodeTopologyMap{
@@ -76,3 +79,19 @@ func (m *NodeTopologyMap) GetKeyValues(topologyKey string) (string, []string) {
 
 	return topologyKey, values
 }
+
+// GetKeyValues returns a node label matching the topologyKey and all values for StrechCluster
+// for that label across all storage nodes
+func (m *NodeTopologyMap) GetKeyValuesStrechCluster() (string, []string) {
+	values := []string{}
+	topologyKey := ""
+	for label, labelValues := range m.Labels {
+		if label == corev1.LabelZoneFailureDomainStable || label == labelZoneFailureDomainWithoutBeta {
+			topologyKey = label
+			values = labelValues
+			break
+		}
+	}
+
+	return topologyKey, values
+}
diff --git a/controllers/storagecluster/cephcluster.go b/controllers/storagecluster/cephcluster.go
@@ -1068,7 +1068,7 @@ func allowUnsupportedCephVersion() bool {
 func generateStretchClusterSpec(sc *ocsv1.StorageCluster) *rookCephv1.StretchClusterSpec {
 	var zones []string
 	stretchClusterSpec := rookCephv1.StretchClusterSpec{}
-	stretchClusterSpec.FailureDomainLabel, zones = sc.Status.NodeTopologies.GetKeyValues(getFailureDomain(sc))
+	stretchClusterSpec.FailureDomainLabel, zones = sc.Status.NodeTopologies.GetKeyValuesStrechCluster()
 
 	for _, zone := range zones {
 		if zone == sc.Spec.NodeTopologies.ArbiterLocation {