feat: Make the Stream return a Document

This commit is contained in:
Clément Renault
2018-07-06 22:05:51 +02:00
parent 6fa164dc56
commit e8c24a0f07
3 changed files with 8 additions and 8 deletions

View File

@ -6,9 +6,9 @@ authors = ["Kerollmops <renault.cle@gmail.com>"]
[dependencies]
env_logger = { version = "0.3", default-features = false }
raptor = { path = "../raptor" }
elapsed = "0.1"
serde = "1.0"
serde_derive = "1.0"
elapsed = "0.1"
[dependencies.fst]
git = "https://github.com/Kerollmops/fst.git"

View File

@ -20,10 +20,10 @@ fn search(map: &DocIndexMap, lev_builder: &LevBuilder, db: &DB, query: &str) {
}
let mut stream = RankedStream::new(&map, map.values(), automatons, 20);
while let Some(document_id) = stream.next() {
print!("{:?} ", document_id);
while let Some(document) = stream.next() {
print!("{:?} ", document.document_id);
let title_key = format!("{}-title", document_id);
let title_key = format!("{}-title", document.document_id);
let title = db.get(title_key.as_bytes()).unwrap().unwrap();
let title = unsafe { from_utf8_unchecked(&title) };
print!("{:?}", title);