From 47aebf3e22fd7dcc64fd127fb0a9427a79808cd8 Mon Sep 17 00:00:00 2001
From: Greg Johnston <greg.johnston@gmail.com>
Date: Tue, 1 Apr 2025 19:25:59 -0400
Subject: [PATCH] fix: allow custom status codes or redirects for route
 fallbacks (#3808)

---
 integrations/axum/src/lib.rs | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/integrations/axum/src/lib.rs b/integrations/axum/src/lib.rs
index 1e1fc7bb8..61e842700 100644
--- a/integrations/axum/src/lib.rs
+++ b/integrations/axum/src/lib.rs
@@ -2061,7 +2061,15 @@ where
                         },
                     )
                     .await;
-                    *res.status_mut() = StatusCode::NOT_FOUND;
+
+                    // set the status to 404
+                    // but if the status was already set (for example, to a 302 redirect) don't
+                    // overwrite it
+                    let status = res.status_mut();
+                    if *status == StatusCode::OK {
+                        *res.status_mut() = StatusCode::NOT_FOUND;
+                    }
+
                     res
                 }
             }