github/kubernetes
1
0
Fork 0
mirror of https://github.com/k3s-io/kubernetes.git synced 2025-08-17 15:50:10 +00:00
Code Issues Projects Releases Wiki Activity

fix allDevices map data race

Browse Source
This commit is contained in:
huyinhou 2022-12-29 18:27:08 +08:00
parent 997cefc9da
commit b9987eeb6c
3 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pkg/kubelet/cm/devicemanager/manager.go
View File

@ -646,7 +646,7 @@ func (m *ManagerImpl) devicesToAllocate(podUID, contName, resource string, requi
func (m *ManagerImpl) filterByAffinity(podUID, contName, resource string, available sets.String) (sets.String, sets.String, sets.String) {
// If alignment information is not available, just pass the available list back.
hint := m.topologyAffinityStore.GetAffinity(podUID, contName)
if !m.deviceHasTopologyAlignment(resource) || hint.NUMANodeAffinity == nil {
if !m.deviceHasTopologyAlignmentLocked(resource) || hint.NUMANodeAffinity == nil {
return sets.NewString(), sets.NewString(), available
}

8
pkg/kubelet/cm/devicemanager/topology_hints.go
View File

@ -136,7 +136,7 @@ func (m *ManagerImpl) GetPodTopologyHints(pod *v1.Pod) map[string][]topologymana
return deviceHints
}
func (m *ManagerImpl) deviceHasTopologyAlignment(resource string) bool {
func (m *ManagerImpl) deviceHasTopologyAlignmentLocked(resource string) bool {
// If any device has Topology NUMANodes available, we assume they care about alignment.
for _, device := range m.allDevices[resource] {
if device.Topology != nil && len(device.Topology.Nodes) > 0 {
@ -146,6 +146,12 @@ func (m *ManagerImpl) deviceHasTopologyAlignment(resource string) bool {
return false
}
func (m *ManagerImpl) deviceHasTopologyAlignment(resource string) bool {
m.mutex.Lock()
defer m.mutex.Unlock()
return m.deviceHasTopologyAlignmentLocked(resource)
}
func (m *ManagerImpl) getAvailableDevices(resource string) sets.String {
m.mutex.Lock()
defer m.mutex.Unlock()

2
pkg/kubelet/cm/devicemanager/topology_hints_test.go
View File

@ -456,7 +456,7 @@ func TestTopologyAlignedAllocation(t *testing.T) {
}
alignment := make(map[int]int)
if m.deviceHasTopologyAlignment(tc.resource) {
if m.deviceHasTopologyAlignmentLocked(tc.resource) {
for d := range allocated {
if m.allDevices[tc.resource][d].Topology != nil {
alignment[int(m.allDevices[tc.resource][d].Topology.Nodes[0].ID)]++