From a7a1745f9ce83bb27588f8f4867a6dc998658ce2 Mon Sep 17 00:00:00 2001
From: Andrew4Coding <andrewdevitoaryo@gmail.com>
Date: Fri, 28 Mar 2025 12:36:25 +0800
Subject: [PATCH] Implement subscribe function in Notification controller.

---
 src/controller/mod.rs          |  2 +-
 src/controller/notification.rs | 10 +++++++++-
 2 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/src/controller/mod.rs b/src/controller/mod.rs
index 002cadc..83e8f32 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![])
+            .mount("/", routes![notification::subscribe])
     });
 }
diff --git a/src/controller/notification.rs b/src/controller/notification.rs
index 063a320..a027fa8 100644
--- a/src/controller/notification.rs
+++ b/src/controller/notification.rs
@@ -3,4 +3,12 @@ use rocket::serde::json::Json;
 use bambangshop_receiver::Result;
 use crate::model::notification::Notification;
 use crate::model::subscriber::SubscriberRequest;
-use crate::service::notification::NotificationService;
\ No newline at end of file
+use crate::service::notification::NotificationService;
+
+#[get("/subscribe/<product_type>")]
+pub fn subscribe(product_type: &str) -> Result<Json<SubscriberRequest>> {
+    return match NotificationService::subscribe(product_type) {
+        Ok(f) => Ok(Json::from(f)),
+        Err(e) => Err(e)
+    }
+}
\ No newline at end of file
-- 
GitLab