Add EntityNotFound --> 404 error handling

web/src/controller/organization_controller.rs

@@ -1,4 +1,4 @@
-use crate::AppState;
+use crate::{AppState, Error};
 use axum::extract::{Path, Query, State};
 use axum::response::IntoResponse;
 use axum::Json;
@@ -76,7 +76,7 @@ impl OrganizationController {
         State(app_state): State<AppState>,
         Path(id): Path<i32>,
         Query(organization_params): Query<organization::Model>,
-    ) -> impl IntoResponse {
+    ) -> Result<Json<entity::organization::Model>, Error> {
         debug!(
             "UPDATE the entire Organization by id: {}, new name: {}",
             id, organization_params.name
@@ -99,10 +99,10 @@ impl OrganizationController {
                     .await
                     .unwrap()
             }
-            None => organization::Model::default(),
+            None => return Err(Error::EntityNotFound),
         };
 
-        Json(updated_organization)
+        Ok(Json(updated_organization))
     }
 
     /// DELETE an Organization entity specified by its primary key

web/src/error.rs

@@ -5,12 +5,18 @@ pub type Result<T> = core::result::Result<T, Error>;
 
 #[derive(Debug)]
 pub enum Error {
-    PlaceholderError,
+    InternalServer,
+    EntityNotFound,
 }
 
 impl IntoResponse for Error {
     fn into_response(self) -> Response {
-        (StatusCode::INTERNAL_SERVER_ERROR, "INTERNAL SERVER ERROR").into_response()
+        match self {
+            Error::InternalServer => {
+                (StatusCode::INTERNAL_SERVER_ERROR, "INTERNAL SERVER ERROR").into_response()
+            }
+            Error::EntityNotFound => (StatusCode::NOT_FOUND, "ENTITY NOT FOUND").into_response(),
+        }
     }
 }