NotAShelf/pakker
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

rate-limiter: re-validate burst window in loop after sleep

Browse source 
Signed-off-by: NotAShelf <raf@notashelf.dev>
Change-Id: Iaa63eef1f055b9a98c90739025bd3ff36a6a6964
This commit is contained in:
raf 2026-02-21 19:50:55 +03:00
commit 5a7762ac13
Signed by: NotAShelf
GPG key ID: 29D95B64378DB4BF
1 changed files with 21 additions and 24 deletions
Download patch file Download diff file Expand all files Collapse all files

13
src/rate_limiter.rs
View file

@ -72,6 +72,7 @@ impl RateLimiter {
let interval = Duration::from_secs(60) / rate.max(1); let interval = Duration::from_secs(60) / rate.max(1);
loop {
let mut inner = self.inner.lock().await; let mut inner = self.inner.lock().await;
let now = Instant::now(); let now = Instant::now();
let platform_requests = let platform_requests =
@ -87,19 +88,15 @@ impl RateLimiter {
if wait_time > Duration::ZERO { if wait_time > Duration::ZERO {
drop(inner); drop(inner);
tokio::time::sleep(wait_time).await; tokio::time::sleep(wait_time).await;
let mut inner = self.inner.lock().await; continue;
let platform_requests = }
inner.requests.entry(platform.to_string()).or_default(); }
platform_requests.push(Instant::now()); platform_requests.push(Instant::now());
return Ok(()); return Ok(());
} }
} }
platform_requests.push(Instant::now());
Ok(())
}
pub async fn wait_for(&self, platform: &str) { pub async fn wait_for(&self, platform: &str) {
let _ = self.acquire(platform).await; let _ = self.acquire(platform).await;
} }