diff --git a/src/repository/subscriber.rs b/src/repository/subscriber.rs index fa3714bbec183f384475aa318375e45060929e35..51eeb32cb65f0db380cf3d0b37ab097d0278d89c 100644 --- a/src/repository/subscriber.rs +++ b/src/repository/subscriber.rs @@ -20,7 +20,7 @@ impl SubscriberRepository { return subscriber; } - + pub fn list_all(product_type: &str) -> Vec<Subscriber> { if SUBSCRIBERS.get(product_type).is_none() { SUBSCRIBERS.insert(String::from(product_type), DashMap::new()); @@ -29,4 +29,16 @@ impl SubscriberRepository { .map(|f| f.value().clone()).collect(); } + pub fn delete(product_type: &str, url: &str) -> Option<Subscriber> { + if SUBSCRIBERS.get(product_type).is_none() { + SUBSCRIBERS.insert(String::from(product_type), DashMap::new()); + } + let result = SUBSCRIBERS.get(product_type).unwrap() + .remove(url); + if !result.is_none() { + return Some(result.unwrap().1); + } + return None; + } + }