GlcoinCore/alby-hub
1
0
Fork 0
mirror of https://github.com/getAlby/hub.git synced 2026-08-13 12:33:39 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

chore: log OS signal on exit in HTTP mode (#502)

Browse source
This commit is contained in:
Roland 2024-06-25 13:30:33 +07:00 • committed by GitHub
parent c9d64a548d
commit 9fdd71b7e0
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 19 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

22
cmd/http/main.go
View file

@ -19,7 +19,13 @@ import (
func main() {
log.Info("NWC Starting in HTTP mode")
ctx, _ := signal.NotifyContext(context.Background(), os.Interrupt, syscall.SIGTERM, os.Kill)
// Create a channel to receive OS signals.
osSignalChannel := make(chan os.Signal, 1)
// Notify the channel on os.Interrupt, syscall.SIGTERM, and os.Kill.
signal.Notify(osSignalChannel, os.Interrupt, syscall.SIGTERM, os.Kill)
ctx, cancel := context.WithCancel(context.Background())
svc, _ := service.NewService(ctx)
echologrus.Logger = logger.Logger
@ -34,10 +40,20 @@ func main() {
logger.Logger.Fatalf("shutting down the server: %v", err)
}
}()
var signal os.Signal
go func() {
// wait for exit signal
signal = <-osSignalChannel
logger.Logger.WithField("signal", signal).Info("Received OS signal")
cancel()
}()
//handle graceful shutdown
<-ctx.Done()
logger.Logger.Infof("Shutting down echo server...")
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
logger.Logger.WithField("signal", signal).Info("Context Done")
logger.Logger.Info("Shutting down echo server...")
ctx, cancel = context.WithTimeout(context.Background(), 10*time.Second)
defer cancel()
e.Shutdown(ctx)
logger.Logger.Info("Echo server exited")