From f48990e98e95aa524dcf25e17f2f9404d439b67f Mon Sep 17 00:00:00 2001
From: DawnFall19 <ignasius.michael09@gmail.com>
Date: Fri, 28 Mar 2025 15:58:30 +0700
Subject: [PATCH] Implement subscribe function in Notification controller.
---
src/controller/mod.rs | 2 +-
src/controller/notification.rs | 13 ++++++++++++-
2 files changed, 13 insertions(+), 2 deletions(-)
diff --git a/src/controller/mod.rs b/src/controller/mod.rs
index 08a540d..1567487 100644
--- a/src/controller/mod.rs
+++ b/src/controller/mod.rs
@@ -7,6 +7,6 @@ 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("/notification", routes![])
+ .mount("/notification", routes![notification::subscribe])
});
}
diff --git a/src/controller/notification.rs b/src/controller/notification.rs
index 7305876..e1420a2 100644
--- a/src/controller/notification.rs
+++ b/src/controller/notification.rs
@@ -2,5 +2,16 @@ use rocket::response::status::Created;
use rocket::serde::json::Json;
use bambangshop::Result;
+use crate::model::notification::Notification;
use crate::model::subscriber::Subscriber;
-use crate::service::notification::NotificationService;
\ No newline at end of file
+use crate::service::notification::NotificationService;
+
+use super::product;
+
+#[post("/subscribe/<product_type>", data = "<subscriber>")]
+pub fn subscribe(product_type: &str, subscriber: Json<Subscriber>) -> Result<Created<Json<Subscriber>>> {
+ return match NotificationService::subscribe(product_type, subscriber.into_inner()) {
+ Ok(f) => Ok(Created::new("/").body(Json::from(f))),
+ Err(e) => Err(e)
+ };
+}
\ No newline at end of file
--
GitLab