Handle deadpool_postgres::PoolError

DigitalPulseSoftware · Jul 14, 2024 · 2964cbc · 2964cbc
1 parent 23b7c27
commit 2964cbc
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/src/errors/api.rs b/src/errors/api.rs
@@ -69,6 +69,13 @@ error_from! { transform tokio_postgres::Error, RouteError, |value| {
     )
 } }
 
+error_from! { transform deadpool_postgres::PoolError, RouteError, |value| {
+    RouteError::ServerError(
+        ErrorCause::Database,
+        ErrorCode::External(value.to_string())
+    )
+} }
+
 impl ResponseError for RouteError {
     fn status_code(&self) -> StatusCode {
         match self {

diff --git a/src/players.rs b/src/players.rs
@@ -59,7 +59,7 @@ async fn player_create(
 
     let uuid = Uuid::new_v4();
 
-    let mut pg_client = pg_pool.get().await.unwrap();
+    let mut pg_client = pg_pool.get().await?;
 
     let create_player_statement = pg_client
         .prepare_typed_cached(
@@ -124,7 +124,7 @@ async fn player_authenticate(
         )));
     }
 
-    let pg_client = pg_pool.get().await.unwrap();
+    let pg_client = pg_pool.get().await?;
 
     let find_token_statement = pg_client
         .prepare_typed_cached(