go/store/nbs: table_set.go: Fix race on ret map in openForAdd.

dolthub · Feb 11, 2025 · 17d5e24 · 17d5e24
1 parent 14e2c0e
commit 17d5e24
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/go/store/nbs/table_set.go b/go/store/nbs/table_set.go
@@ -464,7 +464,10 @@ func (ts tableSet) openForAdd(ctx context.Context, files map[hash.Hash]uint32, s
 	eg, ctx := errgroup.WithContext(ctx)
 	var mu sync.Mutex
 	for fileId, chunkCount := range files {
-		if _, ok := ret[fileId]; ok {
+		mu.Lock()
+		_, ok := ret[fileId]
+		mu.Unlock()
+		if ok {
 			continue
 		}
 		eg.Go(func() error {