feat: set max of the RateLimiter to the rate limit in config

Signed-off-by: Gaius <[email protected]>
dragonflyoss · Feb 10, 2025 · bcc1c69 · bcc1c69
1 parent 40264df
commit bcc1c69
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/dragonfly-client/src/resource/piece.rs b/dragonfly-client/src/resource/piece.rs
@@ -103,7 +103,7 @@ impl Piece {
                 RateLimiter::builder()
                     .initial(config.download.rate_limit.as_u64() as usize)
                     .refill(config.download.rate_limit.as_u64() as usize)
-                    .max((MAX_PIECE_LENGTH * MAX_PIECE_COUNT) as usize)
+                    .max(config.download.rate_limit.as_u64() as usize)
                     .interval(Duration::from_secs(1))
                     .fair(false)
                     .build(),
@@ -112,7 +112,7 @@ impl Piece {
                 RateLimiter::builder()
                     .initial(config.upload.rate_limit.as_u64() as usize)
                     .refill(config.upload.rate_limit.as_u64() as usize)
-                    .max((MAX_PIECE_LENGTH * MAX_PIECE_COUNT) as usize)
+                    .max(config.upload.rate_limit.as_u64() as usize)
                     .interval(Duration::from_secs(1))
                     .fair(false)
                     .build(),
@@ -121,7 +121,7 @@ impl Piece {
                 RateLimiter::builder()
                     .initial(config.proxy.prefetch_rate_limit.as_u64() as usize)
                     .refill(config.proxy.prefetch_rate_limit.as_u64() as usize)
-                    .max((MAX_PIECE_LENGTH * MAX_PIECE_COUNT) as usize)
+                    .max(config.proxy.prefetch_rate_limit.as_u64() as usize)
                     .interval(Duration::from_secs(1))
                     .fair(false)
                     .build(),