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

Signed-off-by: NotAShelf <raf@notashelf.dev>
Change-Id: Iaa63eef1f055b9a98c90739025bd3ff36a6a6964

src/rate_limiter.rs

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