using map instead

Signed-off-by: zounengren <zouyee1989@gmail.com>
gocrane · Jul 8, 2022 · 7d8cb3d · 7d8cb3d
1 parent abdb63c
commit 7d8cb3d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pkg/controller/evpa/container_policy.go b/pkg/controller/evpa/container_policy.go
@@ -29,6 +29,7 @@ func (c *EffectiveVPAController) ReconcileContainerPolicies(evpa *autoscalingapi
 
 	rankedEstimators := RankEstimators(resourceEstimators)
 	needReconciledContainers := make(map[string]autoscalingapi.ContainerResourcePolicy)
+	containerResourceRequirement := make(map[string]*corev1.ResourceRequirements)
 	for _, container := range podTemplate.Spec.Containers {
 		for _, containerPolicy := range evpa.Spec.ResourcePolicy.ContainerPolicies {
 			if containerPolicy.ContainerName == "*" || containerPolicy.ContainerName == container.Name {
@@ -37,10 +38,11 @@ func (c *EffectiveVPAController) ReconcileContainerPolicies(evpa *autoscalingapi
 				needReconciledContainers[container.Name] = *out
 			}
 		}
+		containerResourceRequirement[container.Name] = &container.Resources
 	}
 	for container, containerPolicy := range needReconciledContainers {
 		// get current resource by pod template
-		resourceRequirement, found := utils.GetResourceByPodTemplate(podTemplate, container)
+		resourceRequirement, found := containerResourceRequirement[container]
 		if !found {
 			klog.Warningf("ContainerName %s not found", containerPolicy.ContainerName)
 			continue