snowykami/meilisearch
1
0
Fork 0
mirror of https://github.com/meilisearch/meilisearch.git synced 2025-07-31 10:50:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

Introduce plane_sweep function in proximity criterion

Browse Source
This commit is contained in:
many
2021-03-03 15:41:09 +01:00
committed by Kerollmops
parent 636a9df177
commit ae47bb3594
2 changed files with 188 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
milli/src/search/criteria/mod.rs
View File

@ -5,7 +5,7 @@ use anyhow::bail;
use roaring::RoaringBitmap;
use crate::search::word_derivations;
use crate::Index;
use crate::{DocumentId, Index};
use super::query_tree::{Operation, Query, QueryKind};
use self::typo::Typo;
@ -66,6 +66,7 @@ pub trait Context {
fn word_prefix_pair_proximity_docids(&self, left: &str, right: &str, proximity: u8) -> heed::Result<Option<RoaringBitmap>>;
fn words_fst<'t>(&self) -> &'t fst::Set<Cow<[u8]>>;
fn in_prefix_cache(&self, word: &str) -> bool;
fn docid_word_positions(&self, docid: DocumentId, word: &str) -> heed::Result<Option<RoaringBitmap>>;
}
pub struct CriteriaBuilder<'t> {
rtxn: &'t heed::RoTxn<'t>,
@ -104,6 +105,11 @@ impl<'a> Context for CriteriaBuilder<'a> {
fn in_prefix_cache(&self, word: &str) -> bool {
self.words_prefixes_fst.contains(word)
}
fn docid_word_positions(&self, docid: DocumentId, word: &str) -> heed::Result<Option<RoaringBitmap>> {
let key = (docid, word);
self.index.docid_word_positions.get(self.rtxn, &key)
}
}
impl<'t> CriteriaBuilder<'t> {
@ -368,6 +374,10 @@ pub mod test {
fn in_prefix_cache(&self, word: &str) -> bool {
self.word_prefix_docids.contains_key(&word.to_string())
}
fn docid_word_positions(&self, _docid: DocumentId, _word: &str) -> heed::Result<Option<RoaringBitmap>> {
todo!()
}
}
impl<'a> Default for TestContext<'a> {