diff options
Diffstat (limited to 'src/collector')
-rw-r--r-- | src/collector/facet_collector.rs | 16 | ||||
-rw-r--r-- | src/collector/mod.rs | 11 |
2 files changed, 20 insertions, 7 deletions
diff --git a/src/collector/facet_collector.rs b/src/collector/facet_collector.rs index bc0c46c..494b56d 100644 --- a/src/collector/facet_collector.rs +++ b/src/collector/facet_collector.rs @@ -13,6 +13,7 @@ use termdict::TermStreamerBuilder; use std::collections::BTreeSet; use termdict::TermMerger; use docset::SkipResult; +use std::collections::btree_map; use std::{usize, u64}; use std::iter::Peekable; @@ -433,9 +434,7 @@ pub struct FacetCounts { } -use std::collections::btree_map; - -struct FacetChildIterator<'a> { +pub struct FacetChildIterator<'a> { underlying: btree_map::Range<'a, Facet, u64>, } @@ -501,7 +500,6 @@ impl FacetCounts { #[cfg(test)] mod tests { - use test::Bencher; use core::Index; use schema::{Document, Facet, SchemaBuilder}; use query::AllQuery; @@ -622,6 +620,14 @@ mod tests { } } +} + + +#[cfg(all(test, feature="unstable"))] +mod bench { + + use test::Bencher; + #[bench] fn bench_facet_collector(b: &mut Bencher) { let mut schema_builder = SchemaBuilder::new(); @@ -652,4 +658,4 @@ mod tests { searcher.search(&AllQuery, &mut facet_collector).unwrap(); }); } -} +}
\ No newline at end of file diff --git a/src/collector/mod.rs b/src/collector/mod.rs index f905e52..660f878 100644 --- a/src/collector/mod.rs +++ b/src/collector/mod.rs @@ -89,7 +89,6 @@ impl<'a, C: Collector> Collector for &'a mut C { pub mod tests { use super::*; - use test::Bencher; use DocId; use Score; use core::SegmentReader; @@ -186,6 +185,14 @@ pub mod tests { } } +} + + +#[cfg(all(test, feature="unstable"))] +mod bench { + + use test::Bencher; + #[bench] fn build_collector(b: &mut Bencher) { b.iter(|| { @@ -197,4 +204,4 @@ pub mod tests { count_collector.count() }); } -} +}
\ No newline at end of file |