Show message for force exit

This commit is contained in:
Iwasaki Yudai 2015-09-01 15:07:04 +09:00
parent ff92777011
commit 05d9267f20
2 changed files with 8 additions and 3 deletions

View File

@ -286,8 +286,11 @@ func (app *App) handleAuthToken(w http.ResponseWriter, r *http.Request) {
func (app *App) Exit() (firstCall bool) {
if app.server != nil {
log.Printf("Received Exit command, waiting for all clients to close sessions...")
return app.server.Close()
firstCall = app.server.Close()
if firstCall {
log.Printf("Received Exit command, waiting for all clients to close sessions...")
}
return firstCall
}
return true
}

View File

@ -120,7 +120,9 @@ func registerSignals(app *app.App) {
s := <-sigChan
switch s {
case syscall.SIGINT, syscall.SIGTERM:
if !app.Exit() {
if app.Exit() {
fmt.Println("Send ^C to force exit.")
} else {
os.Exit(5)
}
}