koord-scheduler: retrieve reservationInfo from cache first in preFilter hook (koordinator-sh#906)

Signed-off-by: KunWuLuan <kunwuluan@gmail.com>

Author: KunWuLuan

pkg/scheduler/plugins/reservation/hook.go

@@ -153,6 +153,7 @@ func (h *Hook) prepareMatchReservationState(handle frameworkext.ExtendedHandle,
 		klog.V(6).InfoS("PreFilterHook indexer list reservation on node",
 			"node", node.Name, "count", len(rOnNode))
 		count := 0
+		rCache := getReservationCache()
 		for _, obj := range rOnNode {
 			r, ok := obj.(*schedulingv1alpha1.Reservation)
 			if !ok {
@@ -163,7 +164,12 @@ func (h *Hook) prepareMatchReservationState(handle frameworkext.ExtendedHandle,
 			if !util.IsReservationAvailable(r) {
 				continue
 			}
-			if matchReservation(pod, newReservationInfo(r)) {
+
+			rInfo := rCache.GetInCache(r)
+			if rInfo == nil {
+				rInfo = newReservationInfo(r)
+			}
+			if matchReservation(pod, rInfo) {
 				matchedCache.Add(r)
 				count++
 			} else {

pkg/scheduler/plugins/reservation/plugin.go

@@ -117,7 +117,7 @@ func New(args runtime.Object, handle framework.Handle) (framework.Plugin, error)
 		podLister:        extendedHandle.SharedInformerFactory().Core().V1().Pods().Lister(),
 		client:           extendedHandle.KoordinatorClientSet().SchedulingV1alpha1(),
 		parallelizeUntil: defaultParallelizeUntil(handle),
-		reservationCache: newReservationCache(),
+		reservationCache: getReservationCache(),
 	}
 
 	// handle reservation event in cache; here only scheduled and expired reservations are considered.

pkg/scheduler/plugins/reservation/rcache.go

@@ -228,6 +228,12 @@ type reservationCache struct {
 	assumed  map[string]*assumedInfo                    // reservation key -> assumed (pod allocated) reservation meta
 }
 
+var rCache *reservationCache = newReservationCache()
+
+func getReservationCache() *reservationCache {
+	return rCache
+}
+
 func newReservationCache() *reservationCache {
 	return &reservationCache{
 		inactive: map[string]*schedulingv1alpha1.Reservation{},