diff --git a/src/controller/mod.rs b/src/controller/mod.rs index ef4a9cee11a90f2815de14c2ae3988bcaac5f354..346ff5d623c642f054ca2ad327268c45e3b56287 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![notification::subscribe]) + .mount("notification", routes![notification::subscribe, notification::unsubscribe]) }); } diff --git a/src/controller/notification.rs b/src/controller/notification.rs index e64485726a2ffef5e4169f5a1d0a0e01fe0f5ea8..bfb69596b1d8da5272c7d437706a5dc4bff3a030 100644 --- a/src/controller/notification.rs +++ b/src/controller/notification.rs @@ -1,5 +1,5 @@ use rocket::response::status::Created; -use rocket::serde::json::Json; +use rocket::serde::json::{self, Json}; use bambangshop::Result; use crate::model::subscriber::{self, Subscriber}; @@ -12,3 +12,11 @@ pub fn subscribe(product_type: &str, subscriber: Json<Subscriber>) -> Result<Cre Err(e) => Err(e) } } + +#[post("/unsubscribe/<product_type>?<url>")] +pub fn unsubscribe(product_type: &str, url: &str) -> Result<Json<Subscriber>> { + return match NotificationService::unsubscribe(product_type, url) { + Ok(f) => Ok(Json::from(f)), + Err(e) => Err(e) + } +}