snowykami/meilisearch
1
0
Fork 0
mirror of https://github.com/meilisearch/meilisearch.git synced 2025-01-31 15:31:53 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Restrict the number of facet search results to 1000

Browse Source
This commit is contained in:
Kerollmops 2023-04-13 18:16:08 +02:00 committed by Louis Dureuil
parent c34de05106
commit addb21f110
No known key found for this signature in database
1 changed files with 13 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
milli/src/search/mod.rs
View File

@ -21,6 +21,9 @@ static LEVDIST0: Lazy<LevBuilder> = Lazy::new(|| LevBuilder::new(0, true));
static LEVDIST1: Lazy<LevBuilder> = Lazy::new(|| LevBuilder::new(1, true)); static LEVDIST1: Lazy<LevBuilder> = Lazy::new(|| LevBuilder::new(1, true));
static LEVDIST2: Lazy<LevBuilder> = Lazy::new(|| LevBuilder::new(2, true)); static LEVDIST2: Lazy<LevBuilder> = Lazy::new(|| LevBuilder::new(2, true));
/// The maximum number of facets returned by the facet search route.
const MAX_NUMBER_OF_FACETS: usize = 1000;
pub mod facet; pub mod facet;
mod fst_utils; mod fst_utils;
pub mod new; pub mod new;
@ -291,6 +294,7 @@ impl<'a> SearchForFacetValue<'a> {
let mut stream = fst.search(automaton).into_stream(); let mut stream = fst.search(automaton).into_stream();
let mut result = vec![]; let mut result = vec![];
let mut length = 0;
while let Some(facet_value) = stream.next() { while let Some(facet_value) = stream.next() {
let value = std::str::from_utf8(facet_value)?; let value = std::str::from_utf8(facet_value)?;
let key = FacetGroupKey { field_id, level: 0, left_bound: value }; let key = FacetGroupKey { field_id, level: 0, left_bound: value };
@ -301,6 +305,10 @@ impl<'a> SearchForFacetValue<'a> {
let count = search_candidates.intersection_len(&docids); let count = search_candidates.intersection_len(&docids);
if count != 0 { if count != 0 {
result.push(FacetSearchResult { value: value.to_string(), count }); result.push(FacetSearchResult { value: value.to_string(), count });
length += 1;
}
if length >= MAX_NUMBER_OF_FACETS {
break;
} }
} }
@ -309,6 +317,7 @@ impl<'a> SearchForFacetValue<'a> {
None => { None => {
let mut stream = fst.stream(); let mut stream = fst.stream();
let mut result = vec![]; let mut result = vec![];
let mut length = 0;
while let Some(facet_value) = stream.next() { while let Some(facet_value) = stream.next() {
let value = std::str::from_utf8(facet_value)?; let value = std::str::from_utf8(facet_value)?;
let key = FacetGroupKey { field_id, level: 0, left_bound: value }; let key = FacetGroupKey { field_id, level: 0, left_bound: value };
@ -319,6 +328,10 @@ impl<'a> SearchForFacetValue<'a> {
let count = search_candidates.intersection_len(&docids); let count = search_candidates.intersection_len(&docids);
if count != 0 { if count != 0 {
result.push(FacetSearchResult { value: value.to_string(), count }); result.push(FacetSearchResult { value: value.to_string(), count });
length += 1;
}
if length >= MAX_NUMBER_OF_FACETS {
break;
} }
} }