gaspersic/solana
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

RpcClient now respects the retry-after server response header when getting rate limited

Browse Source 
(cherry picked from commit 7d1637d89a)
This commit is contained in:
Michael Vines
2021-05-05 15:27:44 -07:00
parent 970bba495f
commit fb163187b5
1 changed files with 19 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
client/src/http_sender.rs
View File

@ -7,7 +7,11 @@ use {
rpc_sender::RpcSender, rpc_sender::RpcSender,
}, },
log::*, log::*,
reqwest::{self, header::CONTENT_TYPE, StatusCode}, reqwest::{
self,
header::{CONTENT_TYPE, RETRY_AFTER},
StatusCode,
},
std::{ std::{
sync::{ sync::{
atomic::{AtomicU64, Ordering}, atomic::{AtomicU64, Ordering},
@ -83,14 +87,24 @@ impl RpcSender for HttpSender {
if response.status() == StatusCode::TOO_MANY_REQUESTS if response.status() == StatusCode::TOO_MANY_REQUESTS
&& too_many_requests_retries > 0 && too_many_requests_retries > 0
{ {
let mut duration = Duration::from_millis(500);
if let Some(retry_after) = response.headers().get(RETRY_AFTER) {
if let Ok(retry_after) = retry_after.to_str() {
if let Ok(retry_after) = retry_after.parse::<u64>() {
if retry_after < 120 {
duration = Duration::from_secs(retry_after);
}
}
}
}
too_many_requests_retries -= 1; too_many_requests_retries -= 1;
debug!( debug!(
"Server responded with {:?}, {} retries left", "Too many requests: server responded with {:?}, {} retries left, pausing for {:?}",
response, too_many_requests_retries response, too_many_requests_retries, duration
); );
// Sleep for 500ms to give the server a break sleep(duration);
sleep(Duration::from_millis(500));
continue; continue;
} }
return Err(response.error_for_status().unwrap_err().into()); return Err(response.error_for_status().unwrap_err().into());