diff --git a/src/controller/mod.rs b/src/controller/mod.rs index 6d4d97cf7405e3354b43edee0c7916ba2031eba8..938380099ff468945c84bf573f833b23d23d989c 100644 --- a/src/controller/mod.rs +++ b/src/controller/mod.rs @@ -3,6 +3,6 @@ pub mod notification; pub fn route_stage() -> AdHoc { return AdHoc::on_ignite("Initializing controller routes...", |rocket| async { - rocket.mount("/", routes![]) + rocket.mount("/", routes![notification::subscribe]) }); } diff --git a/src/controller/notification.rs b/src/controller/notification.rs index 0fa74ed62c181f07ec2edfda73a5249a9b820e97..8e6573f89b3c723613a0b0786151092f7e328acb 100644 --- a/src/controller/notification.rs +++ b/src/controller/notification.rs @@ -4,3 +4,11 @@ use crate::model::notification::Notification; use crate::model::subscriber::SubscriberRequest; use crate::service::notification::NotificationService; use bambangshop_receiver::Result; + +#[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), + }; +}