about summary refs log tree commit diff
path: root/kittybox-rs/src/main.rs
diff options
context:
space:
mode:
Diffstat (limited to 'kittybox-rs/src/main.rs')
-rw-r--r--kittybox-rs/src/main.rs7
1 files changed, 4 insertions, 3 deletions
diff --git a/kittybox-rs/src/main.rs b/kittybox-rs/src/main.rs
index 59c3e69..4fb0eec 100644
--- a/kittybox-rs/src/main.rs
+++ b/kittybox-rs/src/main.rs
@@ -174,8 +174,8 @@ async fn main() {
                 "/.kittybox/health",
                 axum::routing::get(
                     |axum::Extension(db): axum::Extension<FileStorage>| async move {
-                    // TODO health-check the database
-                    "OK"
+                        // TODO health-check the database
+                        "OK"
                     }
                 )
                     .layer(axum::Extension(database))
@@ -195,7 +195,8 @@ async fn main() {
             .layer(axum::Extension(kittybox::tokenauth::TokenEndpoint(token_endpoint)))
             .layer(tower::ServiceBuilder::new()
                    .layer(tower_http::trace::TraceLayer::new_for_http())
-                   .into_inner());
+                   .into_inner())
+            .layer(tower_http::catch_panic::CatchPanicLayer::new());
 
         // A little dance to turn a potential file descriptor into
         // a guaranteed async network socket