Encapsulated, gracefull goroutine shutdown in

main.go

package main

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

func main() {
	var wg sync.WaitGroup

	work := make(chan bool)
	go func() {
		for {
			work <- true
			time.Sleep(20 * time.Millisecond)
		}
	}()

	wg.Add(1)
	go func() {
		grace, kill := make(chan os.Signal, 1), make(chan os.Signal, 1)
		signal.Notify(grace, syscall.SIGINT)
		signal.Notify(kill, syscall.SIGTERM)
		for {
			select {
			case _ = <-work:
				log.Println("just doing some work, don't mind me")
				time.Sleep(500 * time.Millisecond)
			case _ = <-grace:
				log.Println("Looks like I got a SIGINT, gonna clean up first")
				wg.Done()
				return
			case _ = <-kill:
				log.Println("Looks like I got a SIGTERM")
				wg.Done()
				return
			}
		}

	}()

	wg.Wait()
}

Why not defer wg.Done()?

And omit the _ =?