maride/gotty
1
0
Fork 0
mirror of https://github.com/sorenisanerd/gotty.git synced 2024-11-22 12:24:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

Show message for force exit

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
app/app.go
View File

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

4
main.go
View File

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