mirror of
https://github.com/matter-labs/teepot.git
synced 2025-07-21 07:03:56 +02:00
bb9c5b195e
- Add `max_retries` field to ApiClient with default of 3 retries - Implement `execute_with_retry()` helper method in helpers.rs - Update all HTTP requests to use retry wrapper for automatic 429 handling - Add `TooManyRequests` error variant with request_id and retry_after fields - Respect Retry-After header duration before retrying requests - Add `set_max_retries()` method to configure retry behavior (0 disables) - Update documentation and add handle_rate_limit example - Enhanced error handling in check_status() for 429 responses The client now transparently handles Intel API rate limiting while remaining configurable for users who need different retry behavior or manual handling. 🤖 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude <noreply@anthropic.com> Signed-off-by: Harald Hoyer <harald@matterlabs.dev>
91 lines
3.2 KiB
Rust
91 lines
3.2 KiB
Rust
// SPDX-License-Identifier: Apache-2.0
|
|
// Copyright (c) 2025 Matter Labs
|
|
|
|
//! Example demonstrating automatic rate limit handling
|
|
//!
|
|
//! The Intel DCAP API client now automatically handles 429 Too Many Requests responses
|
|
//! by retrying up to 3 times by default. This example shows how to configure the retry
|
|
//! behavior and handle cases where all retries are exhausted.
|
|
|
|
use intel_dcap_api::{ApiClient, IntelApiError};
|
|
|
|
#[tokio::main]
|
|
async fn main() -> Result<(), Box<dyn std::error::Error>> {
|
|
// Create API client with default settings (3 retries)
|
|
let mut client = ApiClient::new()?;
|
|
|
|
println!("Example 1: Default behavior (automatic retries)");
|
|
println!("================================================");
|
|
|
|
// Example FMSPC value
|
|
let fmspc = "00606A000000";
|
|
|
|
// The client will automatically retry up to 3 times if rate limited
|
|
match client.get_sgx_tcb_info(fmspc, None, None).await {
|
|
Ok(tcb_info) => {
|
|
println!("✓ Successfully retrieved TCB info");
|
|
println!(
|
|
" TCB Info JSON length: {} bytes",
|
|
tcb_info.tcb_info_json.len()
|
|
);
|
|
println!(
|
|
" Issuer Chain length: {} bytes",
|
|
tcb_info.issuer_chain.len()
|
|
);
|
|
}
|
|
Err(IntelApiError::TooManyRequests {
|
|
request_id,
|
|
retry_after,
|
|
}) => {
|
|
println!("✗ Rate limited even after 3 automatic retries");
|
|
println!(" Request ID: {}", request_id);
|
|
println!(" Last retry-after was: {} seconds", retry_after);
|
|
}
|
|
Err(e) => {
|
|
eprintln!("✗ Other error: {}", e);
|
|
}
|
|
}
|
|
|
|
println!("\nExample 2: Custom retry configuration");
|
|
println!("=====================================");
|
|
|
|
// Configure client to retry up to 5 times
|
|
client.set_max_retries(5);
|
|
println!("Set max retries to 5");
|
|
|
|
match client.get_sgx_tcb_info(fmspc, None, None).await {
|
|
Ok(_) => println!("✓ Request succeeded"),
|
|
Err(IntelApiError::TooManyRequests { .. }) => {
|
|
println!("✗ Still rate limited after 5 retries")
|
|
}
|
|
Err(e) => eprintln!("✗ Error: {}", e),
|
|
}
|
|
|
|
println!("\nExample 3: Disable automatic retries");
|
|
println!("====================================");
|
|
|
|
// Disable automatic retries
|
|
client.set_max_retries(0);
|
|
println!("Disabled automatic retries");
|
|
|
|
match client.get_sgx_tcb_info(fmspc, None, None).await {
|
|
Ok(_) => println!("✓ Request succeeded on first attempt"),
|
|
Err(IntelApiError::TooManyRequests {
|
|
request_id,
|
|
retry_after,
|
|
}) => {
|
|
println!("✗ Rate limited (no automatic retry)");
|
|
println!(" Request ID: {}", request_id);
|
|
println!(" Retry after: {} seconds", retry_after);
|
|
println!(" You would need to implement manual retry logic here");
|
|
}
|
|
Err(e) => eprintln!("✗ Error: {}", e),
|
|
}
|
|
|
|
println!("\nNote: The client handles rate limiting automatically!");
|
|
println!("You only need to handle TooManyRequests errors if:");
|
|
println!("- You disable automatic retries (set_max_retries(0))");
|
|
println!("- All automatic retries are exhausted");
|
|
|
|
Ok(())
|
|
}
|