mirror of
https://github.com/quickwit-oss/tantivy.git
synced 2025-12-23 02:29:57 +00:00
Renaming .values(DocId) to .values_for_doc(DocId) (#1906)
This commit is contained in:
@@ -36,7 +36,7 @@ impl BytesColumn {
|
||||
}
|
||||
|
||||
pub fn term_ords(&self, row_id: RowId) -> impl Iterator<Item = u64> + '_ {
|
||||
self.term_ord_column.values(row_id)
|
||||
self.term_ord_column.values_for_doc(row_id)
|
||||
}
|
||||
|
||||
/// Returns the column of ordinals
|
||||
|
||||
@@ -64,10 +64,10 @@ impl<T: PartialOrd + Copy + Debug + Send + Sync + 'static> Column<T> {
|
||||
}
|
||||
|
||||
pub fn first(&self, row_id: RowId) -> Option<T> {
|
||||
self.values(row_id).next()
|
||||
self.values_for_doc(row_id).next()
|
||||
}
|
||||
|
||||
pub fn values(&self, row_id: RowId) -> impl Iterator<Item = T> + '_ {
|
||||
pub fn values_for_doc(&self, row_id: RowId) -> impl Iterator<Item = T> + '_ {
|
||||
self.value_row_ids(row_id)
|
||||
.map(|value_row_id: RowId| self.values.get_val(value_row_id))
|
||||
}
|
||||
@@ -97,7 +97,7 @@ impl<T: PartialOrd + Copy + Debug + Send + Sync + 'static> Column<T> {
|
||||
/// This method clears the `output` vector.
|
||||
pub fn fill_vals(&self, row_id: RowId, output: &mut Vec<T>) {
|
||||
output.clear();
|
||||
output.extend(self.values(row_id));
|
||||
output.extend(self.values_for_doc(row_id));
|
||||
}
|
||||
|
||||
pub fn first_or_default_col(self, default_value: T) -> Arc<dyn ColumnValues<T>> {
|
||||
|
||||
@@ -96,7 +96,7 @@ fn compute_term_bitset(column: &BytesColumn, row_bitset: &ReadOnlyBitSet) -> Bit
|
||||
let num_terms = column.dictionary().num_terms();
|
||||
let mut term_bitset = BitSet::with_max_value(num_terms as u32);
|
||||
for row_id in row_bitset.iter() {
|
||||
for term_ord in column.term_ord_column.values(row_id) {
|
||||
for term_ord in column.term_ord_column.values_for_doc(row_id) {
|
||||
term_bitset.insert(term_ord as u32);
|
||||
}
|
||||
}
|
||||
|
||||
@@ -105,7 +105,7 @@ impl SegmentCollector for StatsSegmentCollector {
|
||||
fn collect(&mut self, doc: u32, _score: Score) {
|
||||
// Since we know the values are single value, we could call `first_or_default_col` on the
|
||||
// column and fetch single values.
|
||||
for value in self.fast_field_reader.values(doc) {
|
||||
for value in self.fast_field_reader.values_for_doc(doc) {
|
||||
let value = value as f64;
|
||||
self.stats.count += 1;
|
||||
self.stats.sum += value;
|
||||
|
||||
@@ -253,7 +253,7 @@ impl SegmentAggregationCollector for SegmentHistogramCollector {
|
||||
let get_bucket_pos = |val| (get_bucket_pos_f64(val, interval, offset) as i64);
|
||||
|
||||
for doc in docs {
|
||||
for val in accessor.values(*doc) {
|
||||
for val in accessor.values_for_doc(*doc) {
|
||||
let val = self.f64_from_fastfield_u64(val);
|
||||
|
||||
let bucket_pos = get_bucket_pos(val);
|
||||
|
||||
@@ -225,7 +225,7 @@ impl SegmentAggregationCollector for SegmentRangeCollector {
|
||||
let sub_aggregation_accessor =
|
||||
&agg_with_accessor.buckets.values[self.accessor_idx].sub_aggregation;
|
||||
for doc in docs {
|
||||
for val in accessor.values(*doc) {
|
||||
for val in accessor.values_for_doc(*doc) {
|
||||
let bucket_pos = self.get_bucket_pos(val);
|
||||
|
||||
let bucket = &mut self.buckets[bucket_pos];
|
||||
|
||||
@@ -319,7 +319,7 @@ impl SegmentAggregationCollector for SegmentTermCollector {
|
||||
}
|
||||
} else {
|
||||
for doc in docs {
|
||||
for term_id in accessor.values(*doc) {
|
||||
for term_id in accessor.values_for_doc(*doc) {
|
||||
let entry = self
|
||||
.term_buckets
|
||||
.entries
|
||||
|
||||
@@ -181,7 +181,7 @@ impl SegmentStatsCollector {
|
||||
}
|
||||
} else {
|
||||
for doc in docs {
|
||||
for val in field.values(*doc) {
|
||||
for val in field.values_for_doc(*doc) {
|
||||
let val1 = f64_from_fastfield_u64(val, &self.field_type);
|
||||
self.stats.collect(val1);
|
||||
}
|
||||
@@ -234,7 +234,7 @@ impl SegmentAggregationCollector for SegmentStatsCollector {
|
||||
) -> crate::Result<()> {
|
||||
let field = &agg_with_accessor.metrics.values[self.accessor_idx].accessor;
|
||||
|
||||
for val in field.values(doc) {
|
||||
for val in field.values_for_doc(doc) {
|
||||
let val1 = f64_from_fastfield_u64(val, &self.field_type);
|
||||
self.stats.collect(val1);
|
||||
}
|
||||
|
||||
@@ -200,7 +200,7 @@ impl SegmentCollector for FastFieldSegmentCollector {
|
||||
type Fruit = Vec<u64>;
|
||||
|
||||
fn collect(&mut self, doc: DocId, _score: Score) {
|
||||
self.vals.extend(self.reader.values(doc));
|
||||
self.vals.extend(self.reader.values_for_doc(doc));
|
||||
}
|
||||
|
||||
fn harvest(self) -> Vec<u64> {
|
||||
|
||||
@@ -51,7 +51,7 @@ impl FacetReader {
|
||||
|
||||
/// Return the list of facet ordinals associated with a document.
|
||||
pub fn facet_ords(&self, doc: DocId) -> impl Iterator<Item = u64> + '_ {
|
||||
self.facet_column.ords().values(doc)
|
||||
self.facet_column.ords().values_for_doc(doc)
|
||||
}
|
||||
|
||||
/// Accessor to the facet dictionary.
|
||||
|
||||
@@ -467,7 +467,7 @@ mod tests {
|
||||
|
||||
fn get_vals_for_docs(column: &Column<u64>, docs: Range<u32>) -> Vec<u64> {
|
||||
docs.into_iter()
|
||||
.flat_map(|doc| column.values(doc))
|
||||
.flat_map(|doc| column.values_for_doc(doc))
|
||||
.collect()
|
||||
}
|
||||
|
||||
@@ -515,11 +515,11 @@ mod tests {
|
||||
let segment_reader = searcher.segment_reader(0);
|
||||
let fast_fields = segment_reader.fast_fields();
|
||||
let str_column = fast_fields.str("text").unwrap().unwrap();
|
||||
assert!(str_column.ords().values(0u32).eq([1, 0]),);
|
||||
assert!(str_column.ords().values(1u32).next().is_none());
|
||||
assert!(str_column.ords().values(2u32).eq([0]),);
|
||||
assert!(str_column.ords().values(3u32).eq([0, 1]),);
|
||||
assert!(str_column.ords().values(4u32).eq([2]),);
|
||||
assert!(str_column.ords().values_for_doc(0u32).eq([1, 0]),);
|
||||
assert!(str_column.ords().values_for_doc(1u32).next().is_none());
|
||||
assert!(str_column.ords().values_for_doc(2u32).eq([0]),);
|
||||
assert!(str_column.ords().values_for_doc(3u32).eq([0, 1]),);
|
||||
assert!(str_column.ords().values_for_doc(4u32).eq([2]),);
|
||||
|
||||
let mut str_term = String::default();
|
||||
assert!(str_column.ord_to_str(0, &mut str_term).unwrap());
|
||||
|
||||
@@ -338,20 +338,23 @@ mod tests {
|
||||
let column_float_opt: Option<Column<f64>> = columns[1].1.open().unwrap().into();
|
||||
assert!(column_float_opt
|
||||
.unwrap()
|
||||
.values(0)
|
||||
.values_for_doc(0)
|
||||
.eq([1.02f64].into_iter()));
|
||||
}
|
||||
{
|
||||
assert_eq!(columns[2].0, "nested\u{1}child");
|
||||
let column_nest_child_opt: Option<Column<i64>> = columns[2].1.open().unwrap().into();
|
||||
assert!(column_nest_child_opt.unwrap().values(0).eq([3].into_iter()));
|
||||
assert!(column_nest_child_opt
|
||||
.unwrap()
|
||||
.values_for_doc(0)
|
||||
.eq([3].into_iter()));
|
||||
}
|
||||
{
|
||||
assert_eq!(columns[3].0, "nested\u{1}child2");
|
||||
let column_nest_child2_opt: Option<Column<i64>> = columns[3].1.open().unwrap().into();
|
||||
assert!(column_nest_child2_opt
|
||||
.unwrap()
|
||||
.values(0)
|
||||
.values_for_doc(0)
|
||||
.eq([5].into_iter()));
|
||||
}
|
||||
{
|
||||
@@ -390,7 +393,7 @@ mod tests {
|
||||
assert_eq!(columns[0].0, "obj");
|
||||
let dynamic_column = columns[0].1.open().unwrap();
|
||||
let col: Option<Column<i64>> = dynamic_column.into();
|
||||
let vals: Vec<i64> = col.unwrap().values(0).collect();
|
||||
let vals: Vec<i64> = col.unwrap().values_for_doc(0).collect();
|
||||
assert_eq!(&vals, &[18, 19])
|
||||
}
|
||||
|
||||
|
||||
@@ -1490,7 +1490,7 @@ mod tests {
|
||||
|
||||
let in_order_alive_ids: Vec<u64> = segment_reader
|
||||
.doc_ids_alive()
|
||||
.flat_map(|doc| fast_field_reader.values(doc))
|
||||
.flat_map(|doc| fast_field_reader.values_for_doc(doc))
|
||||
.collect();
|
||||
assert_eq!(&in_order_alive_ids[..], &[9, 8, 7, 6, 5, 4, 1, 0]);
|
||||
Ok(())
|
||||
@@ -1550,7 +1550,7 @@ mod tests {
|
||||
let fast_field_reader = segment_reader.fast_fields().u64("id")?;
|
||||
let in_order_alive_ids: Vec<u64> = segment_reader
|
||||
.doc_ids_alive()
|
||||
.flat_map(|doc| fast_field_reader.values(doc))
|
||||
.flat_map(|doc| fast_field_reader.values_for_doc(doc))
|
||||
.collect();
|
||||
assert_eq!(&in_order_alive_ids[..], &[9, 8, 7, 6, 5, 4, 2, 0]);
|
||||
Ok(())
|
||||
@@ -1795,7 +1795,7 @@ mod tests {
|
||||
let ff_reader = segment_reader.fast_fields().u64("id").unwrap();
|
||||
segment_reader
|
||||
.doc_ids_alive()
|
||||
.flat_map(move |doc| ff_reader.values(doc).collect_vec().into_iter())
|
||||
.flat_map(move |doc| ff_reader.values_for_doc(doc).collect_vec().into_iter())
|
||||
})
|
||||
.collect();
|
||||
|
||||
@@ -1806,7 +1806,7 @@ mod tests {
|
||||
let ff_reader = segment_reader.fast_fields().u64("id").unwrap();
|
||||
segment_reader
|
||||
.doc_ids_alive()
|
||||
.flat_map(move |doc| ff_reader.values(doc).collect_vec().into_iter())
|
||||
.flat_map(move |doc| ff_reader.values_for_doc(doc).collect_vec().into_iter())
|
||||
})
|
||||
.collect();
|
||||
|
||||
@@ -1824,7 +1824,7 @@ mod tests {
|
||||
.unwrap()
|
||||
.unwrap();
|
||||
for doc in segment_reader.doc_ids_alive() {
|
||||
all_ips.extend(ip_reader.values(doc));
|
||||
all_ips.extend(ip_reader.values_for_doc(doc));
|
||||
}
|
||||
num_ips += ip_reader.values.num_vals();
|
||||
}
|
||||
@@ -1883,7 +1883,7 @@ mod tests {
|
||||
for reader in searcher.segment_readers() {
|
||||
if let Some(ff_reader) = reader.fast_fields().column_opt::<Ipv6Addr>("ips").unwrap() {
|
||||
for doc in reader.doc_ids_alive() {
|
||||
ips.extend(ff_reader.values(doc));
|
||||
ips.extend(ff_reader.values_for_doc(doc));
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -1915,7 +1915,7 @@ mod tests {
|
||||
for reader in searcher.segment_readers() {
|
||||
if let Some(ff_reader) = reader.fast_fields().column_opt::<Ipv6Addr>("ips").unwrap() {
|
||||
for doc in reader.doc_ids_alive() {
|
||||
ips.extend(ff_reader.values(doc));
|
||||
ips.extend(ff_reader.values_for_doc(doc));
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -1935,12 +1935,12 @@ mod tests {
|
||||
.unwrap()
|
||||
.unwrap();
|
||||
for doc in segment_reader.doc_ids_alive() {
|
||||
let vals: Vec<u64> = ff_reader.values(doc).collect();
|
||||
let vals: Vec<u64> = ff_reader.values_for_doc(doc).collect();
|
||||
assert_eq!(vals.len(), 2);
|
||||
assert_eq!(vals[0], vals[1]);
|
||||
assert_eq!(id_reader.first(doc), Some(vals[0]));
|
||||
|
||||
let bool_vals: Vec<bool> = bool_ff_reader.values(doc).collect();
|
||||
let bool_vals: Vec<bool> = bool_ff_reader.values_for_doc(doc).collect();
|
||||
assert_eq!(bool_vals.len(), 2);
|
||||
assert_ne!(bool_vals[0], bool_vals[1]);
|
||||
|
||||
@@ -2236,8 +2236,8 @@ mod tests {
|
||||
assert_eq!(val_col.get_cardinality(), Cardinality::Multivalued);
|
||||
assert_eq!(id_col.first(0u32), Some(1u64));
|
||||
assert_eq!(id_col.first(1u32), Some(2u64));
|
||||
assert!(val_col.values(0u32).eq([1u64, 1u64].into_iter()));
|
||||
assert!(val_col.values(1u32).eq([2u64, 2u64].into_iter()));
|
||||
assert!(val_col.values_for_doc(0u32).eq([1u64, 1u64].into_iter()));
|
||||
assert!(val_col.values_for_doc(1u32).eq([2u64, 2u64].into_iter()));
|
||||
}
|
||||
|
||||
#[test]
|
||||
|
||||
@@ -1497,7 +1497,7 @@ mod tests {
|
||||
let mut vals: Vec<u64> = Vec::new();
|
||||
let mut test_vals = move |col: &Column<u64>, doc: DocId, expected: &[u64]| {
|
||||
vals.clear();
|
||||
vals.extend(col.values(doc));
|
||||
vals.extend(col.values_for_doc(doc));
|
||||
assert_eq!(&vals[..], expected);
|
||||
};
|
||||
|
||||
|
||||
Reference in New Issue
Block a user