feat: scale disruption cost by the node utilization

Signed-off-by: Cameron McAvoy <[email protected]>

Author: cnmcavoy

pkg/controllers/disruption/types.go

@@ -112,7 +112,7 @@ func NewCandidate(ctx context.Context, kubeClient client.Client, recorder events
 		zone:              node.Labels()[corev1.LabelTopologyZone],
 		reschedulablePods: lo.Filter(pods, func(p *corev1.Pod, _ int) bool { return pod.IsReschedulable(p) }),
 		// We get the disruption cost from all pods in the candidate, not just the reschedulable pods
-		disruptionCost: disruptionutils.ReschedulingCost(ctx, pods) * disruptionutils.LifetimeRemaining(clk, nodePool, node.NodeClaim),
+		disruptionCost: disruptionutils.ReschedulingCost(ctx, pods) * disruptionutils.LifetimeRemaining(clk, nodePool, node.NodeClaim) * node.Utilization(),
 	}, nil
 }
 

pkg/controllers/state/statenode.go

@@ -365,6 +365,18 @@ func (in *StateNode) Available() corev1.ResourceList {
 	return resources.Subtract(in.Allocatable(), in.PodRequests())
 }
 
+// Utilization is the ratio of requested resources to allocatable resources
+func (in *StateNode) Utilization() float64 {
+	alloc := in.Allocatable()
+	requested := in.PodRequests()
+	utilization := 0.0
+	for resource, request := range requested {
+		allocResource := alloc[resource]
+		utilization += float64(request.MilliValue()) / float64(allocResource.MilliValue())
+	}
+	return utilization / float64(len(requested))
+}
+
 func (in *StateNode) DaemonSetRequests() corev1.ResourceList {
 	return resources.Merge(lo.Values(in.daemonSetRequests)...)
 }