mirror of
https://github.com/sorenisanerd/gotty.git
synced 2024-11-09 15:24:25 +00:00
Shutdown server gracefully with Ctrl-C
This commit is contained in:
parent
94a6230355
commit
e613b29cc3
19
app/app.go
19
app/app.go
@ -17,6 +17,7 @@ import (
|
|||||||
"strings"
|
"strings"
|
||||||
"text/template"
|
"text/template"
|
||||||
|
|
||||||
|
"github.com/braintree/manners"
|
||||||
"github.com/elazarl/go-bindata-assetfs"
|
"github.com/elazarl/go-bindata-assetfs"
|
||||||
"github.com/gorilla/websocket"
|
"github.com/gorilla/websocket"
|
||||||
"github.com/kr/pty"
|
"github.com/kr/pty"
|
||||||
@ -26,6 +27,7 @@ type App struct {
|
|||||||
options Options
|
options Options
|
||||||
|
|
||||||
upgrader *websocket.Upgrader
|
upgrader *websocket.Upgrader
|
||||||
|
server *manners.GracefulServer
|
||||||
|
|
||||||
preferences map[string]interface{}
|
preferences map[string]interface{}
|
||||||
titleTemplate *template.Template
|
titleTemplate *template.Template
|
||||||
@ -157,16 +159,21 @@ func (app *App) Run() error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
var err error
|
var err error
|
||||||
|
app.server = manners.NewWithServer(
|
||||||
|
&http.Server{Addr: endpoint, Handler: siteHandler},
|
||||||
|
)
|
||||||
if app.options.EnableTLS {
|
if app.options.EnableTLS {
|
||||||
cert, key := app.loadTLSFiles()
|
cert, key := app.loadTLSFiles()
|
||||||
err = http.ListenAndServeTLS(endpoint, cert, key, siteHandler)
|
err = app.server.ListenAndServeTLS(cert, key)
|
||||||
} else {
|
} else {
|
||||||
err = http.ListenAndServe(endpoint, siteHandler)
|
err = app.server.ListenAndServe()
|
||||||
}
|
}
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
log.Printf("Exiting...")
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -217,6 +224,14 @@ func (app *App) handleWS(w http.ResponseWriter, r *http.Request) {
|
|||||||
context.goHandleClient()
|
context.goHandleClient()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
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()
|
||||||
|
}
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
func wrapLogger(handler http.Handler) http.Handler {
|
func wrapLogger(handler http.Handler) http.Handler {
|
||||||
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||||
log.Printf("%s %s", r.Method, r.URL.Path)
|
log.Printf("%s %s", r.Method, r.URL.Path)
|
||||||
|
@ -61,7 +61,10 @@ func (context *clientContext) goHandleClient() {
|
|||||||
context.processReceive()
|
context.processReceive()
|
||||||
}()
|
}()
|
||||||
|
|
||||||
|
context.app.server.StartRoutine()
|
||||||
go func() {
|
go func() {
|
||||||
|
defer context.app.server.FinishRoutine()
|
||||||
|
|
||||||
<-exit
|
<-exit
|
||||||
context.pty.Close()
|
context.pty.Close()
|
||||||
context.command.Wait()
|
context.command.Wait()
|
||||||
|
25
main.go
25
main.go
@ -7,6 +7,8 @@ import (
|
|||||||
"github.com/codegangsta/cli"
|
"github.com/codegangsta/cli"
|
||||||
|
|
||||||
"github.com/yudai/gotty/app"
|
"github.com/yudai/gotty/app"
|
||||||
|
"os/signal"
|
||||||
|
"syscall"
|
||||||
)
|
)
|
||||||
|
|
||||||
func main() {
|
func main() {
|
||||||
@ -107,6 +109,8 @@ func main() {
|
|||||||
os.Exit(2)
|
os.Exit(2)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
registerSignals(app)
|
||||||
|
|
||||||
err = app.Run()
|
err = app.Run()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
fmt.Println(err)
|
fmt.Println(err)
|
||||||
@ -118,3 +122,24 @@ func main() {
|
|||||||
|
|
||||||
cmd.Run(os.Args)
|
cmd.Run(os.Args)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func registerSignals(app *app.App) {
|
||||||
|
sigChan := make(chan os.Signal, 1)
|
||||||
|
signal.Notify(
|
||||||
|
sigChan,
|
||||||
|
syscall.SIGINT,
|
||||||
|
syscall.SIGTERM,
|
||||||
|
)
|
||||||
|
|
||||||
|
go func() {
|
||||||
|
for {
|
||||||
|
s := <-sigChan
|
||||||
|
switch s {
|
||||||
|
case syscall.SIGINT, syscall.SIGTERM:
|
||||||
|
if !app.Exit() {
|
||||||
|
os.Exit(4)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user