From 70f341c0007b88660d715351e640a3892fce1ccd Mon Sep 17 00:00:00 2001
From: TheoKevH <theodorekevinh@gmail.com>
Date: Fri, 28 Mar 2025 18:58:51 +0700
Subject: [PATCH] Implement add function in Subscriber repository

---
 src/repository/subscriber.rs | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/src/repository/subscriber.rs b/src/repository/subscriber.rs
index 3aab51b..a4cd9d1 100644
--- a/src/repository/subscriber.rs
+++ b/src/repository/subscriber.rs
@@ -10,4 +10,14 @@ lazy_static! {
 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
-- 
GitLab