Signal handling, force quit after two interrupts

This commit is contained in:
Mitchell Hashimoto 2013-06-03 22:40:05 -07:00
parent 5c39a2f7ca
commit 256b97ce49
2 changed files with 31 additions and 0 deletions

View File

@ -96,6 +96,8 @@ func main() {
os.Exit(1)
}
setupSignalHandlers(env)
exitCode, err := env.Cli(os.Args[1:])
if err != nil {
fmt.Fprintf(os.Stderr, "Error executing CLI: %s\n", err.Error())

29
signal.go Normal file
View File

@ -0,0 +1,29 @@
package main
import (
"github.com/mitchellh/packer/packer"
"github.com/mitchellh/packer/packer/plugin"
"log"
"os"
"os/signal"
)
// Prepares the signal handlers so that we handle interrupts properly.
// The signal handler exists in a goroutine.
func setupSignalHandlers(env packer.Environment) {
ch := make(chan os.Signal, 1)
signal.Notify(ch, os.Interrupt)
go func() {
<-ch
log.Println("First interrupt. Ignoring, will let plugins handle...")
<-ch
log.Println("Second interrupt. Exiting now.")
env.Ui().Error("Interrupt signal received twice. Forcefully exiting now.")
// Force kill all the plugins
plugin.CleanupClients()
os.Exit(1)
}()
}