interrupt_handler.go

package main

import (
	"os"
	"os/signal"
	"sync"
)

type InterruptHandler struct {
	interruptCount int
	lock           *sync.Mutex
	C              chan bool
}

func NewInterruptHandler() *InterruptHandler {
	h := &InterruptHandler{
		lock: &sync.Mutex{},
		C:    make(chan bool, 0),
	}

	go h.handleInterrupt()

	return h
}

func (h *InterruptHandler) WasInterrupted() bool {
	h.lock.Lock()
	defer h.lock.Unlock()

	return h.interruptCount > 0
}

func (h *InterruptHandler) handleInterrupt() {
	c := make(chan os.Signal, 1)
	signal.Notify(c, os.Interrupt)

	<-c
	signal.Stop(c)

	h.lock.Lock()
	h.interruptCount++
	if h.interruptCount == 1 {
		close(h.C)
	} else if h.interruptCount > 5 {
		os.Exit(1)
	}
	h.lock.Unlock()

	go h.handleInterrupt()
}