Highlight all the matching words

This commit is contained in:
Kerollmops
2020-07-14 11:51:21 +02:00
parent 085c376655
commit 9ade00e27b
3 changed files with 18 additions and 8 deletions

View File

@ -58,7 +58,7 @@ fn main() -> anyhow::Result<()> {
let before = Instant::now();
let query = result?;
let documents_ids = index.search(&rtxn, &query)?;
let (_, documents_ids) = index.search(&rtxn, &query)?;
let headers = match index.headers(&rtxn)? {
Some(headers) => headers,
None => return Ok(()),

View File

@ -152,14 +152,21 @@ async fn main() -> anyhow::Result<()> {
let before_search = Instant::now();
let rtxn = env_cloned.read_txn().unwrap();
let documents_ids = index.search(&rtxn, &query.query).unwrap();
let (words, documents_ids) = index.search(&rtxn, &query.query).unwrap();
let mut body = Vec::new();
if let Some(headers) = index.headers(&rtxn).unwrap() {
// We write the headers
body.extend_from_slice(headers);
let re = Regex::new(r"(?i)(hello)").unwrap();
let mut regex = format!(r"(?i)\b(");
let number_of_words = words.len();
words.into_iter().enumerate().for_each(|(i, w)| {
regex.push_str(&w);
if i != number_of_words - 1 { regex.push('|') }
});
regex.push_str(r")\b");
let re = Regex::new(&regex).unwrap();
for id in documents_ids {
let content = index.documents.get(&rtxn, &BEU32::new(id)).unwrap();