From fc4279206e844481d9f3d7e788d18c4816043760 Mon Sep 17 00:00:00 2001
From: DawnFall19 <ignasius.michael09@gmail.com>
Date: Fri, 28 Mar 2025 16:59:09 +0700
Subject: [PATCH] Implement publish function in Program service and Program
controller.
---
src/controller/mod.rs | 3 ++-
src/controller/product.rs | 8 ++++++++
src/service/product.rs | 19 +++++++++++++++++++
3 files changed, 29 insertions(+), 1 deletion(-)
diff --git a/src/controller/mod.rs b/src/controller/mod.rs
index fe4a544..349f17e 100644
--- a/src/controller/mod.rs
+++ b/src/controller/mod.rs
@@ -6,7 +6,8 @@ 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])
});
}
diff --git a/src/controller/product.rs b/src/controller/product.rs
index 1cc5384..6802314 100644
--- a/src/controller/product.rs
+++ b/src/controller/product.rs
@@ -37,3 +37,11 @@ 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..c928891 100644
--- a/src/service/product.rs
+++ b/src/service/product.rs
@@ -1,10 +1,15 @@
+use std::ops::Not;
+
use rocket::http::Status;
use rocket::serde::json::Json;
use bambangshop::{Result, compose_error_response};
+use crate::model::notification::Notification;
use crate::model::product::Product;
use crate::repository::product::ProductRepository;
+use super::notification::NotificationService;
+
pub struct ProductService;
impl ProductService {
@@ -42,4 +47,18 @@ impl ProductService {
return Ok(Json::from(product));
}
+
+ pub fn publish(id: usize) -> Result<Product> {
+ let product_opt: Option<Product> = 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 = product_opt.unwrap();
+
+ NotificationService.notify(&product.product_type, "PROMOTION", product.clone());
+ return Ok(product);
+ }
}
--
GitLab