From 18b337e792f073b2b1a5c365c9a093da6a8559e6 Mon Sep 17 00:00:00 2001
From: DawnFall19 <ignasius.michael09@gmail.com>
Date: Fri, 28 Mar 2025 16:03:10 +0700
Subject: [PATCH] Implement unsubscribe function in Notification controller.
---
src/controller/mod.rs | 2 +-
src/controller/notification.rs | 9 +++++++++
2 files changed, 10 insertions(+), 1 deletion(-)
diff --git a/src/controller/mod.rs b/src/controller/mod.rs
index 1567487..fe4a544 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 e1420a2..851bd7f 100644
--- a/src/controller/notification.rs
+++ b/src/controller/notification.rs
@@ -1,3 +1,4 @@
+use reqwest::Url;
use rocket::response::status::Created;
use rocket::serde::json::Json;
@@ -14,4 +15,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
--
GitLab