2023-02-23 13:13:19 +01:00
|
|
|
#[cfg(test)]
|
2023-02-22 15:34:37 +01:00
|
|
|
pub mod detailed;
|
|
|
|
|
|
|
|
use roaring::RoaringBitmap;
|
|
|
|
|
2023-02-23 13:13:19 +01:00
|
|
|
use super::{
|
2023-03-06 08:35:01 +01:00
|
|
|
ranking_rule_graph::{EmptyPathsCache, ProximityGraph, RankingRuleGraph, TypoGraph},
|
2023-03-08 09:53:05 +01:00
|
|
|
small_bitmap::SmallBitmap,
|
2023-02-28 11:49:24 +01:00
|
|
|
RankingRule, RankingRuleQueryTrait,
|
2023-02-23 13:13:19 +01:00
|
|
|
};
|
2023-02-22 15:34:37 +01:00
|
|
|
|
|
|
|
pub struct DefaultSearchLogger;
|
|
|
|
impl<Q: RankingRuleQueryTrait> SearchLogger<Q> for DefaultSearchLogger {
|
2023-03-06 08:35:01 +01:00
|
|
|
fn initial_query(&mut self, _query: &Q) {}
|
|
|
|
|
|
|
|
fn query_for_universe(&mut self, _query: &Q) {}
|
2023-02-22 15:34:37 +01:00
|
|
|
|
2023-02-28 11:49:24 +01:00
|
|
|
fn initial_universe(&mut self, _universe: &RoaringBitmap) {}
|
2023-02-22 15:34:37 +01:00
|
|
|
|
2023-02-28 12:42:29 +01:00
|
|
|
fn ranking_rules(&mut self, _rr: &[&mut dyn RankingRule<Q>]) {}
|
2023-03-02 21:27:42 +01:00
|
|
|
|
2023-02-22 15:34:37 +01:00
|
|
|
fn start_iteration_ranking_rule<'transaction>(
|
|
|
|
&mut self,
|
2023-02-28 11:49:24 +01:00
|
|
|
_ranking_rule_idx: usize,
|
|
|
|
_ranking_rule: &dyn RankingRule<'transaction, Q>,
|
|
|
|
_query: &Q,
|
|
|
|
_universe: &RoaringBitmap,
|
2023-02-22 15:34:37 +01:00
|
|
|
) {
|
|
|
|
}
|
|
|
|
|
|
|
|
fn next_bucket_ranking_rule<'transaction>(
|
|
|
|
&mut self,
|
2023-02-28 11:49:24 +01:00
|
|
|
_ranking_rule_idx: usize,
|
|
|
|
_ranking_rule: &dyn RankingRule<'transaction, Q>,
|
|
|
|
_universe: &RoaringBitmap,
|
2023-03-02 21:27:42 +01:00
|
|
|
_candidates: &RoaringBitmap,
|
2023-02-22 15:34:37 +01:00
|
|
|
) {
|
|
|
|
}
|
2023-02-27 16:14:53 +01:00
|
|
|
fn skip_bucket_ranking_rule<'transaction>(
|
|
|
|
&mut self,
|
2023-02-28 11:49:24 +01:00
|
|
|
_ranking_rule_idx: usize,
|
|
|
|
_ranking_rule: &dyn RankingRule<'transaction, Q>,
|
|
|
|
_candidates: &RoaringBitmap,
|
2023-02-27 16:14:53 +01:00
|
|
|
) {
|
|
|
|
}
|
2023-02-22 15:34:37 +01:00
|
|
|
|
|
|
|
fn end_iteration_ranking_rule<'transaction>(
|
|
|
|
&mut self,
|
2023-02-28 11:49:24 +01:00
|
|
|
_ranking_rule_idx: usize,
|
|
|
|
_ranking_rule: &dyn RankingRule<'transaction, Q>,
|
|
|
|
_universe: &RoaringBitmap,
|
2023-02-22 15:34:37 +01:00
|
|
|
) {
|
|
|
|
}
|
|
|
|
|
2023-02-28 11:49:24 +01:00
|
|
|
fn add_to_results(&mut self, _docids: &[u32]) {}
|
2023-02-22 15:34:37 +01:00
|
|
|
|
2023-02-28 11:49:24 +01:00
|
|
|
fn log_words_state(&mut self, _query_graph: &Q) {}
|
2023-02-23 13:13:19 +01:00
|
|
|
|
|
|
|
fn log_proximity_state(
|
|
|
|
&mut self,
|
2023-02-28 11:49:24 +01:00
|
|
|
_query_graph: &RankingRuleGraph<ProximityGraph>,
|
2023-03-07 14:42:58 +01:00
|
|
|
_paths_map: &[Vec<u16>],
|
2023-02-28 11:49:24 +01:00
|
|
|
_empty_paths_cache: &EmptyPathsCache,
|
2023-03-02 21:27:42 +01:00
|
|
|
_universe: &RoaringBitmap,
|
2023-03-08 09:53:05 +01:00
|
|
|
_distances: Vec<Vec<(u16, SmallBitmap)>>,
|
2023-03-07 14:42:58 +01:00
|
|
|
_cost: u16,
|
2023-02-23 13:13:19 +01:00
|
|
|
) {
|
|
|
|
}
|
2023-02-28 14:19:57 +01:00
|
|
|
|
|
|
|
fn log_typo_state(
|
|
|
|
&mut self,
|
2023-03-02 21:27:42 +01:00
|
|
|
_query_graph: &RankingRuleGraph<TypoGraph>,
|
2023-03-07 14:42:58 +01:00
|
|
|
_paths: &[Vec<u16>],
|
2023-03-02 21:27:42 +01:00
|
|
|
_empty_paths_cache: &EmptyPathsCache,
|
|
|
|
_universe: &RoaringBitmap,
|
2023-03-08 09:53:05 +01:00
|
|
|
_distances: Vec<Vec<(u16, SmallBitmap)>>,
|
2023-03-07 14:42:58 +01:00
|
|
|
_cost: u16,
|
2023-02-28 14:19:57 +01:00
|
|
|
) {
|
|
|
|
}
|
2023-02-22 15:34:37 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
pub trait SearchLogger<Q: RankingRuleQueryTrait> {
|
2023-03-06 08:35:01 +01:00
|
|
|
fn initial_query(&mut self, query: &Q);
|
|
|
|
|
|
|
|
fn query_for_universe(&mut self, query: &Q);
|
|
|
|
|
2023-02-22 15:34:37 +01:00
|
|
|
fn initial_universe(&mut self, universe: &RoaringBitmap);
|
|
|
|
|
2023-02-28 12:42:29 +01:00
|
|
|
fn ranking_rules(&mut self, rr: &[&mut dyn RankingRule<Q>]);
|
2023-02-22 15:34:37 +01:00
|
|
|
|
|
|
|
fn start_iteration_ranking_rule<'transaction>(
|
|
|
|
&mut self,
|
|
|
|
ranking_rule_idx: usize,
|
|
|
|
ranking_rule: &dyn RankingRule<'transaction, Q>,
|
|
|
|
query: &Q,
|
|
|
|
universe: &RoaringBitmap,
|
|
|
|
);
|
|
|
|
fn next_bucket_ranking_rule<'transaction>(
|
|
|
|
&mut self,
|
|
|
|
ranking_rule_idx: usize,
|
|
|
|
ranking_rule: &dyn RankingRule<'transaction, Q>,
|
|
|
|
universe: &RoaringBitmap,
|
2023-03-02 21:27:42 +01:00
|
|
|
candidates: &RoaringBitmap,
|
2023-02-22 15:34:37 +01:00
|
|
|
);
|
2023-02-27 16:14:53 +01:00
|
|
|
fn skip_bucket_ranking_rule<'transaction>(
|
|
|
|
&mut self,
|
|
|
|
ranking_rule_idx: usize,
|
|
|
|
ranking_rule: &dyn RankingRule<'transaction, Q>,
|
|
|
|
candidates: &RoaringBitmap,
|
|
|
|
);
|
2023-02-22 15:34:37 +01:00
|
|
|
fn end_iteration_ranking_rule<'transaction>(
|
|
|
|
&mut self,
|
|
|
|
ranking_rule_idx: usize,
|
|
|
|
ranking_rule: &dyn RankingRule<'transaction, Q>,
|
|
|
|
universe: &RoaringBitmap,
|
|
|
|
);
|
2023-02-27 16:14:53 +01:00
|
|
|
fn add_to_results(&mut self, docids: &[u32]);
|
2023-02-22 15:34:37 +01:00
|
|
|
|
|
|
|
fn log_words_state(&mut self, query_graph: &Q);
|
2023-02-23 13:13:19 +01:00
|
|
|
|
|
|
|
fn log_proximity_state(
|
|
|
|
&mut self,
|
|
|
|
query_graph: &RankingRuleGraph<ProximityGraph>,
|
2023-03-07 14:42:58 +01:00
|
|
|
paths: &[Vec<u16>],
|
2023-02-27 15:04:40 +01:00
|
|
|
empty_paths_cache: &EmptyPathsCache,
|
2023-03-02 21:27:42 +01:00
|
|
|
universe: &RoaringBitmap,
|
2023-03-08 09:53:05 +01:00
|
|
|
distances: Vec<Vec<(u16, SmallBitmap)>>,
|
2023-03-07 14:42:58 +01:00
|
|
|
cost: u16,
|
2023-02-23 13:13:19 +01:00
|
|
|
);
|
2023-02-28 14:19:57 +01:00
|
|
|
|
|
|
|
fn log_typo_state(
|
|
|
|
&mut self,
|
|
|
|
query_graph: &RankingRuleGraph<TypoGraph>,
|
2023-03-07 14:42:58 +01:00
|
|
|
paths: &[Vec<u16>],
|
2023-02-28 14:19:57 +01:00
|
|
|
empty_paths_cache: &EmptyPathsCache,
|
2023-03-02 21:27:42 +01:00
|
|
|
universe: &RoaringBitmap,
|
2023-03-08 09:53:05 +01:00
|
|
|
distances: Vec<Vec<(u16, SmallBitmap)>>,
|
2023-03-07 14:42:58 +01:00
|
|
|
cost: u16,
|
2023-02-28 14:19:57 +01:00
|
|
|
);
|
2023-02-22 15:34:37 +01:00
|
|
|
}
|