From e4ba507acea7f069fd959d1a70a981c8526d6fea Mon Sep 17 00:00:00 2001 From: bitfl0wer Date: Sat, 15 Apr 2023 17:12:33 +0200 Subject: [PATCH] add limit updates for Error, Global, Ip Despite Error, Global and Ip technically being handled seperately, we still need to update their "reset" and "limit" values to match what the API tells us. --- src/limit.rs | 16 ++++++++++++---- 1 file changed, 12 insertions(+), 4 deletions(-) diff --git a/src/limit.rs b/src/limit.rs index a732f5f..e8c8ea2 100644 --- a/src/limit.rs +++ b/src/limit.rs @@ -151,10 +151,18 @@ impl LimitedRequester { let mut_limits_rate = &mut self.limits_rate; match limit_type { - // Error, Global and Ip get handled seperately. - LimitType::Error => {} - LimitType::Global => {} - LimitType::Ip => {} + LimitType::Error => { + let entry = mut_limits_rate.get_mut(&LimitType::Error).unwrap(); + LimitedRequester::update_limit_entry(entry, reset, remaining, limit); + } + LimitType::Global => { + let entry = mut_limits_rate.get_mut(&LimitType::Global).unwrap(); + LimitedRequester::update_limit_entry(entry, reset, remaining, limit); + } + LimitType::Ip => { + let entry = mut_limits_rate.get_mut(&LimitType::Ip).unwrap(); + LimitedRequester::update_limit_entry(entry, reset, remaining, limit); + } LimitType::AuthLogin => { let entry = mut_limits_rate.get_mut(&LimitType::AuthLogin).unwrap(); LimitedRequester::update_limit_entry(entry, reset, remaining, limit);