Graceful Shutdown in Go

gs.go

package main

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

func main() {

	// ctx, cancel := context.WithCancel(context.Background())
	// defer cancel()

	ctx, cancel := signal.NotifyContext(context.Background(), os.Interrupt)
	defer cancel()

	var wg sync.WaitGroup

	wg.Add(1)
	go func() {
		defer wg.Done()
		for {
			select {
			case <-ctx.Done():
				log.Println("Break the loop 1")
				return
			case <-time.After(1 * time.Second):
				log.Println("loop 1")
			}
		}
	}()

	wg.Add(1)
	go func() {
		defer wg.Done()
		for {
			select {
			case <-ctx.Done():
				log.Println("Break the loop 2")
				return
			case <-time.After(2 * time.Second):
				log.Println("loop 2")
			}
		}
	}()

	wg.Wait()

	log.Println("Finishing as a pro!")
}