zeromicro/go-zero
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat: add the SIGINT signal in signals.go to subscribe the user input ctrl+c to exit the application operation (#3611)

Browse Source
This commit is contained in:
#Suyghur
2023-10-15 09:58:15 -05:00
committed by GitHub
parent d71c0da7b7
commit 95b7a3d3ce
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
core/proc/signals.go
View File

@@ -20,7 +20,7 @@ func init() {
// https://golang.org/pkg/os/signal/#Notify // https://golang.org/pkg/os/signal/#Notify
signals := make(chan os.Signal, 1) signals := make(chan os.Signal, 1)
signal.Notify(signals, syscall.SIGUSR1, syscall.SIGUSR2, syscall.SIGTERM) signal.Notify(signals, syscall.SIGUSR1, syscall.SIGUSR2, syscall.SIGTERM, syscall.SIGINT)
for { for {
v := <-signals v := <-signals
@@ -34,14 +34,13 @@ func init() {
profiler.Stop() profiler.Stop()
profiler = nil profiler = nil
} }
case syscall.SIGTERM: case syscall.SIGTERM, syscall.SIGINT:
select { select {
case <-done: case <-done:
// already closed // already closed
default: default:
close(done) close(done)
} }
gracefulStop(signals) gracefulStop(signals)
default: default:
logx.Error("Got unregistered signal:", v) logx.Error("Got unregistered signal:", v)