diff --git a/src/controller/mod.rs b/src/controller/mod.rs index fe2103daa6029656640bce317f991b3d630377cd..fcdeefc1c14dabe88326a8d7b2c658aafd40f40f 100644 --- a/src/controller/mod.rs +++ b/src/controller/mod.rs @@ -1,5 +1,5 @@ pub mod product; -pub mod notification; +pub mod notification; use rocket::fairing::AdHoc; @@ -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]) }); } \ No newline at end of file diff --git a/src/controller/notification.rs b/src/controller/notification.rs index 73058760b5ff495d00487cef3f06ae44b42e54f0..0f18dd25caccdc2aeeda5ef920b6718ed27ea3bb 100644 --- a/src/controller/notification.rs +++ b/src/controller/notification.rs @@ -3,4 +3,12 @@ use rocket::serde::json::Json; use bambangshop::Result; use crate::model::subscriber::Subscriber; -use crate::service::notification::NotificationService; \ No newline at end of file +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) + }; +} \ No newline at end of file