diff --git a/src/repository/subscriber.rs b/src/repository/subscriber.rs index d81e388bfe5fa51bcc629f5cb1eaef8623f5be2a..e56c06bb838dce446aa3c26fe8e83f9323947da9 100644 --- a/src/repository/subscriber.rs +++ b/src/repository/subscriber.rs @@ -10,4 +10,22 @@ lazy_static! { pub struct SubscriberRepository; impl SubscriberRepository { -} \ No newline at end of file + pub fn add(product_type: &str, subscriber: Subscriber) -> Subscriber { + let subscriber_value = subscriber.clone(); + if SUBSCRIBERS.get(product_type).is_none() { + SUBSCRIBERS.insert(String::from(product_type), DashMap::new()); + }; + + SUBSCRIBERS.get(product_type).unwrap() + .insert(subscriber_value.url.clone(), subscriber_value); + return subsriber; + } + + pub fn list_all(product_type: &str) -> Vec<Subscriber> { + if SUBSCRIBERS.get(product_type).is_none() { + SUBSCRIBER.insert(String::from(product_type), DashMap::new()); + } + return SUBSCRIBERS.get(product_type).unwrap().iter() + .map(|f| f.value().clone()).collect(); + } +} \ No newline at end of file