github/rclone
1
Fork 0
mirror of https://github.com/rclone/rclone synced 2024-11-23 00:06:55 +01:00
Code Releases Activity

atexit: add Signalled() function - set if called from a signal #4804

Browse Source
This commit is contained in:
Nick Craig-Wood 2020-11-27 10:49:24 +00:00
parent 1cfce703b2
commit 83406bc473
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
lib/atexit/atexit.go
View File

@ -8,6 +8,7 @@ import (
"os" "os"
"os/signal" "os/signal"
"sync" "sync"
"sync/atomic"
"github.com/rclone/rclone/fs" "github.com/rclone/rclone/fs"
) )
@ -18,6 +19,7 @@ var (
exitChan chan os.Signal exitChan chan os.Signal
exitOnce sync.Once exitOnce sync.Once
registerOnce sync.Once registerOnce sync.Once
signalled int32
) )
// FnHandle is the type of the handle returned by function `Register` // FnHandle is the type of the handle returned by function `Register`
@ -40,6 +42,7 @@ func Register(fn func()) FnHandle {
if sig == nil { if sig == nil {
return return
} }
atomic.StoreInt32(&signalled, 1)
fs.Infof(nil, "Signal received: %s", sig) fs.Infof(nil, "Signal received: %s", sig)
Run() Run()
fs.Infof(nil, "Exiting...") fs.Infof(nil, "Exiting...")
@ -50,6 +53,11 @@ func Register(fn func()) FnHandle {
return &fn return &fn
} }
// Signalled returns true if an exit signal has been received
func Signalled() bool {
return atomic.LoadInt32(&signalled) != 0
}
// Unregister a function using the handle returned by `Register` // Unregister a function using the handle returned by `Register`
func Unregister(handle FnHandle) { func Unregister(handle FnHandle) {
fnsMutex.Lock() fnsMutex.Lock()