From cca92a7ad9670e46556d5b35897394a506d47373 Mon Sep 17 00:00:00 2001
From: Baitinq <manuelpalenzuelamerino@gmail.com>
Date: Thu, 27 Oct 2022 12:49:21 +0200
Subject: Crawler: Abstract database word fetching with search_word_in_db()

---
 indexer/src/main.rs | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

(limited to 'indexer')

diff --git a/indexer/src/main.rs b/indexer/src/main.rs
index aefcc2b..d1ad867 100644
--- a/indexer/src/main.rs
+++ b/indexer/src/main.rs
@@ -109,12 +109,13 @@ async fn search(data: web::Data<AppState>, term: web::Path<String>) -> impl Resp
 
     let mut valid_results: Option<HashSet<CrawledResource>> = None;
     for w in query {
-        let curr_word_results = match database.get(w) {
+        let curr_word_results = match search_word_in_db(&database, w) {
             None => return "[]".to_string(),
-            Some(results) => results,
+            Some(curr_results) => curr_results,
         };
 
         match valid_results {
+            //Initialise valid_results
             None => {
                 valid_results = Some(curr_word_results.to_owned());
             }
@@ -131,6 +132,13 @@ async fn search(data: web::Data<AppState>, term: web::Path<String>) -> impl Resp
     serde_json::to_string(&valid_results.unwrap()).unwrap()
 }
 
+fn search_word_in_db<'a>(
+    db: &'a HashMap<String, HashSet<CrawledResource>>,
+    word: &'a str,
+) -> Option<&'a HashSet<CrawledResource>> {
+    db.get(word)
+}
+
 //TODO!
 fn calculate_word_priority(_word: &str, _html_site: &str) -> u32 {
     rand::thread_rng().gen::<u32>()
-- 
cgit 1.4.1