From 479cf41ea02a7903cfc0da9c74519f5c28c6b5b1 Mon Sep 17 00:00:00 2001
From: Joel Wetzell <git@jwetzell.com>
Date: Fri, 26 Dec 2025 10:24:15 -0600
Subject: [PATCH] use WaitGroup Go function

---
 router.go | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/router.go b/router.go
index f7cf44b..9a1c2cb 100644
--- a/router.go
+++ b/router.go
@@ -109,14 +109,12 @@ func NewRouter(ctx context.Context, config config.Config) (*Router, []module.Mod
 func (r *Router) Run() {
 	r.logger.Info("running")
 	for _, moduleInstance := range r.ModuleInstances {
-		r.moduleWait.Add(1)
-		go func() {
+		r.moduleWait.Go(func() {
 			err := moduleInstance.Run()
 			if err != nil {
 				r.logger.Error("error encountered running module", "error", err)
 			}
-			r.moduleWait.Done()
-		}()
+		})
 	}
 	<-r.Context.Done()
 	r.moduleWait.Wait()