mirror of
https://github.com/kubernetes/kubernetes.git
synced 2026-07-12 03:05:53 -04:00
c7292e0a74
Add the Windows node e2e CPU affinity tests and the supporting suite plumbing: fix the build and drop the dependency on test/e2e_node by porting the kubeletconfig, criproxy and testing-manifests subpackages locally (CRI proxy uses a Windows named pipe).
94 lines
2.9 KiB
Go
94 lines
2.9 KiB
Go
//go:build windows
|
|
|
|
/*
|
|
Copyright 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 e2enodewindows
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"math/bits"
|
|
"strings"
|
|
|
|
v1 "k8s.io/api/core/v1"
|
|
internalapi "k8s.io/cri-api/pkg/apis"
|
|
runtimeapi "k8s.io/cri-api/pkg/apis/runtime/v1"
|
|
)
|
|
|
|
// containerIDForContainer returns the runtime container ID for ctnName
|
|
// inside pod, with the runtime scheme prefix ("containerd://") stripped.
|
|
func containerIDForContainer(pod *v1.Pod, ctnName string) (string, error) {
|
|
for _, cs := range append(pod.Status.ContainerStatuses, pod.Status.InitContainerStatuses...) {
|
|
if cs.Name != ctnName {
|
|
continue
|
|
}
|
|
_, id, found := strings.Cut(cs.ContainerID, "://")
|
|
if !found {
|
|
return "", fmt.Errorf("unsupported containerID format: %q", cs.ContainerID)
|
|
}
|
|
return id, nil
|
|
}
|
|
return "", fmt.Errorf("container %q not found in pod %s/%s status", ctnName, pod.Namespace, pod.Name)
|
|
}
|
|
|
|
// getWindowsContainerCPUAffinity queries the CRI for the CPU group affinities
|
|
// currently set on the named container. Returns nil when no affinity is set.
|
|
func getWindowsContainerCPUAffinity(
|
|
ctx context.Context,
|
|
criClient internalapi.RuntimeService,
|
|
pod *v1.Pod,
|
|
ctnName string,
|
|
) ([]*runtimeapi.WindowsCpuGroupAffinity, error) {
|
|
cntID, err := containerIDForContainer(pod, ctnName)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
resp, err := criClient.ContainerStatus(ctx, cntID, false)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("ContainerStatus(%q): %w", cntID, err)
|
|
}
|
|
if resp == nil || resp.Status == nil || resp.Status.Resources == nil || resp.Status.Resources.Windows == nil {
|
|
return nil, nil
|
|
}
|
|
return resp.Status.Resources.Windows.AffinityCpus, nil
|
|
}
|
|
|
|
// countCPUsInAffinities returns the total number of logical CPUs represented
|
|
// across all WindowsCpuGroupAffinity entries by counting set bits in each mask.
|
|
func countCPUsInAffinities(affinities []*runtimeapi.WindowsCpuGroupAffinity) int {
|
|
total := 0
|
|
for _, a := range affinities {
|
|
total += bits.OnesCount64(a.CpuMask)
|
|
}
|
|
return total
|
|
}
|
|
|
|
// windowsAffinitiesOverlap returns true when two sets of CPU group affinities
|
|
// share at least one CPU (same bit set in the same processor group).
|
|
func windowsAffinitiesOverlap(a, b []*runtimeapi.WindowsCpuGroupAffinity) bool {
|
|
maskA := make(map[uint32]uint64, len(a))
|
|
for _, aff := range a {
|
|
maskA[aff.CpuGroup] |= aff.CpuMask
|
|
}
|
|
for _, aff := range b {
|
|
if maskA[aff.CpuGroup]&aff.CpuMask != 0 {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|