uses pool mutex to select random agent (#301)

synadia-io · Jun 28, 2024 · 375ee51 · 375ee51
1 parent 5b52b46
commit 375ee51
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/internal/node/workload_mgr.go b/internal/node/workload_mgr.go
@@ -670,6 +670,9 @@ func (w *WorkloadManager) createHostServicesConnection(request *agentapi.DeployR
 
 // Picks a pending agent from the pool that will receive the next deployment
 func (w *WorkloadManager) SelectRandomAgent() (*agentapi.AgentClient, error) {
+	w.poolMutex.Lock()
+	defer w.poolMutex.Unlock()
+
 	if len(w.pendingAgents) == 0 {
 		return nil, errors.New("no available agent client in pool")
 	}