fix: data race in bloombuild tests (#15360)

1 year ago · aa7b849e75
parent 42469ccece
commit aa7b849e75
1 changed files with 4 additions and 1 deletions
--- a/pkg/bloombuild/planner/planner_test.go
+++ b/pkg/bloombuild/planner/planner_test.go
@ -713,7 +713,10 @@ func (f *fakeBuilder) Send(req *protos.PlannerToBuilder) error {
 }

 func (f *fakeBuilder) Recv() (*protos.BuilderToPlanner, error) {
-	if len(f.tasks) == 0 {
+	f.mx.Lock()
+	tasksLen := len(f.tasks)
+	f.mx.Unlock()
+	if tasksLen == 0 {
 		// First call to Recv answers with builderID
 		return &protos.BuilderToPlanner{
 			BuilderID: f.id,