add wait group to start

Sellsuki · Oct 12, 2023 · 600f430 · 600f430
1 parent 1627db8
commit 600f430
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/worker_manager.go b/worker_manager.go
@@ -14,8 +14,10 @@ type KafkaWorkerManager struct {
 }
 
 // Start all workers
-func (k KafkaWorkerManager) Start() []error {
-	wg := sync.WaitGroup{}
+func (k KafkaWorkerManager) Start(wg *sync.WaitGroup) []error {
+	if wg == nil {
+		wg = &sync.WaitGroup{}
+	}
 	ctx, cancel := context.WithCancel(context.Background())
 
 	// Wait for OS SIGINT/SIGTERM then gracefully shutdown the workers