Fix #803

onsi · Dec 10, 2024 · 1c6c112 · 1c6c112
1 parent 4feb9d7
commit 1c6c112
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/gmeasure/experiment.go b/gmeasure/experiment.go
@@ -463,13 +463,19 @@ func (e *Experiment) Sample(callback func(idx int), samplingConfig SamplingConfi
 	minSamplingInterval := samplingConfig.MinSamplingInterval
 
 	work := make(chan int)
-	defer close(work)
+	var wg sync.WaitGroup
+	defer func() {
+		close(work)
+		wg.Wait()
+	}()
 	if numParallel > 1 {
 		for worker := 0; worker < numParallel; worker++ {
 			go func() {
+				wg.Add(1)
 				for idx := range work {
 					callback(idx)
 				}
+				wg.Done()
 			}()
 		}
 	}