Golang context with cancellation on signal receive

sigterm_context.go

func contextWithSigterm(ctx context.Context) context.Context {
	ctxWithCancel, cancel := context.WithCancel(ctx)
	go func() {
		defer cancel()

		signalCh := make(chan os.Signal, 1)
		signal.Notify(signalCh, os.Interrupt, syscall.SIGTERM)

		select {
		case <-signalCh:
		case <-ctx.Done():
		}
	}()

	return ctxWithCancel
}

You’re welcome!

LOL 😅

@henvic could you please extend the example in your article?

The problem is that the os.Interrupt is an alias to syscall.SIGINT but at least on Linux we also must handle the SIGTERM:

ctx, stop := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)

I had a problem when the systemd service stuck during reloading.

@stokito, thank you for the feedback. I've updated it.