@@ -21,7 +21,7 @@ import (
2121 "fmt"
2222 "strings"
2323
24- core "k8s.io/api/core/v1"
24+ corev1 "k8s.io/api/core/v1"
2525 "k8s.io/apimachinery/pkg/api/resource"
2626
2727 resource_admission "k8s.io/autoscaler/vertical-pod-autoscaler/pkg/admission-controller/resource"
@@ -57,7 +57,7 @@ func (*resourcesUpdatesPatchCalculator) PatchResourceTarget() PatchResourceTarge
5757 return Pod
5858}
5959
60- func (c *resourcesUpdatesPatchCalculator) CalculatePatches(pod *core .Pod, vpa *vpa_types.VerticalPodAutoscaler) ([]resource_admission.PatchRecord, error) {
60+ func (c *resourcesUpdatesPatchCalculator) CalculatePatches(pod *corev1 .Pod, vpa *vpa_types.VerticalPodAutoscaler) ([]resource_admission.PatchRecord, error) {
6161 result := []resource_admission.PatchRecord{}
6262
6363 containersResources, annotationsPerContainer, err := c.recommendationProvider.GetContainersResourcesForPod(pod, vpa)
@@ -110,7 +110,7 @@ func (c *resourcesUpdatesPatchCalculator) CalculatePatches(pod *core.Pod, vpa *v
110110 return result, nil
111111}
112112
113- func getContainerPatch(pod *core .Pod, i int, annotationsPerContainer vpa_api_util.ContainerToAnnotationsMap, containerResources vpa_api_util.ContainerResources) ([]resource_admission.PatchRecord, string) {
113+ func getContainerPatch(pod *corev1 .Pod, i int, annotationsPerContainer vpa_api_util.ContainerToAnnotationsMap, containerResources vpa_api_util.ContainerResources) ([]resource_admission.PatchRecord, string) {
114114 var patches []resource_admission.PatchRecord
115115 // Add empty resources object if missing.
116116 requests, limits := resourcehelpers.ContainerRequestsAndLimits(pod.Spec.Containers[i].Name, pod)
@@ -130,7 +130,7 @@ func getContainerPatch(pod *core.Pod, i int, annotationsPerContainer vpa_api_uti
130130 return patches, updatesAnnotation
131131}
132132
133- func appendPatchesAndAnnotations(patches []resource_admission.PatchRecord, annotations []string, current core .ResourceList, containerIndex int, resources core .ResourceList, fieldName, resourceName string) ([]resource_admission.PatchRecord, []string) {
133+ func appendPatchesAndAnnotations(patches []resource_admission.PatchRecord, annotations []string, current corev1 .ResourceList, containerIndex int, resources corev1 .ResourceList, fieldName, resourceName string) ([]resource_admission.PatchRecord, []string) {
134134 // Add empty object if it's missing and we're about to fill it.
135135 if current == nil && len(resources) > 0 {
136136 patches = append(patches, GetPatchInitializingEmptyResourcesSubfield(containerIndex, fieldName))
@@ -142,7 +142,7 @@ func appendPatchesAndAnnotations(patches []resource_admission.PatchRecord, annot
142142 return patches, annotations
143143}
144144
145- func (c *resourcesUpdatesPatchCalculator) applyCPUStartupBoost(container *core .Container, vpa *vpa_types.VerticalPodAutoscaler, containerResources *vpa_api_util.ContainerResources) ([]resource_admission.PatchRecord, error) {
145+ func (c *resourcesUpdatesPatchCalculator) applyCPUStartupBoost(container *corev1 .Container, vpa *vpa_types.VerticalPodAutoscaler, containerResources *vpa_api_util.ContainerResources) ([]resource_admission.PatchRecord, error) {
146146 var patches []resource_admission.PatchRecord
147147
148148 startupBoostPolicy := getContainerStartupBoostPolicy(container, vpa)
@@ -164,7 +164,7 @@ func (c *resourcesUpdatesPatchCalculator) applyCPUStartupBoost(container *core.C
164164 return patches, nil
165165}
166166
167- func getContainerStartupBoostPolicy(container *core .Container, vpa *vpa_types.VerticalPodAutoscaler) *vpa_types.StartupBoost {
167+ func getContainerStartupBoostPolicy(container *corev1 .Container, vpa *vpa_types.VerticalPodAutoscaler) *vpa_types.StartupBoost {
168168 policy := vpa_api_util.GetContainerResourcePolicy(container.Name, vpa.Spec.ResourcePolicy)
169169 startupBoost := vpa.Spec.StartupBoost
170170 if policy != nil && policy.StartupBoost != nil {
@@ -226,52 +226,52 @@ func (c *resourcesUpdatesPatchCalculator) calculateBoostedCPU(recommendedCPU, or
226226
227227// capStartupBoostToContainerLimit makes sure startup boost recommendation is not above current limit for the container for CPU.
228228// It attempts to keep the request 1m below the limit to maintain QoS.
229- func capStartupBoostToContainerLimit(recommendation core .ResourceList, containerLimits core .ResourceList) {
230- limit, found := containerLimits[core .ResourceCPU]
229+ func capStartupBoostToContainerLimit(recommendation corev1 .ResourceList, containerLimits corev1 .ResourceList) {
230+ limit, found := containerLimits[corev1 .ResourceCPU]
231231 if !found {
232232 return
233233 }
234234
235- recommendedValue, found := recommendation[core .ResourceCPU]
235+ recommendedValue, found := recommendation[corev1 .ResourceCPU]
236236 if found && recommendedValue.MilliValue() > limit.MilliValue() {
237237 newRecommended := limit.DeepCopy()
238238 if limit.Cmp(resource.MustParse("1m")) > 0 {
239239 newRecommended.Sub(resource.MustParse("1m"))
240240 }
241- recommendation[core .ResourceCPU] = newRecommended
241+ recommendation[corev1 .ResourceCPU] = newRecommended
242242 }
243243}
244244
245- func (c *resourcesUpdatesPatchCalculator) applyControlledCPUResources(container *core .Container, vpa *vpa_types.VerticalPodAutoscaler, containerResources *vpa_api_util.ContainerResources, startupBoostPolicy *vpa_types.StartupBoost) error {
245+ func (c *resourcesUpdatesPatchCalculator) applyControlledCPUResources(container *corev1 .Container, vpa *vpa_types.VerticalPodAutoscaler, containerResources *vpa_api_util.ContainerResources, startupBoostPolicy *vpa_types.StartupBoost) error {
246246 controlledValues := vpa_api_util.GetContainerControlledValues(container.Name, vpa.Spec.ResourcePolicy)
247247
248- recommendedRequest := containerResources.Requests[core .ResourceCPU]
249- originalRequest := container.Resources.Requests[core .ResourceCPU]
248+ recommendedRequest := containerResources.Requests[corev1 .ResourceCPU]
249+ originalRequest := container.Resources.Requests[corev1 .ResourceCPU]
250250 boostedRequest, err := c.calculateBoostedCPU(recommendedRequest, originalRequest, startupBoostPolicy)
251251 if err != nil {
252252 return err
253253 }
254254
255255 if containerResources.Requests == nil {
256- containerResources.Requests = core .ResourceList{}
256+ containerResources.Requests = corev1 .ResourceList{}
257257 }
258- containerResources.Requests[core .ResourceCPU] = *boostedRequest
258+ containerResources.Requests[corev1 .ResourceCPU] = *boostedRequest
259259
260260 switch controlledValues {
261261 case vpa_types.ContainerControlledValuesRequestsOnly:
262262 capStartupBoostToContainerLimit(containerResources.Requests, container.Resources.Limits)
263263 case vpa_types.ContainerControlledValuesRequestsAndLimits:
264264 if containerResources.Limits == nil {
265- containerResources.Limits = core .ResourceList{}
265+ containerResources.Limits = corev1 .ResourceList{}
266266 }
267267 newLimits, _ := vpa_api_util.GetProportionalLimit(
268- container.Resources.Limits, // originalLimits
269- container.Resources.Requests, // originalRequests
270- core .ResourceList{core .ResourceCPU: *boostedRequest}, // newRequests
271- core .ResourceList{}, // defaultLimit
268+ container.Resources.Limits, // originalLimits
269+ container.Resources.Requests, // originalRequests
270+ corev1 .ResourceList{corev1 .ResourceCPU: *boostedRequest}, // newRequests
271+ corev1 .ResourceList{}, // defaultLimit
272272 )
273- if newLimit, ok := newLimits[core .ResourceCPU]; ok {
274- containerResources.Limits[core .ResourceCPU] = newLimit
273+ if newLimit, ok := newLimits[corev1 .ResourceCPU]; ok {
274+ containerResources.Limits[corev1 .ResourceCPU] = newLimit
275275 }
276276 default:
277277 // Do nothing
0 commit comments