Add fallback handler to axum Router (#258)

By default the fallback handler now logs the path that could not be found. This can be vital for debugging, as otherwise there is no feedback on the server if a method is called that isn't registered. Signed-off-by: Leon Matthes <[email protected]>
dapr · Jan 28, 2025 · 52f095f · 52f095f
1 parent a532ee1
commit 52f095f
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/dapr/src/server/http.rs b/dapr/src/server/http.rs
@@ -1,5 +1,5 @@
 use axum::{
-    extract::{Path, State},
+    extract::{OriginalUri, Path, State},
     http::StatusCode,
     response::IntoResponse,
     routing::{delete, get, put},
@@ -207,14 +207,23 @@ impl DaprHttpServer {
             .route(
                 "/actors/:actor_type/:actor_id/method/timer/:timer_name",
                 put(invoke_timer).with_state(rt.clone()),
-            );
+            )
+            .fallback(fallback_handler);
 
         self.actor_runtime
             .configure_method_routes(app, rt.clone())
             .await
     }
 }
 
+async fn fallback_handler(OriginalUri(uri): OriginalUri) -> impl IntoResponse {
+    log::warn!("Returning 404 for request: {uri}");
+    (
+        StatusCode::NOT_FOUND,
+        format!("The URI '{uri}' could not be found!"),
+    )
+}
+
 async fn health_check() -> impl IntoResponse {
     log::debug!("recieved health check request");
     StatusCode::OK