summaryrefslogtreecommitdiffstats
path: root/src/collector
diff options
context:
space:
mode:
Diffstat (limited to 'src/collector')
-rw-r--r--src/collector/facet_collector.rs16
-rw-r--r--src/collector/mod.rs11
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