refactor: move logger.Debug call to waitForShutdownSignal

cmd/twhelp/cmd_consumer.go

@@ -391,7 +391,7 @@ func runConsumer(c *cli.Context, name string, registerHandlers registerConsumerH
 			logger.Info("consumer is up and running", slog.String("name", name))
 		}
 
-		waitForShutdownSignal(ctx)
+		waitForShutdownSignal(ctx, logger)
 
 		if closeErr := router.Close(); closeErr != nil {
 			logger.Warn("couldn't close router", slog.Any("error", err))

cmd/twhelp/cmd_serve.go

@@ -175,9 +175,7 @@ var cmdServe = &cli.Command{
 
 		idleConnsClosed := make(chan struct{})
 		go func() {
-			waitForShutdownSignal(c.Context)
-
-			logger.Debug("received shutdown signal")
+			waitForShutdownSignal(c.Context, logger)
 
 			ctx, cancel := context.WithTimeout(c.Context, c.Duration(apiServerShutdownTimeoutFlag.Name))
 			defer cancel()

cmd/twhelp/utils.go

@@ -2,6 +2,7 @@ package main
 
 import (
 	"context"
+	"log/slog"
 	"os"
 	"os/signal"
 	"syscall"
@@ -15,8 +16,9 @@ func newShutdownSignalContext(parent context.Context) (context.Context, context.
 	return signal.NotifyContext(parent, shutdownSignals...)
 }
 
-func waitForShutdownSignal(ctx context.Context) {
+func waitForShutdownSignal(ctx context.Context, logger *slog.Logger) {
 	ctx, cancel := newShutdownSignalContext(ctx)
 	defer cancel()
 	<-ctx.Done()
+	logger.Debug("received shutdown signal")
 }