mirror of
https://github.com/k3s-io/kubernetes.git
synced 2026-07-24 03:40:56 +00:00
71 lines
2.5 KiB
Go
71 lines
2.5 KiB
Go
/*
|
|
Copyright 2025 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 cgroups
|
|
|
|
import (
|
|
"math"
|
|
"strconv"
|
|
|
|
v1 "k8s.io/api/core/v1"
|
|
kubecm "k8s.io/kubernetes/pkg/kubelet/cm"
|
|
)
|
|
|
|
// convertCPUSharesToCgroupV2Value is copied from ConvertCPUSharesToCgroupV2Value in opencontainers/cgroups.
|
|
// https://github.com/opencontainers/cgroups/pull/20/files
|
|
func convertCPUSharesToCgroupV2Value(cpuShares uint64) uint64 {
|
|
// The value of 0 means "unset".
|
|
if cpuShares == 0 {
|
|
return 0
|
|
}
|
|
if cpuShares <= 2 {
|
|
return 1
|
|
}
|
|
if cpuShares >= 262144 {
|
|
return 10000
|
|
}
|
|
l := math.Log2(float64(cpuShares))
|
|
// Quadratic function which fits min, max, and default.
|
|
exponent := (l*l+125*l)/612.0 - 7.0/34.0
|
|
|
|
return uint64(math.Ceil(math.Pow(10, exponent)))
|
|
}
|
|
|
|
func getExpectedCPUShares(rr *v1.ResourceRequirements, podOnCgroupv2 bool) []string {
|
|
// This function is moved out from cgroups.go because opencontainers/cgroups can only be compiled in linux platforms.
|
|
cpuRequest := rr.Requests.Cpu()
|
|
cpuLimit := rr.Limits.Cpu()
|
|
var shares int64
|
|
if cpuRequest.IsZero() && !cpuLimit.IsZero() {
|
|
shares = int64(kubecm.MilliCPUToShares(cpuLimit.MilliValue()))
|
|
} else {
|
|
shares = int64(kubecm.MilliCPUToShares(cpuRequest.MilliValue()))
|
|
}
|
|
if podOnCgroupv2 {
|
|
// Because of https://github.com/kubernetes/kubernetes/issues/131216, the way of conversion has been changed.
|
|
// runc: https://github.com/opencontainers/runc/pull/4785
|
|
// crun: https://github.com/containers/crun/issues/1721
|
|
// This is dependent on the container runtime version. In order not to break the tests when we upgrade the
|
|
// container runtimes, we check if either the old or the new conversion matches the actual value for now.
|
|
// TODO: Remove the old conversion once container runtimes are updated.
|
|
oldConverted := 1 + ((shares-2)*9999)/262142
|
|
converted := convertCPUSharesToCgroupV2Value(uint64(shares))
|
|
return []string{strconv.FormatInt(oldConverted, 10), strconv.FormatInt(int64(converted), 10)}
|
|
} else {
|
|
return []string{strconv.FormatInt(shares, 10)}
|
|
}
|
|
}
|