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

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

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