mirror of
https://github.com/quickwit-oss/tantivy.git
synced 2026-05-23 11:40:40 +00:00
delegating the search logic to a query trait.
This commit is contained in:
5
src/query/mod.rs
Normal file
5
src/query/mod.rs
Normal file
@@ -0,0 +1,5 @@
|
||||
mod query;
|
||||
mod multi_term_query;
|
||||
|
||||
pub use self::query::Query;
|
||||
pub use self::multi_term_query::MultiTermQuery;
|
||||
88
src/query/multi_term_query.rs
Normal file
88
src/query/multi_term_query.rs
Normal file
@@ -0,0 +1,88 @@
|
||||
use schema::Term;
|
||||
use query::Query;
|
||||
use common::TimerTree;
|
||||
use common::OpenTimer;
|
||||
use std::io;
|
||||
use core::searcher::Searcher;
|
||||
use collector::Collector;
|
||||
use core::searcher::SegmentLocalId;
|
||||
use core::reader::SegmentReader;
|
||||
use postings::Postings;
|
||||
use postings::SegmentPostings;
|
||||
use postings::intersection;
|
||||
|
||||
pub struct MultiTermQuery {
|
||||
terms: Vec<Term>,
|
||||
}
|
||||
|
||||
impl Query for MultiTermQuery {
|
||||
|
||||
|
||||
fn search<C: Collector>(&self, searcher: &Searcher, collector: &mut C) -> io::Result<TimerTree> {
|
||||
let mut timer_tree = TimerTree::new();
|
||||
{
|
||||
let mut search_timer = timer_tree.open("search");
|
||||
for (segment_ord, segment_reader) in searcher.segments().iter().enumerate() {
|
||||
let mut segment_search_timer = search_timer.open("segment_search");
|
||||
{
|
||||
let _ = segment_search_timer.open("set_segment");
|
||||
try!(collector.set_segment(segment_ord as SegmentLocalId, &segment_reader));
|
||||
}
|
||||
let mut postings = self.search_segment(segment_reader, segment_search_timer.open("get_postings"));
|
||||
{
|
||||
let _collection_timer = segment_search_timer.open("collection");
|
||||
while postings.next() {
|
||||
collector.collect(postings.doc());
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
Ok(timer_tree)
|
||||
}
|
||||
}
|
||||
|
||||
impl MultiTermQuery {
|
||||
pub fn new(terms: Vec<Term>) -> MultiTermQuery {
|
||||
MultiTermQuery {
|
||||
terms: terms,
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
fn search_segment<'a, 'b>(&'b self, reader: &'b SegmentReader, mut timer: OpenTimer<'a>) -> Box<Postings + 'b> {
|
||||
if self.terms.len() == 1 {
|
||||
match reader.get_term(&self.terms[0]) {
|
||||
Some(term_info) => {
|
||||
let postings: SegmentPostings<'b> = reader.read_postings(&term_info);
|
||||
Box::new(postings)
|
||||
},
|
||||
None => {
|
||||
Box::new(SegmentPostings::empty())
|
||||
},
|
||||
}
|
||||
} else {
|
||||
let mut segment_postings: Vec<SegmentPostings> = Vec::new();
|
||||
{
|
||||
let mut decode_timer = timer.open("decode_all");
|
||||
for term in self.terms.iter() {
|
||||
match reader.get_term(term) {
|
||||
Some(term_info) => {
|
||||
let _decode_one_timer = decode_timer.open("decode_one");
|
||||
let segment_posting = reader.read_postings(&term_info);
|
||||
segment_postings.push(segment_posting);
|
||||
}
|
||||
None => {
|
||||
// currently this is a strict intersection.
|
||||
return Box::new(SegmentPostings::empty());
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
Box::new(intersection(segment_postings))
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
8
src/query/query.rs
Normal file
8
src/query/query.rs
Normal file
@@ -0,0 +1,8 @@
|
||||
use std::io;
|
||||
use collector::Collector;
|
||||
use core::searcher::Searcher;
|
||||
use common::TimerTree;
|
||||
|
||||
pub trait Query {
|
||||
fn search<C: Collector>(&self, searcher: &Searcher, collector: &mut C) -> io::Result<TimerTree>;
|
||||
}
|
||||
Reference in New Issue
Block a user