diff --git a/src/controller/mod.rs b/src/controller/mod.rs index 87c80418faf585d60151a9352cef6ab1ab6888b0..ef4a9cee11a90f2815de14c2ae3988bcaac5f354 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..e64485726a2ffef5e4169f5a1d0a0e01fe0f5ea8 100644 --- a/src/controller/notification.rs +++ b/src/controller/notification.rs @@ -2,5 +2,13 @@ use rocket::response::status::Created; use rocket::serde::json::Json; use bambangshop::Result; -use crate::model::subscriber::Subscriber; +use crate::model::subscriber::{self, 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) + } +}