STAC-21470: Fix duplication

rb3ckers · Aug 21, 2024 · b1e5a1b · b1e5a1b
1 parent 91ac414
commit b1e5a1b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/internal/mirror/reflector.go b/internal/mirror/reflector.go
@@ -78,7 +78,7 @@ func (r *Reflector) AddMirrors(urls []string, persistent bool) {
 
 	for _, url := range urls {
 		log.Printf("Adding '%s' to mirror list.", url)
-		r.mirrors[url] = NewMirror(url, r.config, r.MirrorFailureChan, persistent, r.templateSendQueue)
+		r.mirrors[url] = NewMirror(url, r.config, r.MirrorFailureChan, persistent, r.templateSendQueue.Clone())
 	}
 }
 

diff --git a/internal/mirror/sendqueue.go b/internal/mirror/sendqueue.go
@@ -30,6 +30,9 @@ func MakeSendQueue(maxQueueSize int) *SendQueue {
 }
 
 func (s *SendQueue) Clone() *SendQueue {
+	s.Lock()
+	defer s.Unlock()
+
 	completedCopied := make(map[uint64]interface{}, 0)
 	for c, _ := range s.epochsCompleted {
 		completedCopied[c] = nil