grishka/lemmy
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Removing check_only

Browse source
This commit is contained in:
Dessalines 2022-03-24 16:29:08 -04:00
parent ed9f70276d
commit 2da0684e95
2 changed files with 2 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
crates/utils/src/rate_limit/mod.rs
View file

@ -86,7 +86,6 @@ impl RateLimited {
&ip_addr, &ip_addr,
rate_limit.message, rate_limit.message,
rate_limit.message_per_second, rate_limit.message_per_second,
false,
)?; )?;
drop(limiter); drop(limiter);
@ -98,7 +97,6 @@ impl RateLimited {
&ip_addr, &ip_addr,
rate_limit.post, rate_limit.post,
rate_limit.post_per_second, rate_limit.post_per_second,
false,
)?; )?;
} }
RateLimitType::Register => { RateLimitType::Register => {
@ -107,7 +105,6 @@ impl RateLimited {
&ip_addr, &ip_addr,
rate_limit.register, rate_limit.register,
rate_limit.register_per_second, rate_limit.register_per_second,
false,
)?; )?;
} }
RateLimitType::Image => { RateLimitType::Image => {
@ -116,7 +113,6 @@ impl RateLimited {
&ip_addr, &ip_addr,
rate_limit.image, rate_limit.image,
rate_limit.image_per_second, rate_limit.image_per_second,
false,
)?; )?;
} }
RateLimitType::Comment => { RateLimitType::Comment => {
@ -125,7 +121,6 @@ impl RateLimited {
&ip_addr, &ip_addr,
rate_limit.comment, rate_limit.comment,
rate_limit.comment_per_second, rate_limit.comment_per_second,
false,
)?; )?;
} }
}; };

5
crates/utils/src/rate_limit/rate_limiter.rs
View file

@ -53,7 +53,6 @@ impl RateLimiter {
ip: &IpAddr, ip: &IpAddr,
rate: i32, rate: i32,
per: i32, per: i32,
check_only: bool,
) -> Result<(), LemmyError> { ) -> Result<(), LemmyError> {
self.insert_ip(ip); self.insert_ip(ip);
if let Some(bucket) = self.buckets.get_mut(&type_) { if let Some(bucket) = self.buckets.get_mut(&type_) {
@ -68,7 +67,7 @@ impl RateLimiter {
rate_limit.last_checked = current; rate_limit.last_checked = current;
rate_limit.allowance += time_passed * (rate as f64 / per as f64); rate_limit.allowance += time_passed * (rate as f64 / per as f64);
if !check_only && rate_limit.allowance > rate as f64 { if rate_limit.allowance > rate as f64 {
rate_limit.allowance = rate as f64; rate_limit.allowance = rate as f64;
} }
@ -91,9 +90,7 @@ impl RateLimiter {
"too_many_requests", "too_many_requests",
)) ))
} else { } else {
if !check_only {
rate_limit.allowance -= 1.0; rate_limit.allowance -= 1.0;
}
Ok(()) Ok(())
} }
} else { } else {