How to Implement Graceful Shutdown with Context in Go

Implement graceful shutdown by creating a context that cancels on OS signals and passing it to your goroutines.

package main

import (
	"context"
	"log"
	"os"
	"os/signal"
	"syscall"
	"time"
)

func main() {
	ctx, cancel := context.WithCancel(context.Background())
	defer cancel()

	go func() {
		<-ctx.Done()
		log.Println("Shutting down...")
	}()

	sigChan := make(chan os.Signal, 1)
	signal.Notify(sigChan, syscall.SIGINT, syscall.SIGTERM)

	select {
	case <-sigChan:
		log.Println("Received interrupt signal")
		cancel()
	case <-time.After(5 * time.Second):
		log.Println("Timeout")
	}
}

Graceful shutdown lets your program finish current tasks before closing when a user stops it. It works like a polite waiter who finishes serving the current table before leaving the restaurant. You use this to prevent data loss or corrupted files when your app exits.