only capture gateway errors

jamsocket · Feb 10, 2025 · 2e1b92d · 2e1b92d
1 parent c3a1aa3
commit 2e1b92d
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/proxy.rs b/src/proxy.rs
@@ -5,7 +5,7 @@ use axum::{
     response::Response,
 };
 use handlebars::Handlebars;
-use http::{uri::Authority, Uri};
+use http::{uri::Authority, StatusCode, Uri};
 use plane_dynamic_proxy::{
     body::{to_simple_body, SimpleBody},
     proxy::ProxyClient,
@@ -63,7 +63,9 @@ pub async fn proxy_request(
 
     let (response, handler) = client.request(req).await.expect("Infallable");
 
-    if !response.status().is_success() {
+    if response.status() == StatusCode::GATEWAY_TIMEOUT
+        || response.status() == StatusCode::BAD_GATEWAY
+    {
         return render_error(State(server_state.clone()), path).await;
     }