diff --git a/src/controller/mod.rs b/src/controller/mod.rs index fb4642da4f170b27aaaec35bb2797ec379b38352..5124c8193f66beddad34ceb71264aa53f3f937cd 100644 --- a/src/controller/mod.rs +++ b/src/controller/mod.rs @@ -15,6 +15,6 @@ pub fn route_stage() -> AdHoc { product::delete ], ) - .mount("/notification", routes![]) + .mount("/notification", routes![notification::subscribe]) }); } diff --git a/src/controller/notification.rs b/src/controller/notification.rs index 17316a91906b51a2a5d368ee47f8325f6293ad67..bc0ea02f075ff3207a974ad9e8bb1036a70dda3e 100644 --- a/src/controller/notification.rs +++ b/src/controller/notification.rs @@ -4,3 +4,14 @@ use rocket::serde::json::Json; use crate::model::subscriber::Subscriber; use crate::service::notification::NotificationService; use bambangshop::Result; + +#[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), + }; +}