7c760fea38
Previously, the state was keyed off of containerID intead of podUID and containerName. Unfortunately, this is no longer possible as we move to a to model where we we allocate CPUs to containers at pod adit time rather than container start time. This patch is the first step towards full migration to the new semantics. Only the unit tests in cpumanager/state are passing. In subsequent commits we will update the CPUManager itself to use these new semantics. This patch also includes code to do migration from the old checkpoint format to the new one, assuming the existence of a ContainerMap with the proper mapping of (containerID)->(podUID, containerName). A subsequent commit will update code in higher layers to make sure that this ContainerMap is made available to this state logic.
59 lines
1.7 KiB
Go
59 lines
1.7 KiB
Go
/*
|
|
Copyright 2017 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 state
|
|
|
|
import (
|
|
"k8s.io/kubernetes/pkg/kubelet/cm/cpuset"
|
|
)
|
|
|
|
// ContainerCPUAssignments type used in cpu manager state
|
|
type ContainerCPUAssignments map[string]map[string]cpuset.CPUSet
|
|
|
|
// Clone returns a copy of ContainerCPUAssignments
|
|
func (as ContainerCPUAssignments) Clone() ContainerCPUAssignments {
|
|
ret := make(ContainerCPUAssignments)
|
|
for pod := range as {
|
|
ret[pod] = make(map[string]cpuset.CPUSet)
|
|
for container, cset := range as[pod] {
|
|
ret[pod][container] = cset
|
|
}
|
|
}
|
|
return ret
|
|
}
|
|
|
|
// Reader interface used to read current cpu/pod assignment state
|
|
type Reader interface {
|
|
GetCPUSet(podUID string, containerName string) (cpuset.CPUSet, bool)
|
|
GetDefaultCPUSet() cpuset.CPUSet
|
|
GetCPUSetOrDefault(podUID string, containerName string) cpuset.CPUSet
|
|
GetCPUAssignments() ContainerCPUAssignments
|
|
}
|
|
|
|
type writer interface {
|
|
SetCPUSet(podUID string, containerName string, cpuset cpuset.CPUSet)
|
|
SetDefaultCPUSet(cpuset cpuset.CPUSet)
|
|
SetCPUAssignments(ContainerCPUAssignments)
|
|
Delete(podUID string, containerName string)
|
|
ClearState()
|
|
}
|
|
|
|
// State interface provides methods for tracking and setting cpu/pod assignment
|
|
type State interface {
|
|
Reader
|
|
writer
|
|
}
|