slightly improved http error status codes

spruceid · Jul 26, 2023 · 4f1e5fa · 4f1e5fa
1 parent 03bad6a
commit 4f1e5fa
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/src/routes/mod.rs b/src/routes/mod.rs
@@ -69,6 +69,7 @@ pub async fn delegate(
                 (
                     match e {
                         TxError::OrbitNotFound => Status::NotFound,
+                        TxError::DbErr(DbErr::ConnectionAcquire) => Status::ServiceUnavailable,
                         _ => Status::Unauthorized,
                     },
                     e.to_string(),
@@ -183,7 +184,16 @@ pub async fn invoke(
                     _ => unreachable!(),
                 },
             )
-            .map_err(|e| (Status::Unauthorized, e.to_string()));
+            .map_err(|e| {
+                (
+                    match e {
+                        TxError::OrbitNotFound => Status::NotFound,
+                        TxError::DbErr(DbErr::ConnectionAcquire) => Status::ServiceUnavailable,
+                        _ => Status::Unauthorized,
+                    },
+                    e.to_string(),
+                )
+            });
 
         timer.observe_duration();
         res