mirror of
https://github.com/rclone/rclone.git
synced 2024-11-29 20:04:52 +01:00
e65059e431
The race detector complained whenever SIGUSR2 was sent to rclone so this adds a mutex to prevent concurrent access.
34 lines
711 B
Go
34 lines
711 B
Go
// Accounting and limiting reader
|
|
// Unix specific functions.
|
|
|
|
// +build darwin dragonfly freebsd linux netbsd openbsd solaris
|
|
|
|
package fs
|
|
|
|
import (
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
)
|
|
|
|
// startSignalHandler() sets a signal handler to catch SIGUSR2 and toggle throttling.
|
|
func startSignalHandler() {
|
|
signals := make(chan os.Signal, 1)
|
|
signal.Notify(signals, syscall.SIGUSR2)
|
|
|
|
go func() {
|
|
// This runs forever, but blocks until the signal is received.
|
|
for {
|
|
<-signals
|
|
tokenBucketMu.Lock()
|
|
tokenBucket, prevTokenBucket = prevTokenBucket, tokenBucket
|
|
s := "disabled"
|
|
if tokenBucket != nil {
|
|
s = "enabled"
|
|
}
|
|
tokenBucketMu.Unlock()
|
|
Log(nil, "Bandwidth limit %s by user", s)
|
|
}
|
|
}()
|
|
}
|