diff --git a/src/repository/subscriber.rs b/src/repository/subscriber.rs index 857ec9c26549b0cca2f38fe8a03660c14a038c5a..e92612da8f4fc204e569dcce8d1d23d5c2345747 100644 --- a/src/repository/subscriber.rs +++ b/src/repository/subscriber.rs @@ -3,11 +3,22 @@ use lazy_static::lazy_static; use crate::model::subscriber::Subscriber; lazy_static! { - pub static ref SUBSCRIBERS: DashMap<String, Subscriber> = DashMap::new(); + pub static ref SUBSCRIBERS: DashMap<String, DashMap<String, Subscriber>> = DashMap::new(); } pub struct SubscriberRepository; impl SubscriberRepository { - + 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 subscriber; + } } \ No newline at end of file