From f46a5bf4924bb7cc702d677512d4aa9831254cb9 Mon Sep 17 00:00:00 2001 From: Andrew4Coding <andrewdevitoaryo@gmail.com> Date: Fri, 28 Mar 2025 11:35:43 +0800 Subject: [PATCH] Implement publish function in Program service and Program controller. --- src/controller/mod.rs | 2 +- src/controller/product.rs | 9 +++++++++ src/service/product.rs | 18 ++++++++++++++++++ 3 files changed, 28 insertions(+), 1 deletion(-) diff --git a/src/controller/mod.rs b/src/controller/mod.rs index df6df1a..9107e72 100644 --- a/src/controller/mod.rs +++ b/src/controller/mod.rs @@ -6,7 +6,7 @@ use rocket::fairing::AdHoc; pub fn route_stage() -> AdHoc { return AdHoc::on_ignite("Initializing controller routes...", |rocket| async { rocket - .mount("/product", routes![product::create, product::list, product::read, product::delete]) + .mount("/product", routes![product::create, product::list, product::read, product::delete, product::publish]) .mount("/notification", routes![notification::subscribe, notification::unsubscribe]) }); } \ No newline at end of file diff --git a/src/controller/product.rs b/src/controller/product.rs index 1cc5384..9e976c8 100644 --- a/src/controller/product.rs +++ b/src/controller/product.rs @@ -37,3 +37,12 @@ pub fn delete(id: usize) -> Result<Json<Product>> { Err(e) => Err(e) }; } + + +#[post("/<id>/publish")] +pub fn publish(id: usize) -> Result<Json<Product>> { + return match ProductService::publish(id) { + Ok(f) => Ok(Json::from(f)), + Err(e) => Err(e) + }; +} \ No newline at end of file diff --git a/src/service/product.rs b/src/service/product.rs index 775bf09..44424d0 100644 --- a/src/service/product.rs +++ b/src/service/product.rs @@ -5,6 +5,8 @@ use bambangshop::{Result, compose_error_response}; use crate::model::product::Product; use crate::repository::product::ProductRepository; +use super::notification::NotificationService; + pub struct ProductService; impl ProductService { @@ -42,4 +44,20 @@ impl ProductService { return Ok(Json::from(product)); } + + pub fn publish(id: usize) -> Result<Product> { + let product_opt = ProductRepository::get_by_id(id); + if product_opt.is_none() { + return Err(compose_error_response( + Status::NotFound, + String::from("Product not found.") + )); + }; + + let product = product_opt.unwrap(); + + NotificationService.notify(&product.product_type, "PROMOTION", product.clone()); + + return Ok(product); + } } -- GitLab