This commit is contained in:
Brian Morin 2025-01-19 13:52:31 -08:00
parent 6f49cce887
commit d85fdb16cb
2 changed files with 9 additions and 13 deletions

View file

@ -136,7 +136,7 @@ impl VectorTokenBucket {
impl TokenBucket for VectorTokenBucket { impl TokenBucket for VectorTokenBucket {
fn get_capacity(&self) -> f32 { fn get_capacity(&self) -> f32 {
if self.total_limit <= 0 { if self.total_limit == 0 {
// Handle edge cases by telling the caller we have no capacity. // Handle edge cases by telling the caller we have no capacity.
return -1.0; return -1.0;
} }
@ -147,7 +147,7 @@ impl TokenBucket for VectorTokenBucket {
return -1.0; return -1.0;
} }
return 1.0 - (timestamps.len() as f32 / self.total_limit as f32); 1.0 - (timestamps.len() as f32 / self.total_limit as f32)
} }
fn get_delay(&self) -> Option<Duration> { fn get_delay(&self) -> Option<Duration> {

View file

@ -237,7 +237,7 @@ impl RiotApi {
Err(e) => Some(Err(e)), Err(e) => Some(Err(e)),
} }
} else { } else {
return None; None
} }
} }
@ -296,20 +296,16 @@ impl RiotApi {
/// ///
/// # Returns /// # Returns
/// A future resolving to a `Result` containg either a `ResponseInfo` (success) or a `RiotApiError` (failure). /// A future resolving to a `Result` containg either a `ResponseInfo` (success) or a `RiotApiError` (failure).
pub fn execute_raw( pub async fn execute_raw(
&self, &self,
method_id: &'static str, method_id: &'static str,
region_platform: &'static str, region_platform: &'static str,
request: RequestBuilder, request: RequestBuilder,
) -> impl Future<Output = Result<ResponseInfo>> + '_ { ) -> Result<ResponseInfo> {
async move { self.regional_requester(region_platform)
self.regional_requester(region_platform) .execute(&self.config, method_id, request, None)
.execute(&self.config, method_id, request, None) .await
.await .expect("regional_requester.excute only returns None when min_capacity is Some(f32)")
.expect(
"regional_requester.excute only returns None when min_capacity is Some(f32)",
)
}
} }
/// This method should generally not be used directly. Consider using endpoint wrappers instead. /// This method should generally not be used directly. Consider using endpoint wrappers instead.