diff --git a/src/controller/mod.rs b/src/controller/mod.rs index a97743537af2e94759840871e20ee8ac3d1f0024..2a921fbab019b84771dcc23125ad82e59f09a2ba 100644 --- a/src/controller/mod.rs +++ b/src/controller/mod.rs @@ -4,6 +4,6 @@ use rocket::fairing::AdHoc; pub fn route_stage() -> AdHoc { return AdHoc::on_ignite("Initializing controller routes...", |rocket| async { rocket - .mount("/", routes![notification::subscribe]) + .mount("/", routes![notification::subscribe, notification::unsubscribe]) }); } diff --git a/src/controller/notification.rs b/src/controller/notification.rs index 2bb89f10a537ee3921d5eafe31667b6eb118203c..cb6c6589c26635f75b5f73c45abaf9c6674ff7c8 100644 --- a/src/controller/notification.rs +++ b/src/controller/notification.rs @@ -11,4 +11,12 @@ pub fn subscribe(product_type: &str) -> Result<Json<SubscriberRequest>> { Ok(f) => Ok(Json::from(f)), Err(e) => Err(e) }; -} \ No newline at end of file +} + +#[get("/unsubscribe/<product_type>")] + pub fn unsubscribe(product_type: &str) -> Result<Json<SubscriberRequest>> { + return match NotificationService::unsubscribe(product_type) { + Ok(f) => Ok(Json::from(f)), + Err(e) => Err(e) + }; + } \ No newline at end of file