cleanup: safely get pg conn in create_message_query

devflowinc · Aug 26, 2024 · bc16798 · bc16798
1 parent 50d437b
commit bc16798
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/server/src/operators/message_operator.rs b/server/src/operators/message_operator.rs
@@ -69,7 +69,15 @@ pub async fn create_message_query(
 ) -> Result<(), ServiceError> {
     use crate::data::schema::messages::dsl::messages;
 
-    let mut conn = pool.get().await.unwrap();
+    let mut conn = match pool.get().await {
+        Ok(conn) => conn,
+        Err(e) => {
+            log::error!("Error getting connection from pool: {:?}", e);
+            return Err(ServiceError::InternalServerError(
+                "Error getting postgres connection from pool".into(),
+            ));
+        }
+    };
 
     diesel::insert_into(messages)
         .values(&new_message)