mirrors/magic-nix-cache
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

Move the tripping to a separate impl on the atomicbool

Browse source
This commit is contained in:
Graham Christensen 2024-06-13 13:07:00 -04:00
parent 25359d9b17
commit 51bb9f972d
1 changed files with 31 additions and 27 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

58
gha-cache/src/api.rs
View file

@ -54,7 +54,9 @@ pub enum Error {
#[error("Failed to initialize the client: {0}")] #[error("Failed to initialize the client: {0}")]
InitError(Box<dyn std::error::Error + Send + Sync>), InitError(Box<dyn std::error::Error + Send + Sync>),
#[error("GitHub Actions Cache throttled Magic Nix Cache. Not trying to use it again on this run.")] #[error(
"GitHub Actions Cache throttled Magic Nix Cache. Not trying to use it again on this run."
)]
CircuitBreakerTripped, CircuitBreakerTripped,
#[error("Request error: {0}")] #[error("Request error: {0}")]
@ -280,18 +282,6 @@ impl Api {
self.circuit_breaker_429_tripped.load(Ordering::Relaxed) self.circuit_breaker_429_tripped.load(Ordering::Relaxed)
} }
fn circuit_breaker_trip_on_429(&self, e: &Error) {
if let Error::ApiError {
status: reqwest::StatusCode::TOO_MANY_REQUESTS,
info: ref _info,
} = e
{
tracing::info!("Disabling GitHub Actions Cache due to 429: Too Many Requests");
self.circuit_breaker_429_tripped
.store(true, Ordering::Relaxed);
}
}
/// Mutates the cache version/namespace. /// Mutates the cache version/namespace.
pub fn mutate_version(&mut self, data: &[u8]) { pub fn mutate_version(&mut self, data: &[u8]) {
self.version_hasher.update(data); self.version_hasher.update(data);
@ -408,13 +398,7 @@ impl Api {
drop(permit); drop(permit);
if let Err(Error::ApiError { circuit_breaker_429_tripped.check_result(&r);
status: reqwest::StatusCode::TOO_MANY_REQUESTS,
info: ref _info,
}) = r
{
circuit_breaker_429_tripped.store(true, Ordering::Relaxed);
}
r r
}) })
@ -475,9 +459,7 @@ impl Api {
.check_json() .check_json()
.await; .await;
if let Err(ref e) = res { self.circuit_breaker_429_tripped.check_result(&res);
self.circuit_breaker_trip_on_429(e);
}
match res { match res {
Ok(entry) => Ok(Some(entry)), Ok(entry) => Ok(Some(entry)),
@ -520,9 +502,7 @@ impl Api {
.check_json() .check_json()
.await; .await;
if let Err(ref e) = res { self.circuit_breaker_429_tripped.check_result(&res);
self.circuit_breaker_trip_on_429(e);
}
res res
} }
@ -549,7 +529,7 @@ impl Api {
.check() .check()
.await .await
{ {
self.circuit_breaker_trip_on_429(&e); self.circuit_breaker_429_tripped.check_err(&e);
return Err(e); return Err(e);
} }
@ -619,3 +599,27 @@ async fn handle_error(res: reqwest::Response) -> Error {
Error::ApiError { status, info } Error::ApiError { status, info }
} }
trait AtomicCircuitBreaker {
fn check_err(&self, e: &Error);
fn check_result<T>(&self, r: &std::result::Result<T, Error>);
}
impl AtomicCircuitBreaker for AtomicBool {
fn check_result<T>(&self, r: &std::result::Result<T, Error>) {
if let Err(ref e) = r {
self.check_err(e)
}
}
fn check_err(&self, e: &Error) {
if let Error::ApiError {
status: reqwest::StatusCode::TOO_MANY_REQUESTS,
info: ref _info,
} = e
{
tracing::info!("Disabling GitHub Actions Cache due to 429: Too Many Requests");
self.store(true, Ordering::Relaxed);
}
}
}