defer wg.Done

Co-authored-by: Vegar Sechmann Molvig <[email protected]>
nais · Dec 22, 2023 · 78e00b1 · 78e00b1
1 parent febc0fe
commit 78e00b1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/internal/device-agent/states/connected/connected.go b/internal/device-agent/states/connected/connected.go
@@ -281,6 +281,7 @@ func (c *Connected) launchHealthCheck(ctx context.Context) context.CancelFunc {
 				for i, gw := range gateways {
 					wg.Add(1)
 					go func(i int, gw *pb.Gateway) {
+						defer wg.Done()
 						err := ping(c.logger, gw.Ipv4)
 						pos := fmt.Sprintf("[%02d/%02d]", i+1, total)
 						if err == nil {
@@ -290,7 +291,6 @@ func (c *Connected) launchHealthCheck(ctx context.Context) context.CancelFunc {
 							gw.Healthy = false
 							c.logger.Debugf("%s %s: unable to ping %s: %v", pos, gw.Name, gw.Ipv4, err)
 						}
-						wg.Done()
 					}(i, gw)
 				}
 				wg.Wait()