Handle error for Post request

Cargo.toml

@@ -19,6 +19,7 @@ tracing-subscriber = "0.3.17"
 tower-http = { version = "0.4.3", features = ["trace"] }
 md5 = "0.7.0"
 base62 = "2.0.2"
+anyhow = "1.0.73"
 
 [dependencies.sqlx]
 version = "0.7.1"

src/app_error.rs

@@ -0,0 +1,16 @@
+use axum::{
+    http::StatusCode,
+    response::{IntoResponse, Response},
+};
+
+pub struct AppError(pub anyhow::Error);
+
+impl IntoResponse for AppError {
+    fn into_response(self) -> Response {
+        (
+            StatusCode::INTERNAL_SERVER_ERROR,
+            format!("Something went wrong: {}", self.0),
+        )
+            .into_response()
+    }
+}

src/lib.rs

@@ -1,3 +1,4 @@
+pub mod app_error;
 pub mod configuration;
 pub mod key_generator;
 pub mod model;

src/routes/full_url.rs

@@ -1,6 +1,8 @@
 use crate::{
-    key_generator::generate, model::UrlRequestModel, schema::CreateShortUrlSchema, AppState,
+    app_error::AppError, key_generator::generate, model::UrlRequestModel,
+    schema::CreateShortUrlSchema, AppState,
 };
+use anyhow::anyhow;
 use axum::{extract::State, Json};
 use chrono::Utc;
 use sqlx::Error;
@@ -10,30 +12,25 @@ use uuid::Uuid;
 pub async fn full_url(
     State(data): State<Arc<AppState>>,
     Json(payload): Json<CreateShortUrlSchema>,
-) -> String {
+) -> Result<String, AppError> {
     let mut retry_count = 3;
 
     while retry_count > 0 {
         let query_result = insert_in_db(&payload.url, State(data.clone())).await;
 
         match query_result {
             Ok(request) => {
-                println!("okay received testing: {:?}", request);
-                return request.extension;
+                return Ok(request.extension);
             }
             Err(e) => {
-                println!("failed with error: {:?}", e);
-                //TODO : - For now assume is failing because of error code 23505, that stands for
-                //duplicate key
                 if retry_count == 1 {
-                    return "error".to_string();
+                    return Err(AppError(e.into()));
                 }
                 retry_count -= 1;
             }
         }
     }
-
-    "internal server error".to_string()
+    Err(AppError(anyhow!("Something went wrong")))
 }
 
 async fn insert_in_db(