From 1c74525b8d551fa7865274443aec3b67608cc8a2 Mon Sep 17 00:00:00 2001
From: Mikhail Scherba <mikhail.scherba@flant.com>
Date: Mon, 17 Feb 2025 13:18:22 +0300
Subject: [PATCH] refactor queueset locks #2

Signed-off-by: Mikhail Scherba <mikhail.scherba@flant.com>
---
 pkg/task/queue/queue_set.go | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/pkg/task/queue/queue_set.go b/pkg/task/queue/queue_set.go
index ff756f1a..6924c787 100644
--- 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) {