Function logSignalNotifyContext

cli/agent.go:628–647

logSignalNotifyContext is like signal.NotifyContext but logs the received signal before canceling the context.

(parent context.Context, logger slog.Logger, signals ...os.Signal)

// logSignalNotifyContext is like signal.NotifyContext but logs the received
// signal before canceling the context.
func logSignalNotifyContext(parent context.Context, logger slog.Logger, signals ...os.Signal) (context.Context, context.CancelFunc) {
	ctx, cancel := context.WithCancelCause(parent)
	c := make(chan os.Signal, 1)
	signal.Notify(c, signals...)

	go func() {
		select {
		case sig := <-c:
			logger.Info(ctx, "agent received signal", slog.F("signal", sig.String()))
			cancel(xerrors.Errorf("signal: %s", sig.String()))
		case <-ctx.Done():
			logger.Info(ctx, "ctx canceled, stopping signal handler")
		}
	}()

	return ctx, func() {
		cancel(context.Canceled)
		signal.Stop(c)
	}
}