Merge pull request #1941 from huone1/numa-aware/map-bugfix

[numa-aware]fix concurrent map read and map write
volcano-sh · Jan 11, 2022 · 2a45ed2 · 2a45ed2
2 parents 0c62e0c + 4a1f8a6
commit 2a45ed2
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/pkg/scheduler/plugins/numaaware/numaaware.go b/pkg/scheduler/plugins/numaaware/numaaware.go
@@ -45,6 +45,7 @@ const (
 )
 
 type numaPlugin struct {
+	sync.Mutex
 	// Arguments given for the plugin
 	pluginArguments framework.Arguments
 	hintProviders   []policy.HintProvider
@@ -142,6 +143,8 @@ func (pp *numaPlugin) OnSessionOpen(ssn *framework.Session) {
 			}
 		}
 
+		pp.Lock()
+		defer pp.Unlock()
 		if _, ok := pp.assignRes[task.UID]; !ok {
 			pp.assignRes[task.UID] = make(map[string]api.ResNumaSets)
 		}