Graceful shutdown with Context + channels (https://play.golang.org/p/ZBxNMddpbj2)

gracefully_shutdown.go

// The point of this snippet is to showcase how to handle gracefully any incoming
// SIGINT (manual ctrl-c) or SIGTERM (process is killed) for a long-running task
// such as a server.
package main

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

func LongRunning(ctx context.Context) error {
	timeout := time.After(1 * time.Second)
	select {
	case <-ctx.Done():
		return fmt.Errorf("you canceled my job!")
	case <-timeout:
		return nil
	}

	return nil
}

func main() {
	ctx, cancel := context.WithCancel(context.Background())
	done := make(chan bool)
	go func() {
		defer close(done)
		if err := LongRunning(ctx); err != nil {
			log.Fatalf("error running LongRunning: %v", err)
		}
	}()

	wait := make(chan os.Signal, 1)
	signal.Notify(wait, syscall.SIGINT, syscall.SIGTERM)
	<-wait
	cancel() // gracefully stop my long running function
	<-done
}