diff --git a/src/controller/mod.rs b/src/controller/mod.rs index 1567487f77b9d46c16d18eef8ebf1b8d571c60d8..fe4a544b65cec3c65bedd9d1fa666dc986aff6f2 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 6e4c2f83fce5892ee67a46eee39dbcb7d70942fc..aa4fa8c550ff20ece3c890b2e9ac0abfbf555ec2 100644 --- a/src/controller/notification.rs +++ b/src/controller/notification.rs @@ -11,4 +11,12 @@ pub fn subscribe(product_type: &str, subscriber: Json<Subscriber>) -> Result<Cre Ok(f) => Ok(Created::new("/").body(Json::from(f))), 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) + }; } \ No newline at end of file