mirror of
https://github.com/k3s-io/kubernetes.git
synced 2025-07-25 04:33:26 +00:00
82 lines
3.2 KiB
Go
82 lines
3.2 KiB
Go
/*
|
|
Copyright 2016 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 util
|
|
|
|
import (
|
|
v1 "k8s.io/api/core/v1"
|
|
"k8s.io/apimachinery/pkg/api/resource"
|
|
)
|
|
|
|
// For each of these resources, a container that doesn't request the resource explicitly
|
|
// will be treated as having requested the amount indicated below, for the purpose
|
|
// of computing priority only. This ensures that when scheduling zero-request pods, such
|
|
// pods will not all be scheduled to the node with the smallest in-use request,
|
|
// and that when scheduling regular pods, such pods will not see zero-request pods as
|
|
// consuming no resources whatsoever. We chose these values to be similar to the
|
|
// resources that we give to cluster addon pods (#10653). But they are pretty arbitrary.
|
|
// As described in #11713, we use request instead of limit to deal with resource requirements.
|
|
const (
|
|
// DefaultMilliCPURequest defines default milli cpu request number.
|
|
DefaultMilliCPURequest int64 = 100 // 0.1 core
|
|
// DefaultMemoryRequest defines default memory request size.
|
|
DefaultMemoryRequest int64 = 200 * 1024 * 1024 // 200 MB
|
|
)
|
|
|
|
// GetNonzeroRequests returns the default cpu in milli-cpu and memory in bytes resource requests if none is found or
|
|
// what is provided on the request.
|
|
func GetNonzeroRequests(requests *v1.ResourceList) (int64, int64) {
|
|
cpu := GetRequestForResource(v1.ResourceCPU, requests, true)
|
|
mem := GetRequestForResource(v1.ResourceMemory, requests, true)
|
|
return cpu.MilliValue(), mem.Value()
|
|
|
|
}
|
|
|
|
// GetRequestForResource returns the requested values unless nonZero is true and there is no defined request
|
|
// for CPU and memory.
|
|
// If nonZero is true and the resource has no defined request for CPU or memory, it returns a default value.
|
|
func GetRequestForResource(resourceName v1.ResourceName, requests *v1.ResourceList, nonZero bool) resource.Quantity {
|
|
if requests == nil {
|
|
return resource.Quantity{}
|
|
}
|
|
switch resourceName {
|
|
case v1.ResourceCPU:
|
|
// Override if un-set, but not if explicitly set to zero
|
|
if _, found := (*requests)[v1.ResourceCPU]; !found && nonZero {
|
|
return *resource.NewMilliQuantity(DefaultMilliCPURequest, resource.DecimalSI)
|
|
}
|
|
return requests.Cpu().DeepCopy()
|
|
case v1.ResourceMemory:
|
|
// Override if un-set, but not if explicitly set to zero
|
|
if _, found := (*requests)[v1.ResourceMemory]; !found && nonZero {
|
|
return *resource.NewQuantity(DefaultMemoryRequest, resource.DecimalSI)
|
|
}
|
|
return requests.Memory().DeepCopy()
|
|
case v1.ResourceEphemeralStorage:
|
|
quantity, found := (*requests)[v1.ResourceEphemeralStorage]
|
|
if !found {
|
|
return resource.Quantity{}
|
|
}
|
|
return quantity.DeepCopy()
|
|
default:
|
|
quantity, found := (*requests)[resourceName]
|
|
if !found {
|
|
return resource.Quantity{}
|
|
}
|
|
return quantity.DeepCopy()
|
|
}
|
|
}
|