diff --git a/src/repository/subscriber.rs b/src/repository/subscriber.rs index 98b0faf96a7b80a7f5f8d0223f09ddcc5405b6ac..0615c68cc97b92cab7d2354196b28d5267fd0be8 100644 --- a/src/repository/subscriber.rs +++ b/src/repository/subscriber.rs @@ -29,4 +29,16 @@ impl SubscriberRepository { return SUBSCRIBERS.get(product_type).unwrap().iter() .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; + } } \ No newline at end of file