quick fix

main.go

@@ -31,8 +31,9 @@ var (
 	signalFlag = flag.String("s", "", `Send signal to the daemon:
   drain — graceful shutdown, will wait until all jobs finish before exiting
   stop — best effort graceful shutdown, interrupting the job as soon as possible`)
-	stop = make(chan struct{})
-	done = make(chan struct{})
+	stop            = make(chan struct{})
+	done            = make(chan struct{})
+	shutDownMessage = "anklet service shut down"
 )
 
 func termHandler(ctx context.Context, logger *slog.Logger) daemon.SignalHandlerFunc {
@@ -130,8 +131,6 @@ func main() {
 	if err != nil {
 		log.Printf("Error: %s", err.Error())
 	}
-
-	logger.InfoContext(ctx, "anklet terminated")
 }
 
 func worker(ctx context.Context, logger *slog.Logger, rateLimiter *http.Client, loadedConfig config.Config) {
@@ -201,20 +200,24 @@ func worker(ctx context.Context, logger *slog.Logger, rateLimiter *http.Client,
 				})
 			}
 
+			logger.InfoContext(serviceCtx, "started service")
+
 			for {
 				select {
 				case <-serviceCtx.Done():
-					logger.WarnContext(serviceCtx, "service shut down")
-					os.Exit(0)
+					logger.WarnContext(serviceCtx, shutDownMessage)
+					return
 				default:
 					run.Plugin(serviceCtx, logger)
 					if ctx.Err() != nil || runOnce == "true" {
-						logger.WarnContext(serviceCtx, "service shut down")
-						os.Exit(0)
+						logger.WarnContext(serviceCtx, shutDownMessage)
+						return
 					}
 					time.Sleep(time.Duration(service.SleepInterval) * time.Second)
 				}
 			}
 		}(service)
 	}
+	wg.Wait()
+	os.Exit(0)
 }