wg.Add len of decorators

vbauerster · Aug 16, 2024 · d8ed1db · d8ed1db
1 parent a4b4b53
commit d8ed1db
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/bar.go b/bar.go
@@ -581,8 +581,8 @@ func (s bState) completed() bool {
 }
 
 func (s bState) decoratorEwmaUpdate(n int64, dur time.Duration, wg *sync.WaitGroup) {
+	wg.Add(len(s.ewmaDecorators))
 	for _, d := range s.ewmaDecorators {
-		wg.Add(1)
 		d := d
 		go func() {
 			d.EwmaUpdate(n, dur)
@@ -592,8 +592,8 @@ func (s bState) decoratorEwmaUpdate(n int64, dur time.Duration, wg *sync.WaitGro
 }
 
 func (s bState) decoratorAverageAdjust(start time.Time, wg *sync.WaitGroup) {
+	wg.Add(len(s.averageDecorators))
 	for _, d := range s.averageDecorators {
-		wg.Add(1)
 		d := d
 		go func() {
 			d.AverageAdjust(start)
@@ -603,8 +603,8 @@ func (s bState) decoratorAverageAdjust(start time.Time, wg *sync.WaitGroup) {
 }
 
 func (s bState) decoratorShutdownNotify(wg *sync.WaitGroup) {
+	wg.Add(len(s.shutdownListeners))
 	for _, d := range s.shutdownListeners {
-		wg.Add(1)
 		d := d
 		go func() {
 			d.OnShutdown()