추가 소비 기록 수정 엔드포인트 만들기

Author: tyange

src/main.rs

@@ -26,6 +26,7 @@ use crate::routes::set_budget::set_budget;
 use crate::routes::update_budget::update_budget;
 use crate::routes::update_portfolio::update_portfolio;
 use crate::routes::update_post::update_post;
+use crate::routes::update_spending::update_spending;
 use crate::routes::upload_image::upload_image;
 use crate::{models::AppState, routes::add_user::add_user};
 use db::init_db;
@@ -101,7 +102,10 @@ async fn main() -> Result<(), std::io::Error> {
                 "/budget/spending",
                 get(get_spending).post(create_spending.with(ApiKeyAuth)),
             )
-            .at("/budget/spending/:record_id", delete(delete_spending))
+            .at(
+                "/budget/spending/:record_id",
+                put(update_spending).delete(delete_spending),
+            )
             .at("/budget/weekly", get(get_weekly_summary))
             .at("/budget/weekly/:week_key", get(get_weekly_summary_by_key))
             .nest("/images", StaticFilesEndpoint::new(upload_base_path))

src/models.rs

@@ -187,6 +187,13 @@ pub struct CreateSpendingRequest {
     pub transacted_at: String,
 }
 
+#[derive(Debug, Deserialize)]
+pub struct UpdateSpendingRequest {
+    pub amount: i64,
+    pub merchant: Option<String>,
+    pub transacted_at: String,
+}
+
 #[derive(Debug, Serialize)]
 pub struct CreateSpendingResponse {
     pub record_id: i64,

src/routes.rs

@@ -12,6 +12,7 @@ pub mod get_spending;
 pub mod login;
 pub mod update_portfolio;
 pub mod update_post;
+pub mod update_spending;
 pub mod upload_image;
 pub mod upload_post;
 pub mod get_posts_with_tags;

src/routes/update_spending.rs

@@ -0,0 +1,80 @@
+use std::sync::Arc;
+
+use poem::{
+    handler,
+    http::StatusCode,
+    web::{Data, Json, Path},
+    Error,
+};
+use sqlx::{query, query_as};
+
+use crate::{
+    models::{AppState, SpendingRecordResponse, UpdateSpendingRequest},
+    utils::parse_transacted_at,
+};
+
+#[handler]
+pub async fn update_spending(
+    Path(record_id): Path<i64>,
+    data: Data<&Arc<AppState>>,
+    Json(payload): Json<UpdateSpendingRequest>,
+) -> Result<Json<SpendingRecordResponse>, Error> {
+    if payload.amount <= 0 {
+        return Err(Error::from_string(
+            "amount는 0보다 커야 합니다.",
+            StatusCode::BAD_REQUEST,
+        ));
+    }
+
+    let transacted_at = parse_transacted_at(&payload.transacted_at).map_err(|_| {
+        Error::from_string(
+            "transacted_at 형식이 올바르지 않습니다. 예: 2026-03-03T12:20:00",
+            StatusCode::BAD_REQUEST,
+        )
+    })?;
+    let week_key = crate::utils::iso_week_key_from_datetime(&transacted_at);
+    let transacted_at_text = transacted_at.format("%Y-%m-%d %H:%M:%S").to_string();
+
+    let updated = query(
+        "UPDATE spending_records
+         SET amount = ?, merchant = ?, transacted_at = ?, week_key = ?
+         WHERE record_id = ?",
+    )
+    .bind(payload.amount)
+    .bind(payload.merchant)
+    .bind(&transacted_at_text)
+    .bind(&week_key)
+    .bind(record_id)
+    .execute(&data.db)
+    .await
+    .map_err(|e| {
+        Error::from_string(
+            format!("소비 기록 수정 실패: {}", e),
+            StatusCode::INTERNAL_SERVER_ERROR,
+        )
+    })?;
+
+    if updated.rows_affected() == 0 {
+        return Err(Error::from_string(
+            format!("해당 소비 기록(record_id={})을 찾을 수 없습니다.", record_id),
+            StatusCode::NOT_FOUND,
+        ));
+    }
+
+    let record = query_as::<_, SpendingRecordResponse>(
+        "SELECT record_id, amount, merchant, transacted_at, created_at
+         FROM spending_records
+         WHERE record_id = ?",
+    )
+    .bind(record_id)
+    .fetch_one(&data.db)
+    .await
+    .map_err(|e| {
+        Error::from_string(
+            format!("수정된 소비 기록 조회 실패: {}", e),
+            StatusCode::INTERNAL_SERVER_ERROR,
+        )
+    })?;
+
+    Ok(Json(record))
+}