From 56e4da02f2268897267c070434ba02d6f19271fc Mon Sep 17 00:00:00 2001 From: TheoKevH <theodorekevinh@gmail.com> Date: Fri, 28 Mar 2025 20:22:23 +0700 Subject: [PATCH] Implement subscribe function in Notification controller --- src/controller/mod.rs | 2 +- src/controller/notification.rs | 8 ++++++++ 2 files changed, 9 insertions(+), 1 deletion(-) diff --git a/src/controller/mod.rs b/src/controller/mod.rs index 5f156e9..e5f9997 100644 --- a/src/controller/mod.rs +++ b/src/controller/mod.rs @@ -5,6 +5,6 @@ use rocket::fairing::AdHoc; pub fn route_stage() -> AdHoc { return AdHoc::on_ignite("Initializing controller routes...", |rocket| async { rocket - .mount("/", routes![]) + .mount("/", routes![notification::subscribe]) }); } diff --git a/src/controller/notification.rs b/src/controller/notification.rs index b189ea1..db0effa 100644 --- a/src/controller/notification.rs +++ b/src/controller/notification.rs @@ -4,3 +4,11 @@ use bambangshop_receiver::Result; use crate::model::notification::Notification; use crate::model::subscriber::SubscriberRequest; use crate::service::notification::NotificationService; + +#[get("/subscribe/<product_type>")] +pub fn subscribe(product_type: &str) -> Result<Json<SubscriberRequest>> { + return match NotificationService::subscribe(product_type) { + Ok(f) => Ok(Json::from(f)), + Err(e) => Err(e) + }; +} \ No newline at end of file -- GitLab