snowykami/meilisearch
1
0
Fork 0
mirror of https://github.com/meilisearch/meilisearch.git synced 2024-11-22 18:17:39 +08:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

fix: Allow documents to bypass attribute distinction

Browse Source 
In situations where the attribute is not present.
This commit is contained in:
Clément Renault 2018-10-18 15:08:04 +02:00
parent 98899d3ea0
commit 86cb01b243
1 changed files with 17 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
src/rank/ranked_stream.rs
View File

@ -100,7 +100,7 @@ where C: Criterion
}
pub fn retrieve_distinct_documents<K>(mut self, range: Range<usize>) -> Vec<Document>
where F: Fn(&DocumentId) -> K,
where F: Fn(&DocumentId) -> Option<K>,
K: Hash + Eq,
{
let mut documents = self.retrieve_all_documents();
@ -122,8 +122,10 @@ where C: Criterion
let mut seen = DistinctMap::new(limit);
for document in documents {
let key = distinct(&document.id);
let accepted = seen.digest(key);
let accepted = match distinct(&document.id) {
Some(key) => seen.digest(key),
None => seen.accept_without_key(),
};
if accepted {
if seen.len() == range.end { break }
@ -154,7 +156,18 @@ impl<K: Hash + Eq> DistinctMap<K> {
pub fn digest(&mut self, key: K) -> bool {
let seen = self.inner.entry(key).or_insert(0);
if *seen < self.limit { *seen += 1; self.len += 1; true } else { false }
if *seen < self.limit {
*seen += 1;
self.len += 1;
true
} else {
false
}
}
pub fn accept_without_key(&mut self) -> bool {
self.len += 1;
true
}
pub fn len(&self) -> usize {