actions/cargo-binstall
1
0
Fork 0
mirror of https://github.com/cargo-bins/cargo-binstall.git synced 2025-06-17 08:06:38 +00:00
Code Issues Projects Releases Packages Wiki Activity

Fix launch_baseline_find_tasks: Retry on rate limit

Browse source 
Signed-off-by: Jiahao XU <Jiahao_XU@outlook.com>
This commit is contained in:
Jiahao XU 2024-06-06 00:03:15 +10:00
parent 53b3c02b3b
commit 6b3e38453f
No known key found for this signature in database
GPG key ID: 76D1E687CA3C4928
1 changed files with 20 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

31
crates/binstalk-fetchers/src/gh_crate_meta.rs
View file

@ -1,4 +1,4 @@
use std::{borrow::Cow, fmt, iter, path::Path, sync::Arc};
use std::{borrow::Cow, fmt, iter, path::Path, sync::Arc, time::Duration};
use binstalk_git_repo_api::gh_api_client::{GhApiError, GhReleaseArtifact, GhReleaseArtifactUrl};
use compact_str::{CompactString, ToCompactString};
@ -6,6 +6,7 @@ use either::Either;
use leon::Template;
use once_cell::sync::OnceCell;
use strum::IntoEnumIterator;
use tokio::time::sleep;
use tracing::{debug, info, trace, warn};
use url::Url;
@ -14,6 +15,8 @@ use crate::{
SignaturePolicy, SignatureVerifier, TargetDataErased,
};
static DEFAULT_GH_API_RETRY_DURATION: Duration = Duration::from_secs(1);
pub(crate) mod hosting;
pub struct GhCrateMeta {
@ -102,17 +105,23 @@ impl GhCrateMeta {
};
if let Some(artifact) = gh_release_artifact {
match get_gh_release_artifact_url(gh_api_client, artifact).await {
Ok(Some(artifact_url)) => {
resolved.gh_release_artifact_url = Some(artifact_url);
return Ok(Some(resolved));
loop {
match get_gh_release_artifact_url(gh_api_client.clone(), artifact.clone())
.await
{
Ok(Some(artifact_url)) => {
resolved.gh_release_artifact_url = Some(artifact_url);
return Ok(Some(resolved));
}
Ok(None) => return Ok(None),
Err(GhApiError::RateLimit { retry_after }) => {
sleep(retry_after.unwrap_or(DEFAULT_GH_API_RETRY_DURATION)).await;
}
Err(GhApiError::Unauthorized) if !is_repo_private => break,
Err(err) => return Err(err.into()),
}
Ok(None) => return Ok(None),
Err(GhApiError::RateLimit { .. }) => (),
Err(GhApiError::Unauthorized) if !is_repo_private => (),
Err(err) => return Err(err.into()),
}
}