fix: concurrent map iteration and map write

Signed-off-by: Cookie Wang <[email protected]>

pkg/queue/queue.go

@@ -85,6 +85,8 @@ func (r *Memory) Current() (*Counts, error) {
 	r.mut.RLock()
 	defer r.mut.RUnlock()
 	cts := NewCounts()
-	cts.Counts = r.countMap
+	for key, val := range r.countMap {
+		cts.Counts[key] = val
+	}
 	return cts, nil
 }

pkg/queue/queue_test.go

@@ -0,0 +1,21 @@
+package queue
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/require"
+)
+
+func TestCurrent(t *testing.T) {
+	r := require.New(t)
+	memory := NewMemory()
+	memory.Resize("host1", 1)
+	memory.Resize("host2", 1)
+	current, err := memory.Current()
+	r.NoError(err)
+	r.Equal(current.Counts, memory.countMap)
+
+	memory.Resize("host1", 1)
+	memory.Resize("host3", 1)
+	r.NotEqual(current.Counts, memory.countMap)
+}