From 6b98a2cc23346659ca87f2a197eaf1afcdf967c2 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);