chore: improve buffer functionality

Author: cisse21

router/batchrouter/handle_lifecycle.go

@@ -166,6 +166,7 @@ func (brt *Handle) Setup(
 	brt.jobBuffer = &JobBuffer{
 		sourceDestMap: make(map[string]chan *jobsdb.JobT),
 		uploadTimers:  make(map[string]*time.Timer),
+		batchWorkers:  make(map[string]*batchWorker),
 		brt:           brt,
 	}
 
@@ -273,14 +274,12 @@ func (brt *Handle) Shutdown() {
 	brt.logger.Info("Initiating batch router shutdown")
 	brt.backgroundCancel()
 
-	// Stop all job buffer timers
+	// Stop all job buffer timers and workerPool
 	if brt.jobBuffer != nil {
-		brt.jobBuffer.mu.Lock()
-		for key, timer := range brt.jobBuffer.uploadTimers {
-			timer.Stop()
-			brt.logger.Debugf("Stopped timer for source-destination: %s", key)
-		}
-		brt.jobBuffer.mu.Unlock()
+		brt.logger.Debug("Stopping JobBuffer and its resources")
+
+		// Stop the job buffer (which will also stop the worker pool)
+		brt.jobBuffer.Stop()
 	}
 
 	// Wait for all background goroutines to complete