Support pod-level resources.

Author: nelchael

go.mod

@@ -16,6 +16,7 @@ require (
 	k8s.io/api v0.36.0
 	k8s.io/apimachinery v0.36.0
 	k8s.io/client-go v0.36.0
+	k8s.io/component-helpers v0.36.0
 	sigs.k8s.io/karpenter v1.12.0
 )
 

go.sum

@@ -270,8 +270,8 @@ k8s.io/client-go v0.36.0 h1:pOYi7C4RHChYjMiHpZSpSbIM6ZxVbRXBy7CuiIwqA3c=
 k8s.io/client-go v0.36.0/go.mod h1:ZKKcpwF0aLYfkHFCjillCKaTK/yBkEDHTDXCFY6AS9Y=
 k8s.io/component-base v0.36.0 h1:hFjEktssxiJhrK1zfybkH4kJOi8iZuF+mIDCqS5+jRo=
 k8s.io/component-base v0.36.0/go.mod h1:JZvIfcNHk+uck+8LhJzhSBtydWXaZNQwX2OdL+Mnwsk=
-k8s.io/component-helpers v0.35.0 h1:wcXv7HJRksgVjM4VlXJ1CNFBpyDHruRI99RrBtrJceA=
-k8s.io/component-helpers v0.35.0/go.mod h1:ahX0m/LTYmu7fL3W8zYiIwnQ/5gT28Ex4o2pymF63Co=
+k8s.io/component-helpers v0.36.0 h1:KznLAOD7oPxjaeheW4SOQijz9UtMO8Nvp89+lR8FYks=
+k8s.io/component-helpers v0.36.0/go.mod h1:BqZG+01Z97KR8GN9Stb8SiRmtn/EpZogriuQtpMCsLg=
 k8s.io/klog/v2 v2.140.0 h1:Tf+J3AH7xnUzZyVVXhTgGhEKnFqye14aadWv7bzXdzc=
 k8s.io/klog/v2 v2.140.0/go.mod h1:o+/RWfJ6PwpnFn7OyAG3QnO47BFsymfEfrz6XyYSSp0=
 k8s.io/kube-openapi v0.0.0-20260317180543-43fb72c5454a h1:xCeOEAOoGYl2jnJoHkC3hkbPJgdATINPMAxaynU2Ovg=

pkg/model/pod.go

@@ -22,6 +22,7 @@ import (
 
 	v1 "k8s.io/api/core/v1"
 	"k8s.io/apimachinery/pkg/api/resource"
+	helperResource "k8s.io/component-helpers/resource"
 )
 
 // Pod is our pod model used for internal storage and display
@@ -85,24 +86,10 @@ func (p *Pod) Phase() v1.PodPhase {
 func (p *Pod) Requested() v1.ResourceList {
 	p.mu.RLock()
 	defer p.mu.RUnlock()
-	requested := v1.ResourceList{}
-	for _, c := range p.pod.Spec.InitContainers {
-		if c.RestartPolicy == nil || *c.RestartPolicy != v1.ContainerRestartPolicyAlways {
-			continue
-		}
-		for rn, q := range c.Resources.Requests {
-			existing := requested[rn]
-			existing.Add(q)
-			requested[rn] = existing
-		}
-	}
-	for _, c := range p.pod.Spec.Containers {
-		for rn, q := range c.Resources.Requests {
-			existing := requested[rn]
-			existing.Add(q)
-			requested[rn] = existing
-		}
-	}
+	requested := helperResource.PodRequests(&p.pod, helperResource.PodResourcesOptions{
+		SkipPodLevelResources:       false,
+		SkipContainerLevelResources: false,
+	})
 	requested[v1.ResourcePods] = resource.MustParse("1")
 	return requested
 }

pkg/model/pod_test.go

@@ -74,6 +74,46 @@ func testPod(namespace, name string) *v1.Pod {
 	}
 	return p
 }
+
+func testPodWithPodLevelResources(namespace, name string) *v1.Pod {
+	restartAlways := v1.ContainerRestartPolicyAlways
+	p := &v1.Pod{
+		ObjectMeta: metav1.ObjectMeta{
+			Namespace: namespace,
+			Name:      name,
+		},
+		Status: v1.PodStatus{
+			Phase: v1.PodPending,
+		},
+		Spec: v1.PodSpec{
+			Resources: &v1.ResourceRequirements{
+				Requests: v1.ResourceList{
+					v1.ResourceCPU:    resource.MustParse("10"),
+					v1.ResourceMemory: resource.MustParse("12Gi"),
+				},
+			},
+			InitContainers: []v1.Container{
+				{
+					Image: "normalinit",
+					Name:  "container",
+				},
+				{
+					Image:         "sidecar",
+					Name:          "container",
+					RestartPolicy: &restartAlways,
+				},
+			},
+			Containers: []v1.Container{
+				{
+					Image: "test-image",
+					Name:  "container",
+				},
+			},
+		},
+	}
+	return p
+}
+
 func TestNewPod(t *testing.T) {
 	pod := testPod("default", "mypod")
 	pod.Spec.NodeName = "mynode"
@@ -100,6 +140,15 @@ func TestNewPod(t *testing.T) {
 	if exp, got := resource.MustParse("2Gi"), p.Requested()[v1.ResourceMemory]; exp.Cmp(got) != 0 {
 		t.Errorf("expected Memory = %s, got %s", exp.String(), got.String())
 	}
+
+	podWithPodLevelResources := testPodWithPodLevelResources("default", "mypod")
+	pWithPodLevelResources := model.NewPod(podWithPodLevelResources)
+	if exp, got := resource.MustParse("10"), pWithPodLevelResources.Requested()[v1.ResourceCPU]; exp.Cmp(got) != 0 {
+		t.Errorf("expected CPU = %s, got %s", exp.String(), got.String())
+	}
+	if exp, got := resource.MustParse("12Gi"), pWithPodLevelResources.Requested()[v1.ResourceMemory]; exp.Cmp(got) != 0 {
+		t.Errorf("expected Memory = %s, got %s", exp.String(), got.String())
+	}
 }
 
 func TestPodUpdate(t *testing.T) {