Skip to content
Snippets Groups Projects

Resolve "Implement DB Integration"

Merged Phil Höfer requested to merge 35-implement-db-integration into main
4 files
+ 62
28
Compare changes
  • Side-by-side
  • Inline
Files
4
@@ -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);
Loading