Adopt zones in scheduler spread by constraints

Signed-off-by: Tanggui Bian <softwarebtg@163.com>

pkg/scheduler/core/spreadconstraint/group_clusters.go

@@ -136,22 +136,23 @@ func (info *GroupClustersInfo) generateZoneInfo(spreadConstraints []policyv1alph
 	}
 
 	for _, clusterInfo := range info.Clusters {
-		zone := clusterInfo.Cluster.Spec.Zone
+		zones := clusterInfo.Cluster.Spec.Zones
-		if zone == "" {
+		if len(zones) == 0 {
 			continue
 		}
 
-		zoneInfo, ok := info.Zones[zone]
+		for _, zone := range zones {
-		if !ok {
+			zoneInfo, ok := info.Zones[zone]
-			zoneInfo = ZoneInfo{
+			if !ok {
-				Name:     zone,
+				zoneInfo = ZoneInfo{
-				Clusters: make([]ClusterDetailInfo, 0),
+					Name:     zone,
+					Clusters: make([]ClusterDetailInfo, 0),
+				}
 			}
+			zoneInfo.Clusters = append(zoneInfo.Clusters, clusterInfo)
+			zoneInfo.AvailableReplicas += clusterInfo.AvailableReplicas
+			info.Zones[zone] = zoneInfo
 		}
-
-		zoneInfo.Clusters = append(zoneInfo.Clusters, clusterInfo)
-		zoneInfo.AvailableReplicas += clusterInfo.AvailableReplicas
-		info.Zones[zone] = zoneInfo
 	}
 
 	for zone, zoneInfo := range info.Zones {

pkg/scheduler/core/spreadconstraint/select_clusters_test.go

@@ -18,7 +18,7 @@ func NewClusterWithTopology(name, provider, region, zone string) *clusterv1alpha
 		Spec: clusterv1alpha1.ClusterSpec{
 			Provider: provider,
 			Region:   region,
-			Zone:     zone,
+			Zones:    []string{zone},
 		},
 	}
 }