disallow burstiness in rate limiter

paradigmxyz · Oct 15, 2023 · 5ac99d4 · 5ac99d4
1 parent a247da8
commit 5ac99d4
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/crates/cli/src/parse/source.rs b/crates/cli/src/parse/source.rs
@@ -18,9 +18,9 @@ pub(crate) async fn parse_source(args: &Args) -> Result<Source, ParseError> {
     let chain_id = provider.get_chainid().await.map_err(ParseError::ProviderError)?.as_u64();
 
     let rate_limiter = match args.requests_per_second {
-        Some(rate_limit) => match NonZeroU32::new(rate_limit) {
-            Some(value) => {
-                let quota = Quota::per_second(value);
+        Some(rate_limit) => match (NonZeroU32::new(1), NonZeroU32::new(rate_limit)) {
+            (Some(one), Some(value)) => {
+                let quota = Quota::per_second(value).allow_burst(one);
                 Some(RateLimiter::direct(quota))
             }
             _ => None,