Konstantin Lebedev
2 years ago
committed by
GitHub
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with
4 additions and
2 deletions
-
weed/util/grace/signal_handling.go
|
|
@ -12,7 +12,7 @@ import ( |
|
|
|
|
|
|
|
|
var signalChan chan os.Signal |
|
|
var signalChan chan os.Signal |
|
|
var hooks = make([]func(), 0) |
|
|
var hooks = make([]func(), 0) |
|
|
var hookLock sync.Mutex |
|
|
|
|
|
|
|
|
var hookLock sync.RWMutex |
|
|
|
|
|
|
|
|
func init() { |
|
|
func init() { |
|
|
signalChan = make(chan os.Signal, 1) |
|
|
signalChan = make(chan os.Signal, 1) |
|
|
@ -27,10 +27,12 @@ func init() { |
|
|
// syscall.SIGQUIT,
|
|
|
// syscall.SIGQUIT,
|
|
|
) |
|
|
) |
|
|
go func() { |
|
|
go func() { |
|
|
for _ = range signalChan { |
|
|
|
|
|
|
|
|
for range signalChan { |
|
|
|
|
|
hookLock.RLock() |
|
|
for _, hook := range hooks { |
|
|
for _, hook := range hooks { |
|
|
hook() |
|
|
hook() |
|
|
} |
|
|
} |
|
|
|
|
|
hookLock.RUnlock() |
|
|
os.Exit(0) |
|
|
os.Exit(0) |
|
|
} |
|
|
} |
|
|
}() |
|
|
}() |
|
|
|
