TrueCloudLab/rclone
16
0
Fork 2
Code Issues Pull requests 1 Projects Releases Packages Wiki Activity Actions 1

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.",