diff --git a/src/controller/mod.rs b/src/controller/mod.rs index 08a540d7202c90e21865ad69008ba590c1fb343a..1567487f77b9d46c16d18eef8ebf1b8d571c60d8 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 622e59354050dae49d5acc5032af464032c71e8b..4fbd079aa71fdd2fe622bf7b0b6b21bdb5884b4f 100644 --- a/src/controller/notification.rs +++ b/src/controller/notification.rs @@ -4,3 +4,11 @@ use rocket::serde::json::Json; use bambangshop::Result; use crate::model::subscriber::Subscriber; use crate::service::notification::NotificationService; + +#[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) + }; +}