Fix Data Race in checkoutSeriesWithLock

m3db · Feb 28, 2021 · 7bf92e8 · 7bf92e8
1 parent da44ea0
commit 7bf92e8
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/dbnode/storage/bootstrap/util.go b/src/dbnode/storage/bootstrap/util.go
@@ -240,6 +240,9 @@ func (a *TestDataAccumulator) checkoutSeriesWithLock(
 	mockSeries.EXPECT().
 		LoadBlock(gomock.Any(), gomock.Any()).
 		DoAndReturn(func(bl block.DatabaseBlock, _ series.WriteType) error {
+			a.Lock()
+			defer a.Unlock()
+
 			reader, err := bl.Stream(context.NewBackground())
 			if err != nil {
 				streamErr = err