From c063a2ed3294c0853f8d7f1a95115f1906588192 Mon Sep 17 00:00:00 2001 From: Gaius Date: Mon, 10 Feb 2025 11:20:55 +0800 Subject: [PATCH] feat: set max of the RateLimiter to the rate limit in config (#966) Signed-off-by: Gaius --- dragonfly-client/src/resource/piece.rs | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/dragonfly-client/src/resource/piece.rs b/dragonfly-client/src/resource/piece.rs index f81ef66a..afe678f4 100644 --- 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(),