refactor queueset locks #2

Signed-off-by: Mikhail Scherba <mikhail.scherba@flant.com>
flant · Feb 17, 2025 · 1c74525 · 1c74525
1 parent 10fa483
commit 1c74525
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/pkg/task/queue/queue_set.go b/pkg/task/queue/queue_set.go
@@ -116,6 +116,8 @@ func (tqs *TaskQueueSet) Iterate(doFn func(queue *TaskQueue)) {
 		return
 	}
 
+	tqs.m.RLock()
+	defer tqs.m.RUnlock()
 	if len(tqs.Queues) == 0 {
 		return
 	}
@@ -126,14 +128,11 @@ func (tqs *TaskQueueSet) Iterate(doFn func(queue *TaskQueue)) {
 	}
 	// TODO sort names
 
-	tqs.m.RLock()
 	for _, q := range tqs.Queues {
 		if q.Name != tqs.MainName {
 			doFn(q)
 		}
 	}
-
-	tqs.m.RUnlock()
 }
 
 func (tqs *TaskQueueSet) Remove(name string) {