From a04e46f71983ba8bb5c8056ac232a158aacf8e90 Mon Sep 17 00:00:00 2001 From: Anna Shaleva Date: Fri, 30 Apr 2021 15:53:27 +0300 Subject: [PATCH] cli: reload RPC-server TLS config on SIGHUP --- cli/server/server.go | 18 +++++++++++++++++- 1 file changed, 17 insertions(+), 1 deletion(-) diff --git a/cli/server/server.go b/cli/server/server.go index daf0438e3..e0899ec52 100644 --- a/cli/server/server.go +++ b/cli/server/server.go @@ -5,6 +5,7 @@ import ( "fmt" "os" "os/signal" + "syscall" "github.com/nspcc-dev/neo-go/cli/options" "github.com/nspcc-dev/neo-go/pkg/config" @@ -325,6 +326,9 @@ func startServer(ctx *cli.Context) error { go serv.Start(errChan) rpcServer.Start(errChan) + sighupCh := make(chan os.Signal, 1) + signal.Notify(sighupCh, syscall.SIGHUP) + fmt.Fprintln(ctx.App.Writer, logo()) fmt.Fprintln(ctx.App.Writer, serv.UserAgent) fmt.Fprintln(ctx.App.Writer) @@ -336,8 +340,20 @@ Main: case err := <-errChan: shutdownErr = fmt.Errorf("server error: %w", err) cancel() - + case sig := <-sighupCh: + switch sig { + case syscall.SIGHUP: + log.Info("SIGHUP received, restarting rpc-server") + serverErr := rpcServer.Shutdown() + if serverErr != nil { + errChan <- fmt.Errorf("error while restarting rpc-server: %w", serverErr) + break + } + rpcServer = server.New(chain, cfg.ApplicationConfiguration.RPC, serv, serv.GetOracle(), log) + rpcServer.Start(errChan) + } case <-grace.Done(): + signal.Stop(sighupCh) serv.Shutdown() if serverErr := rpcServer.Shutdown(); serverErr != nil { shutdownErr = fmt.Errorf("error on shutdown: %w", serverErr)