Make NRT plugin implement PostBind Extension

Signed-off-by: pprokop <[email protected]>

Author: PiotrProkop

pkg/noderesourcetopology/cache/cache.go

@@ -56,4 +56,10 @@ type Interface interface {
 
 	// UnreserveNodeResources decrement from the node assumed resources the resources required by the given pod.
 	UnreserveNodeResources(nodeName string, pod *corev1.Pod)
+
+	// PostBind is called after a pod is successfully bound. These plugins are
+	// informational. A common application of this extension point is for cleaning
+	// up. If a plugin needs to clean-up its state after a pod is scheduled and
+	// bound, PostBind is the extension point that it should register.
+	PostBind(nodeName string, pod *corev1.Pod)
 }

pkg/noderesourcetopology/cache/overreserve.go

@@ -261,3 +261,5 @@ func (ov *OverReserve) Store() *nrtStore {
 func logIDFromTime() string {
 	return fmt.Sprintf("resync%v", time.Now().UnixMilli())
 }
+
+func (ov *OverReserve) PostBind(nodeName string, pod *corev1.Pod) {}

pkg/noderesourcetopology/cache/passthrough.go

@@ -48,3 +48,4 @@ func (pt Passthrough) NodeMaybeOverReserved(nodeName string, pod *corev1.Pod)  {
 func (pt Passthrough) NodeHasForeignPods(nodeName string, pod *corev1.Pod)     {}
 func (pt Passthrough) ReserveNodeResources(nodeName string, pod *corev1.Pod)   {}
 func (pt Passthrough) UnreserveNodeResources(nodeName string, pod *corev1.Pod) {}
+func (pt Passthrough) PostBind(nodeName string, pod *corev1.Pod)               {}

pkg/noderesourcetopology/plugin.go

@@ -93,6 +93,7 @@ var _ framework.FilterPlugin = &TopologyMatch{}
 var _ framework.ReservePlugin = &TopologyMatch{}
 var _ framework.ScorePlugin = &TopologyMatch{}
 var _ framework.EnqueueExtensions = &TopologyMatch{}
+var _ framework.PostBindPlugin = &TopologyMatch{}
 
 // Name returns name of the plugin. It is used in logs, etc.
 func (tm *TopologyMatch) Name() string {

pkg/noderesourcetopology/postbind.go

@@ -0,0 +1,28 @@
+/*
+Copyright 2023 The Kubernetes Authors.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+package noderesourcetopology
+
+import (
+	"context"
+
+	corev1 "k8s.io/api/core/v1"
+	"k8s.io/kubernetes/pkg/scheduler/framework"
+)
+
+func (tm *TopologyMatch) PostBind(ctx context.Context, state *framework.CycleState, pod *corev1.Pod, nodeName string) {
+	tm.nrtCache.PostBind(nodeName, pod)
+}