extern/rclone
1
0
Fork 0
mirror of https://github.com/rclone/rclone.git synced 2024-11-07 17:14:44 +01:00
Code Issues Packages Projects Releases Wiki Activity

rc: fix setting bwlimit to unlimited

Browse Source
This commit is contained in:
Nick Craig-Wood 2018-04-24 09:43:07 +01:00
parent 764440068e
commit 9f04ce282e
1 changed files with 14 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
fs/accounting/token_bucket.go
View File

@ -115,6 +115,19 @@ func limitBandwidth(n int) {
tokenBucketMu.Unlock() tokenBucketMu.Unlock()
} }
// SetBwLimit sets the current bandwidth limit
func SetBwLimit(bandwidth fs.SizeSuffix) {
tokenBucketMu.Lock()
defer tokenBucketMu.Unlock()
if bandwidth > 0 {
tokenBucket = newTokenBucket(bandwidth)
fs.Logf(nil, "Bandwidth limit set to %v", bandwidth)
} else {
tokenBucket = nil
fs.Logf(nil, "Bandwidth limit reset to unlimited")
}
}
// Remote control for the token bucket // Remote control for the token bucket
func init() { func init() {
rc.Add(rc.Call{ rc.Add(rc.Call{
@ -137,10 +150,7 @@ func init() {
return out, errors.New("need exactly 1 bandwidth setting") return out, errors.New("need exactly 1 bandwidth setting")
} }
bw := bws[0] bw := bws[0]
tokenBucketMu.Lock() SetBwLimit(bw.Bandwidth)
tokenBucket = newTokenBucket(bw.Bandwidth)
tokenBucketMu.Unlock()
fs.Logf(nil, "Bandwidth limit set to %v", bw.Bandwidth)
return rc.Params{"rate": bw.Bandwidth.String()}, nil return rc.Params{"rate": bw.Bandwidth.String()}, nil
}, },
Title: "Set the bandwidth limit.", Title: "Set the bandwidth limit.",