snowykami/meilisearch
1
0
Fork 0
mirror of https://github.com/meilisearch/meilisearch.git synced 2024-11-23 10:37:41 +08:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

Speed up the facets stats infos subcommand

Browse Source
This commit is contained in:
Clément Renault 2020-11-17 21:18:34 +01:00
parent 7a6e6eb5e2
commit 45e0feab4e
No known key found for this signature in database
GPG Key ID: 92ADA4E935E71FA4
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/subcommand/infos.rs
View File

@ -407,6 +407,7 @@ fn facet_values_docids(index: &Index, rtxn: &heed::RoTxn, debug: bool, field_nam
} }
fn facet_stats(index: &Index, rtxn: &heed::RoTxn, field_name: String) -> anyhow::Result<()> { fn facet_stats(index: &Index, rtxn: &heed::RoTxn, field_name: String) -> anyhow::Result<()> {
use heed::types::ByteSlice;
use crate::facet::FacetType; use crate::facet::FacetType;
use crate::heed_codec::facet::{ use crate::heed_codec::facet::{
FacetValueStringCodec, FacetLevelValueF64Codec, FacetLevelValueI64Codec, FacetValueStringCodec, FacetLevelValueF64Codec, FacetLevelValueI64Codec,
@ -423,17 +424,17 @@ fn facet_stats(index: &Index, rtxn: &heed::RoTxn, field_name: String) -> anyhow:
let iter = index.facet_field_id_value_docids.prefix_iter(&rtxn, &[field_id])?; let iter = index.facet_field_id_value_docids.prefix_iter(&rtxn, &[field_id])?;
let iter = match field_type { let iter = match field_type {
FacetType::String => { FacetType::String => {
let iter = iter.remap_key_type::<FacetValueStringCodec>() let iter = iter.remap_types::<FacetValueStringCodec, ByteSlice>()
.map(|r| r.map(|_| 0u8)); .map(|r| r.map(|_| 0u8));
Box::new(iter) as Box<dyn Iterator<Item=_>> Box::new(iter) as Box<dyn Iterator<Item=_>>
}, },
FacetType::Float => { FacetType::Float => {
let iter = iter.remap_key_type::<FacetLevelValueF64Codec>() let iter = iter.remap_types::<FacetLevelValueF64Codec, ByteSlice>()
.map(|r| r.map(|((_, level, _, _), _)| level)); .map(|r| r.map(|((_, level, _, _), _)| level));
Box::new(iter) Box::new(iter)
}, },
FacetType::Integer => { FacetType::Integer => {
let iter = iter.remap_key_type::<FacetLevelValueI64Codec>() let iter = iter.remap_types::<FacetLevelValueI64Codec, ByteSlice>()
.map(|r| r.map(|((_, level, _, _), _)| level)); .map(|r| r.map(|((_, level, _, _), _)| level));
Box::new(iter) Box::new(iter)
}, },