diff --git a/src/service/notification.rs b/src/service/notification.rs index 9c57880e1d6c59c954462af094a4d52bfb9a9e38..843c426cb3c8da3f1e3753bd640f677786a38909 100644 --- a/src/service/notification.rs +++ b/src/service/notification.rs @@ -1,17 +1,50 @@ use std::thread; - use rocket::http::Status; - use rocket::log; +use rocket::http::Status; +use rocket::log; + +use rocket::serde::json::to_string; +use rocket::tokio; + +use bambangshop_receiver::{APP_CONFIG, REQWEST_CLIENT, Result, compose_error_response}; +use crate::model::notification::Notification; +use crate::model::subscriber::SubscriberRequest; +use crate::repository::notification::NotificationRepository; + +pub struct NotificationService; - use rocket::serde::json::to_string; - use rocket::tokio; - - use bambangshop_receiver::{APP_CONFIG, REQWEST_CLIENT, Result, compose_error_response}; - use crate::model::notification::Notification; - use crate::model::subscriber::SubscriberRequest; - use crate::repository::notification::NotificationRepository; - - pub struct NotificationService; - - impl NotificationService { - } \ No newline at end of file +impl NotificationService { + pub fn subscribe(product_type: &str) -> Result<SubscriberRequest> { + let product_type_clone = String::from(product_type); + return thread::spawn(move || Self::subscribe_request(product_type_clone)) + .join().unwrap(); + } +} + +#[tokio::main] +async fn subscribe_request(product_type: String) -> Result<SubscriberRequest> { + let product_type_upper: String = product_type.to_uppercase(); + let product_type_str: &str = product_type_upper.as_str(); + let notification_receiver_url: String = format!("{}/receive", APP_CONFIG.get_instance_root_url()); + let payload: SubscriberRequest = SubscriberRequest { + name: APP_CONFIG.get_instance_name().to_string(), + url: notification_receiver_url, + }; + + let request_url: String = format!("{}/notification/subscribe/{}", APP_CONFIG.get_publisher_root_url(), product_type_str); + let request = REQWEST_CLIENT + .post(request_url.clone()) + .header("Content-Type", "application/json") + .header("Accept", "application/json") + .body(to_string(&payload).unwrap()) + .send().await; + log::warn!("Sent subscribe request to: {}", request_url); + + return match request { + Ok(f) => match f.json::<SubscriberRequest>().await { + Ok(x) => Ok(x), + Err(y) => Err(compose_error_response(Status::NotAcceptable, y.to_string())), + }, + Err(e) => Err(compose_error_response(Status::NotFound, e.to_string())), + }; +} \ No newline at end of file