scheduler: optimize when only 1 reservation/node (#2136)

Signed-off-by: wangjianyu.wjy <wangjianyu.wjy@alibaba-inc.com> Co-authored-by: wangjianyu.wjy <wangjianyu.wjy@alibaba-inc.com>
koordinator-sh · Jul 17, 2024 · beac34f · beac34f
1 parent 290ae09
commit beac34f
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/pkg/scheduler/plugins/reservation/nominator.go b/pkg/scheduler/plugins/reservation/nominator.go
@@ -142,6 +142,10 @@ func (pl *Plugin) NominateReservation(ctx context.Context, cycleState *framework
 		return nil, nil
 	}
 
+	if len(reservationInfos) == 1 && state.hasAffinity {
+		return reservationInfos[0], nil
+	}
+
 	rInfo := pl.GetNominatedReservation(pod, nodeName)
 	if rInfo != nil {
 		return rInfo, nil
@@ -164,6 +168,10 @@ func (pl *Plugin) NominateReservation(ctx context.Context, cycleState *framework
 		return nil, nil
 	}
 
+	if len(reservations) == 1 {
+		return reservations[0], nil
+	}
+
 	nominated, _ := findMostPreferredReservationByOrder(reservations)
 	if nominated != nil {
 		return nominated, nil