Gaialite signal handling is broken, repair it
This commit is contained in:
parent
f4d3e65ef0
commit
48192ea3d4
|
@ -5,6 +5,8 @@ import (
|
||||||
"net"
|
"net"
|
||||||
"net/http"
|
"net/http"
|
||||||
"os"
|
"os"
|
||||||
|
"os/signal"
|
||||||
|
"syscall"
|
||||||
|
|
||||||
"github.com/cosmos/cosmos-sdk/client"
|
"github.com/cosmos/cosmos-sdk/client"
|
||||||
"github.com/cosmos/cosmos-sdk/client/context"
|
"github.com/cosmos/cosmos-sdk/client/context"
|
||||||
|
@ -21,7 +23,6 @@ import (
|
||||||
"github.com/rakyll/statik/fs"
|
"github.com/rakyll/statik/fs"
|
||||||
"github.com/spf13/cobra"
|
"github.com/spf13/cobra"
|
||||||
"github.com/spf13/viper"
|
"github.com/spf13/viper"
|
||||||
cmn "github.com/tendermint/tendermint/libs/common"
|
|
||||||
"github.com/tendermint/tendermint/libs/log"
|
"github.com/tendermint/tendermint/libs/log"
|
||||||
tmserver "github.com/tendermint/tendermint/rpc/lib/server"
|
tmserver "github.com/tendermint/tendermint/rpc/lib/server"
|
||||||
)
|
)
|
||||||
|
@ -57,6 +58,25 @@ func ServeCommand(cdc *codec.Codec) *cobra.Command {
|
||||||
|
|
||||||
var listener net.Listener
|
var listener net.Listener
|
||||||
var fingerprint string
|
var fingerprint string
|
||||||
|
|
||||||
|
sigs := make(chan os.Signal, 1)
|
||||||
|
signal.Notify(sigs, syscall.SIGINT, syscall.SIGTERM)
|
||||||
|
closeListener := func() {
|
||||||
|
err := listener.Close()
|
||||||
|
logger.Error("error closing listener", "err", err)
|
||||||
|
}
|
||||||
|
go func() {
|
||||||
|
sig := <-sigs
|
||||||
|
switch sig {
|
||||||
|
case syscall.SIGTERM:
|
||||||
|
defer closeListener()
|
||||||
|
os.Exit(128 + int(syscall.SIGTERM))
|
||||||
|
case syscall.SIGINT:
|
||||||
|
defer closeListener()
|
||||||
|
os.Exit(128 + int(syscall.SIGINT))
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
|
||||||
if viper.GetBool(flagInsecure) {
|
if viper.GetBool(flagInsecure) {
|
||||||
listener, err = tmserver.StartHTTPServer(
|
listener, err = tmserver.StartHTTPServer(
|
||||||
listenAddr, handler, logger,
|
listenAddr, handler, logger,
|
||||||
|
@ -89,6 +109,7 @@ func ServeCommand(cdc *codec.Codec) *cobra.Command {
|
||||||
}
|
}
|
||||||
defer cleanupFunc()
|
defer cleanupFunc()
|
||||||
}
|
}
|
||||||
|
|
||||||
listener, err = tmserver.StartHTTPAndTLSServer(
|
listener, err = tmserver.StartHTTPAndTLSServer(
|
||||||
listenAddr, handler,
|
listenAddr, handler,
|
||||||
certFile, keyFile,
|
certFile, keyFile,
|
||||||
|
@ -102,13 +123,6 @@ func ServeCommand(cdc *codec.Codec) *cobra.Command {
|
||||||
}
|
}
|
||||||
logger.Info("REST server started")
|
logger.Info("REST server started")
|
||||||
|
|
||||||
// wait forever and cleanup
|
|
||||||
cmn.TrapSignal(func() {
|
|
||||||
defer cleanupFunc()
|
|
||||||
err := listener.Close()
|
|
||||||
logger.Error("error closing listener", "err", err)
|
|
||||||
})
|
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue