cloudflared-mirror/cmd/cloudflared/tunnel/signal.go

24 lines
543 B
Go
Raw Permalink Normal View History

package tunnel
import (
"os"
"os/signal"
"syscall"
"github.com/rs/zerolog"
)
// waitForSignal closes graceShutdownC to indicate that we should start graceful shutdown sequence
func waitForSignal(graceShutdownC chan struct{}, logger *zerolog.Logger) {
signals := make(chan os.Signal, 10)
signal.Notify(signals, syscall.SIGTERM, syscall.SIGINT)
defer signal.Stop(signals)
select {
case s := <-signals:
logger.Info().Msgf("Initiating graceful shutdown due to signal %s ...", s)
close(graceShutdownC)
case <-graceShutdownC:
}
}