snowykami/meilisearch
1
0
Fork 0
mirror of https://github.com/meilisearch/meilisearch.git synced 2024-11-23 10:37:41 +08:00
Code Issues Actions 8 Packages Projects Releases Wiki Activity

Correctly compute the bucket candidates for the Typo criterion

Browse Source
This commit is contained in:
Clément Renault 2021-02-17 17:45:56 +01:00 committed by Kerollmops
parent 5344abc008
commit 229130ed25
No known key found for this signature in database
GPG Key ID: 92ADA4E935E71FA4
1 changed files with 20 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
milli/src/search/criteria/typo.rs
View File

@ -16,6 +16,7 @@ pub struct Typo<'t> {
query_tree: Option<Operation>,
number_typos: u8,
candidates: Candidates,
bucket_candidates: Option<RoaringBitmap>,
parent: Option<Box<dyn Criterion>>,
}
@ -31,6 +32,7 @@ impl<'t> Typo<'t> {
query_tree,
number_typos: 0,
candidates: candidates.map_or_else(Candidates::default, Candidates::Allowed),
bucket_candidates: None,
parent: None,
})
}
@ -45,6 +47,7 @@ impl<'t> Typo<'t> {
query_tree: None,
number_typos: 0,
candidates: Candidates::default(),
bucket_candidates: None,
parent: Some(parent),
})
}
@ -68,10 +71,15 @@ impl<'t> Criterion for Typo<'t> {
candidates.difference_with(&new_candidates);
self.number_typos += 1;
let bucket_candidates = match self.parent {
Some(_) => self.bucket_candidates.take(),
None => Some(new_candidates.clone()),
};
return Ok(Some(CriterionResult {
query_tree: Some(new_query_tree),
candidates: new_candidates,
bucket_candidates: None,
bucket_candidates,
}));
},
(Some(query_tree), Forbidden(candidates)) => {
@ -83,26 +91,33 @@ impl<'t> Criterion for Typo<'t> {
candidates.union_with(&new_candidates);
self.number_typos += 1;
let bucket_candidates = match self.parent {
Some(_) => self.bucket_candidates.take(),
None => Some(new_candidates.clone()),
};
return Ok(Some(CriterionResult {
query_tree: Some(new_query_tree),
candidates: new_candidates,
bucket_candidates: None,
bucket_candidates,
}));
},
(None, Allowed(_)) => {
let candidates = take(&mut self.candidates).into_inner();
return Ok(Some(CriterionResult {
query_tree: None,
candidates: take(&mut self.candidates).into_inner(),
bucket_candidates: None,
candidates: candidates.clone(),
bucket_candidates: Some(candidates),
}));
},
(None, Forbidden(_)) => {
match self.parent.as_mut() {
Some(parent) => {
match parent.next()? {
Some(CriterionResult { query_tree, candidates, .. }) => {
Some(CriterionResult { query_tree, candidates, bucket_candidates }) => {
self.query_tree = query_tree;
self.candidates = Candidates::Allowed(candidates);
self.bucket_candidates = bucket_candidates;
},
None => return Ok(None),
}