Resolve "Implement DB Integration"

Closes #35 (closed)

src/predictors/basic_set.rs

@@ -102,12 +102,15 @@ pub fn from_file_path(file_path: &str) -> Result<SetPredictor, std::io::Error> {
     let mut reader = ReaderBuilder::new().from_reader(file);
     let mut markov_chain: SetPredictor = SetPredictor::new();
 
+    let mut count = 0;
     for result in reader.records() {
         let record = result?;
         if let Some(query) = record.get(5) {
             markov_chain.update(query);
+            count += 1;
         }
     }
+    println!("{} queries read from file", count);
 
     Ok(markov_chain)
 }
@@ -127,14 +130,17 @@ pub fn from_file_path_and_config(file_paths: Vec<&str>, config: HashMap<String,
         match File::open(path) {
             Ok(file) => {
                 println!("Reading data file...");
+                let mut count = 0;
                 let mut reader = ReaderBuilder::new().from_reader(file);
 
                 for result in reader.records() {
                     let record = result?;
                     if let Some(query) = record.get(5) {
                         markov_chain.update(query);
+                        count += 1;
                     }
                 }
+                println!("{} queries read from file", count);
             },
             Err(e) => {
                 println!("Error while reading: {}",e);