ginkgo/ginkgo/interrupt_handler.go at refactor · blgm/ginkgo

Repository navigation

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()

}