diff --git a/src/controller/mod.rs b/src/controller/mod.rs index 9d3748372824be39fe02e9c3e201cf252d0a4266..63b88722379049e997798dd4305733e2a0d56e1a 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![]) + .mount("/", routes![notification::subscribe]) }); } diff --git a/src/controller/notification.rs b/src/controller/notification.rs index 063a3209acba95c83d4afba919876c03d9214663..ca866c49c85f7be5579588c761c541551d94db54 100644 --- a/src/controller/notification.rs +++ b/src/controller/notification.rs @@ -3,4 +3,12 @@ use rocket::serde::json::Json; use bambangshop_receiver::Result; use crate::model::notification::Notification; use crate::model::subscriber::SubscriberRequest; -use crate::service::notification::NotificationService; \ No newline at end of file +use crate::service::notification::NotificationService; + +#[get("/<product_type>")] +pub fn subscribe(product_type: &str) -> Result<Json<SubscriberRequest>> { + return match NotificationService::subscribe(product_type) { + Ok(r) => Ok(Json::from(r)), + Err(e) => Err(e) + }; +} \ No newline at end of file