pass in notifier to update_database func

src/cli/main.rs

@@ -247,16 +247,9 @@ async fn update_database(
     mut db: LocalBrokerDb,
     collectors: Vec<Collector>,
     days: Option<u32>,
+    notifier: &Option<NatsNotifier>,
     send_heartbeat: bool,
 ) {
-    let notifier = match NatsNotifier::new(None).await {
-        Ok(n) => Some(n),
-        Err(e) => {
-            error!("want to set up notifier but failed: {}", e);
-            None
-        }
-    };
-
     let now = Utc::now();
 
     let latest_ts_map: HashMap<String, NaiveDateTime> = db
@@ -311,7 +304,7 @@ async fn update_database(
             Ok(items) => {
                 let inserted = db.insert_items(&items, true).await.unwrap();
                 if !inserted.is_empty() {
-                    if let Some(n) = &notifier {
+                    if let Some(n) = notifier {
                         if let Err(e) = n.send(&inserted).await {
                             error!("{}", e);
                         }
@@ -412,16 +405,24 @@ fn main() {
                     let rt = get_tokio_runtime();
 
                     let collectors = load_collectors().unwrap();
-
                     rt.block_on(async {
+                        let notifier = match NatsNotifier::new(None).await {
+                            Ok(n) => Some(n),
+                            Err(_e) => {
+                                info!("no nats notifier configured, skip pushing notification");
+                                None
+                            }
+                        };
+
                         let db = LocalBrokerDb::new(path.as_str()).await.unwrap();
                         let mut interval =
                             tokio::time::interval(std::time::Duration::from_secs(update_interval));
 
                         loop {
                             interval.tick().await;
                             // updating from the latest data available
-                            update_database(db.clone(), collectors.clone(), None, true).await;
+                            update_database(db.clone(), collectors.clone(), None, &notifier, true)
+                                .await;
                             info!("wait for {} seconds before next update", update_interval);
                         }
                     });
@@ -535,7 +536,14 @@ fn main() {
 
             rt.block_on(async {
                 let db = LocalBrokerDb::new(&db_path).await.unwrap();
-                update_database(db, collectors, days, false).await;
+                let notifier = match NatsNotifier::new(None).await {
+                    Ok(n) => Some(n),
+                    Err(_e) => {
+                        info!("no nats notifier configured, skip pushing notification");
+                        None
+                    }
+                };
+                update_database(db, collectors, days, &notifier, false).await;
             });
         }
         Commands::Search { query, json, url } => {
@@ -644,6 +652,7 @@ fn main() {
                         return;
                     }
                 };
+
                 if let Err(e) = notifier.start_subscription(subject).await {
                     error!("{}", e);
                     return;