fix calculation of speed limit #41

cenkalti · Nov 7, 2021 · d1d9dc7 · d1d9dc7
1 parent 3054d79
commit d1d9dc7
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/torrent/session.go b/torrent/session.go
@@ -194,11 +194,13 @@ func NewSession(cfg Config) (*Session, error) {
  },
  },
  }
+ dlSpeed := cfg.SpeedLimitDownload * 1024
  if cfg.SpeedLimitDownload > 0 {
- c.bucketDownload = ratelimit.NewBucketWithRate(float64(cfg.SpeedLimitDownload), cfg.SpeedLimitDownload)
+ c.bucketDownload = ratelimit.NewBucketWithRate(float64(dlSpeed), dlSpeed)
  }
+ ulSpeed := cfg.SpeedLimitUpload * 1024
  if cfg.SpeedLimitUpload > 0 {
- c.bucketUpload = ratelimit.NewBucketWithRate(float64(cfg.SpeedLimitUpload), cfg.SpeedLimitUpload)
+ c.bucketUpload = ratelimit.NewBucketWithRate(float64(ulSpeed), ulSpeed)
  }
  err = c.startBlocklistReloader()
  if err != nil {