Modify the compute_query_term_subset_docids function to accept the universe

This commit is contained in:
Clément Renault 2024-06-21 11:43:04 +02:00
parent 2099b4f0dd
commit 50a7393c55
No known key found for this signature in database
GPG key ID: F250A4C4E3AE5F5F
6 changed files with 11 additions and 14 deletions

View file

@ -31,6 +31,7 @@ impl<'ctx> SearchContext<'ctx> {
}
pub fn compute_query_term_subset_docids(
ctx: &mut SearchContext<'_>,
universe: Option<&RoaringBitmap>,
term: &QueryTermSubset,
) -> Result<RoaringBitmap> {
let mut docids = RoaringBitmap::new();
@ -49,7 +50,10 @@ pub fn compute_query_term_subset_docids(
}
}
Ok(docids)
match universe {
Some(universe) => Ok(docids & universe),
None => Ok(docids),
}
}
pub fn compute_query_term_subset_docids_within_field_id(
@ -147,10 +151,7 @@ pub fn compute_query_graph_docids(
term_subset,
positions: _,
term_ids: _,
}) => {
let node_docids = compute_query_term_subset_docids(ctx, term_subset)?;
predecessors_docids & node_docids
}
}) => compute_query_term_subset_docids(ctx, Some(&predecessors_docids), term_subset)?,
QueryNodeData::Deleted => {
panic!()
}