mirror of
https://github.com/ceph/ceph-csi.git
synced 2024-11-13 01:40:23 +00:00
119 lines
4.1 KiB
Go
119 lines
4.1 KiB
Go
/*
|
|
Copyright 2014 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 resource
|
|
|
|
import (
|
|
"fmt"
|
|
"math"
|
|
"strconv"
|
|
|
|
"k8s.io/apimachinery/pkg/api/resource"
|
|
api "k8s.io/kubernetes/pkg/apis/core"
|
|
)
|
|
|
|
// addResourceList adds the resources in newList to list
|
|
func addResourceList(list, new api.ResourceList) {
|
|
for name, quantity := range new {
|
|
if value, ok := list[name]; !ok {
|
|
list[name] = *quantity.Copy()
|
|
} else {
|
|
value.Add(quantity)
|
|
list[name] = value
|
|
}
|
|
}
|
|
}
|
|
|
|
// maxResourceList sets list to the greater of list/newList for every resource
|
|
// either list
|
|
func maxResourceList(list, new api.ResourceList) {
|
|
for name, quantity := range new {
|
|
if value, ok := list[name]; !ok {
|
|
list[name] = *quantity.Copy()
|
|
continue
|
|
} else {
|
|
if quantity.Cmp(value) > 0 {
|
|
list[name] = *quantity.Copy()
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// PodRequestsAndLimits returns a dictionary of all defined resources summed up for all
|
|
// containers of the pod.
|
|
func PodRequestsAndLimits(pod *api.Pod) (reqs api.ResourceList, limits api.ResourceList) {
|
|
reqs, limits = api.ResourceList{}, api.ResourceList{}
|
|
for _, container := range pod.Spec.Containers {
|
|
addResourceList(reqs, container.Resources.Requests)
|
|
addResourceList(limits, container.Resources.Limits)
|
|
}
|
|
// init containers define the minimum of any resource
|
|
for _, container := range pod.Spec.InitContainers {
|
|
maxResourceList(reqs, container.Resources.Requests)
|
|
maxResourceList(limits, container.Resources.Limits)
|
|
}
|
|
return
|
|
}
|
|
|
|
// ExtractContainerResourceValue extracts the value of a resource
|
|
// in an already known container
|
|
func ExtractContainerResourceValue(fs *api.ResourceFieldSelector, container *api.Container) (string, error) {
|
|
divisor := resource.Quantity{}
|
|
if divisor.Cmp(fs.Divisor) == 0 {
|
|
divisor = resource.MustParse("1")
|
|
} else {
|
|
divisor = fs.Divisor
|
|
}
|
|
|
|
switch fs.Resource {
|
|
case "limits.cpu":
|
|
return convertResourceCPUToString(container.Resources.Limits.Cpu(), divisor)
|
|
case "limits.memory":
|
|
return convertResourceMemoryToString(container.Resources.Limits.Memory(), divisor)
|
|
case "limits.ephemeral-storage":
|
|
return convertResourceEphemeralStorageToString(container.Resources.Limits.StorageEphemeral(), divisor)
|
|
case "requests.cpu":
|
|
return convertResourceCPUToString(container.Resources.Requests.Cpu(), divisor)
|
|
case "requests.memory":
|
|
return convertResourceMemoryToString(container.Resources.Requests.Memory(), divisor)
|
|
case "requests.ephemeral-storage":
|
|
return convertResourceEphemeralStorageToString(container.Resources.Requests.StorageEphemeral(), divisor)
|
|
}
|
|
|
|
return "", fmt.Errorf("unsupported container resource : %v", fs.Resource)
|
|
}
|
|
|
|
// convertResourceCPUToString converts cpu value to the format of divisor and returns
|
|
// ceiling of the value.
|
|
func convertResourceCPUToString(cpu *resource.Quantity, divisor resource.Quantity) (string, error) {
|
|
c := int64(math.Ceil(float64(cpu.MilliValue()) / float64(divisor.MilliValue())))
|
|
return strconv.FormatInt(c, 10), nil
|
|
}
|
|
|
|
// convertResourceMemoryToString converts memory value to the format of divisor and returns
|
|
// ceiling of the value.
|
|
func convertResourceMemoryToString(memory *resource.Quantity, divisor resource.Quantity) (string, error) {
|
|
m := int64(math.Ceil(float64(memory.Value()) / float64(divisor.Value())))
|
|
return strconv.FormatInt(m, 10), nil
|
|
}
|
|
|
|
// convertResourceEphemeralStorageToString converts ephemeral storage value to the format of divisor and returns
|
|
// ceiling of the value.
|
|
func convertResourceEphemeralStorageToString(ephemeralStorage *resource.Quantity, divisor resource.Quantity) (string, error) {
|
|
m := int64(math.Ceil(float64(ephemeralStorage.Value()) / float64(divisor.Value())))
|
|
return strconv.FormatInt(m, 10), nil
|
|
}
|