mirror of https://github.com/k3s-io/k3s
84 lines
1.7 KiB
Go
84 lines
1.7 KiB
Go
/*
|
|
Copyright The containerd 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 v2
|
|
|
|
import (
|
|
"math"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
type CPUMax string
|
|
|
|
func NewCPUMax(quota *int64, period *uint64) CPUMax {
|
|
max := "max"
|
|
if quota != nil {
|
|
max = strconv.FormatInt(*quota, 10)
|
|
}
|
|
return CPUMax(strings.Join([]string{max, strconv.FormatUint(*period, 10)}, " "))
|
|
}
|
|
|
|
type CPU struct {
|
|
Weight *uint64
|
|
Max CPUMax
|
|
Cpus string
|
|
Mems string
|
|
}
|
|
|
|
func (c CPUMax) extractQuotaAndPeriod() (int64, uint64) {
|
|
var (
|
|
quota int64
|
|
period uint64
|
|
)
|
|
values := strings.Split(string(c), " ")
|
|
if values[0] == "max" {
|
|
quota = math.MaxInt64
|
|
} else {
|
|
quota, _ = strconv.ParseInt(values[0], 10, 64)
|
|
}
|
|
period, _ = strconv.ParseUint(values[1], 10, 64)
|
|
return quota, period
|
|
}
|
|
|
|
func (r *CPU) Values() (o []Value) {
|
|
if r.Weight != nil {
|
|
o = append(o, Value{
|
|
filename: "cpu.weight",
|
|
value: *r.Weight,
|
|
})
|
|
}
|
|
if r.Max != "" {
|
|
o = append(o, Value{
|
|
filename: "cpu.max",
|
|
value: r.Max,
|
|
})
|
|
}
|
|
if r.Cpus != "" {
|
|
o = append(o, Value{
|
|
filename: "cpuset.cpus",
|
|
value: r.Cpus,
|
|
})
|
|
}
|
|
if r.Mems != "" {
|
|
o = append(o, Value{
|
|
filename: "cpuset.mems",
|
|
value: r.Mems,
|
|
})
|
|
}
|
|
return o
|
|
}
|