packer-cn/signal.go

40 lines
980 B
Go
Raw Normal View History

package main
import (
"log"
"os"
"os/signal"
"syscall"
2015-06-09 00:26:28 -04:00
"github.com/mitchellh/packer/packer"
"github.com/mitchellh/packer/packer/plugin"
)
// Prepares the signal handlers so that we handle interrupts properly.
// The signal handler exists in a goroutine.
2015-05-25 20:29:10 -04:00
func setupSignalHandlers(ui packer.Ui) {
ch := make(chan os.Signal, 1)
signal.Notify(ch, os.Interrupt)
signal.Notify(ch, syscall.SIGTERM)
go func() {
2013-07-25 22:36:46 -04:00
// First interrupt. We mostly ignore this because it allows the
// plugins time to cleanup.
<-ch
log.Println("First interrupt. Ignoring to allow plugins to clean up.")
2013-07-25 22:36:46 -04:00
2015-05-25 20:29:10 -04:00
ui.Error("Interrupt signal received. Cleaning up...")
2013-07-25 22:36:46 -04:00
// Second interrupt. Go down hard.
<-ch
log.Println("Second interrupt. Exiting now.")
2015-05-25 20:29:10 -04:00
ui.Error("Interrupt signal received twice. Forcefully exiting now.")
// Force kill all the plugins, but mark that we're killing them
// first so that we don't get panics everywhere.
plugin.CleanupClients()
os.Exit(1)
}()
}