[YUNIKORN-3004] Lock User Trackers and Group Trackers map during dele…

…te operations (#1007) Closes: #1007 Signed-off-by: Manikandan R <[email protected]>
apache · Dec 21, 2024 · 2113cd7 · 2113cd7
1 parent 72a2553
commit 2113cd7
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/pkg/scheduler/ugm/manager.go b/pkg/scheduler/ugm/manager.go
@@ -159,7 +159,9 @@ func (m *Manager) DecreaseTrackedResource(queuePath, applicationID string, usage
 	if userTracker.decreaseTrackedResource(queuePath, applicationID, usage, removeApp) {
 		log.Log(log.SchedUGM).Info("Removing user from manager",
 			zap.String("user", user.User))
+		m.Lock()
 		delete(m.userTrackers, user.User)
+		m.Unlock()
 	}
 	// if the app did not have a group we're done otherwise update the groupTracker
 	if appGroup == common.Empty {
@@ -184,7 +186,9 @@ func (m *Manager) DecreaseTrackedResource(queuePath, applicationID string, usage
 			zap.String("queue path", queuePath),
 			zap.String("application", applicationID),
 			zap.Bool("removeApp", removeApp))
+		m.Lock()
 		delete(m.groupTrackers, appGroup)
+		m.Unlock()
 	}
 }