snowykami/meilisearch
1
0
Fork 0
mirror of https://github.com/meilisearch/meilisearch.git synced 2024-11-26 03:55:07 +08:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Merge #944

Browse Source 
944: Fix facet count r=MarinPostma a=MarinPostma

fix bug reported in: https://github.com/meilisearch/MeiliSearch/issues/929#issuecomment-683683728

Co-authored-by: mpostma <postma.marin@protonmail.com>
This commit is contained in:
bors[bot] 2020-09-01 08:43:47 +00:00 committed by GitHub
commit a07c3743f0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 46 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
meilisearch-core/src/bucket_sort.rs
View File

@ -100,6 +100,12 @@ where
debug!("found {} documents", docids.len());
debug!("number of postings {:?}", queries.len());
if let Some(f) = facet_count_docids {
// hardcoded value, until approximation optimization
result.exhaustive_facets_count = Some(true);
result.facets = Some(facet_count(f, &docids));
}
if let Some(facets_docids) = facets_docids {
let intersection = sdset::duo::OpBuilder::new(docids.as_ref(), facets_docids.as_set())
.intersection()
@ -107,12 +113,6 @@ where
docids = Cow::Owned(intersection);
}
if let Some(f) = facet_count_docids {
// hardcoded value, until approximation optimization
result.exhaustive_facets_count = Some(true);
result.facets = Some(facet_count(f, &docids));
}
let before = Instant::now();
mk_arena!(arena);
let mut bare_matches = cleanup_bare_matches(&mut arena, &docids, queries);
@ -243,6 +243,12 @@ where
debug!("found {} documents", docids.len());
debug!("number of postings {:?}", queries.len());
if let Some(f) = facet_count_docids {
// hardcoded value, until approximation optimization
result.exhaustive_facets_count = Some(true);
result.facets = Some(facet_count(f, &docids));
}
if let Some(facets_docids) = facets_docids {
let intersection = OpBuilder::new(docids.as_ref(), facets_docids.as_set())
.intersection()
@ -250,12 +256,6 @@ where
docids = Cow::Owned(intersection);
}
if let Some(f) = facet_count_docids {
// hardcoded value, until approximation optimization
result.exhaustive_facets_count = Some(true);
result.facets = Some(facet_count(f, &docids));
}
let before = Instant::now();
mk_arena!(arena);
let mut bare_matches = cleanup_bare_matches(&mut arena, &docids, queries);

35
meilisearch-http/tests/search.rs
View File

@ -1466,7 +1466,6 @@ async fn test_facet_count() {
server.update_all_settings(body).await;
// same as before, but now facets are set:
test_post_get_search!(server, query, |response, _status_code|{
println!("{}", response);
assert!(response.get("exhaustiveFacetsCount").is_some());
assert_eq!(response.get("facetsDistribution").unwrap().as_object().unwrap().values().count(), 1);
// assert that case is preserved
@ -1696,3 +1695,37 @@ async fn update_documents_with_facet_distribution() {
let (response2, _) = server.search_post(search).await;
assert_json_eq!(expected_facet_distribution, response2["facetsDistribution"].clone());
}
#[actix_rt::test]
async fn test_facet_count_with_facet_filter() {
let mut server = common::Server::test_server().await;
let body = json!({
"attributesForFaceting": ["gender"]
});
server.update_all_settings(body).await;
let query = json!({
"q": "a",
"facetsDistribution": ["gender"],
"facetFilters": ["gender:male"],
});
test_post_get_search!(server, query, |response, _status_code|{
assert!(response.get("exhaustiveFacetsCount").is_some());
let facets_distribution = response.get("facetsDistribution").unwrap().as_object();
assert_eq!(facets_distribution.unwrap()["gender"]["male"], 37);
assert_eq!(facets_distribution.unwrap()["gender"]["female"], 39);
});
// facet distribution should remain the same when facet filter changes
let query = json!({
"q": "a",
"facetsDistribution": ["gender"],
"facetFilters": [["gender:male", "gender:female"]],
});
test_post_get_search!(server, query, |response, _status_code|{
assert!(response.get("exhaustiveFacetsCount").is_some());
let facets_distribution = response.get("facetsDistribution").unwrap().as_object();
assert_eq!(facets_distribution.unwrap()["gender"]["male"], 37);
assert_eq!(facets_distribution.unwrap()["gender"]["female"], 39);
});
}