mirror of
https://github.com/quickwit-oss/tantivy.git
synced 2026-01-04 08:12:54 +00:00
Compare commits
21 Commits
columnread
...
sparse_cod
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
4c7437f2e0 | ||
|
|
cc82f94c72 | ||
|
|
f842da758c | ||
|
|
97ccd6d712 | ||
|
|
cb252a42af | ||
|
|
d9609dd6b6 | ||
|
|
f03667d967 | ||
|
|
10f10a322f | ||
|
|
f757471077 | ||
|
|
21e0adefda | ||
|
|
ea8e6d7b1d | ||
|
|
dac7da780e | ||
|
|
20c87903b2 | ||
|
|
f9c3947803 | ||
|
|
e9a384bb15 | ||
|
|
d231671fe2 | ||
|
|
fa3d786a2f | ||
|
|
75aafeeb9b | ||
|
|
6f066c7f65 | ||
|
|
22e56aaee3 | ||
|
|
d641979127 |
1
.gitattributes
vendored
1
.gitattributes
vendored
@@ -1 +0,0 @@
|
|||||||
cpp/* linguist-vendored
|
|
||||||
1
.gitignore
vendored
1
.gitignore
vendored
@@ -9,7 +9,6 @@ target/release
|
|||||||
Cargo.lock
|
Cargo.lock
|
||||||
benchmark
|
benchmark
|
||||||
.DS_Store
|
.DS_Store
|
||||||
cpp/simdcomp/bitpackingbenchmark
|
|
||||||
*.bk
|
*.bk
|
||||||
.idea
|
.idea
|
||||||
trace.dat
|
trace.dat
|
||||||
|
|||||||
@@ -259,11 +259,7 @@ impl BitSet {
|
|||||||
// we do not check saturated els.
|
// we do not check saturated els.
|
||||||
let higher = el / 64u32;
|
let higher = el / 64u32;
|
||||||
let lower = el % 64u32;
|
let lower = el % 64u32;
|
||||||
self.len += if self.tinysets[higher as usize].insert_mut(lower) {
|
self.len += u64::from(self.tinysets[higher as usize].insert_mut(lower));
|
||||||
1
|
|
||||||
} else {
|
|
||||||
0
|
|
||||||
};
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Inserts an element in the `BitSet`
|
/// Inserts an element in the `BitSet`
|
||||||
@@ -272,11 +268,7 @@ impl BitSet {
|
|||||||
// we do not check saturated els.
|
// we do not check saturated els.
|
||||||
let higher = el / 64u32;
|
let higher = el / 64u32;
|
||||||
let lower = el % 64u32;
|
let lower = el % 64u32;
|
||||||
self.len -= if self.tinysets[higher as usize].remove_mut(lower) {
|
self.len -= u64::from(self.tinysets[higher as usize].remove_mut(lower));
|
||||||
1
|
|
||||||
} else {
|
|
||||||
0
|
|
||||||
};
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Returns true iff the elements is in the `BitSet`.
|
/// Returns true iff the elements is in the `BitSet`.
|
||||||
@@ -285,7 +277,7 @@ impl BitSet {
|
|||||||
self.tinyset(el / 64u32).contains(el % 64)
|
self.tinyset(el / 64u32).contains(el % 64)
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Returns the first non-empty `TinySet` associated to a bucket lower
|
/// Returns the first non-empty `TinySet` associated with a bucket lower
|
||||||
/// or greater than bucket.
|
/// or greater than bucket.
|
||||||
///
|
///
|
||||||
/// Reminder: the tiny set with the bucket `bucket`, represents the
|
/// Reminder: the tiny set with the bucket `bucket`, represents the
|
||||||
|
|||||||
@@ -161,8 +161,7 @@ impl FixedSize for u8 {
|
|||||||
|
|
||||||
impl BinarySerializable for bool {
|
impl BinarySerializable for bool {
|
||||||
fn serialize<W: Write>(&self, writer: &mut W) -> io::Result<()> {
|
fn serialize<W: Write>(&self, writer: &mut W) -> io::Result<()> {
|
||||||
let val = if *self { 1 } else { 0 };
|
writer.write_u8(u8::from(*self))
|
||||||
writer.write_u8(val)
|
|
||||||
}
|
}
|
||||||
fn deserialize<R: Read>(reader: &mut R) -> io::Result<bool> {
|
fn deserialize<R: Read>(reader: &mut R) -> io::Result<bool> {
|
||||||
let val = reader.read_u8()?;
|
let val = reader.read_u8()?;
|
||||||
|
|||||||
@@ -50,7 +50,7 @@ to get tantivy to fit your use case:
|
|||||||
|
|
||||||
*Example 1* You could for instance use hadoop to build a very large search index in a timely manner, copy all of the resulting segment files in the same directory and edit the `meta.json` to get a functional index.[^2]
|
*Example 1* You could for instance use hadoop to build a very large search index in a timely manner, copy all of the resulting segment files in the same directory and edit the `meta.json` to get a functional index.[^2]
|
||||||
|
|
||||||
*Example 2* You could also disable your merge policy and enforce daily segments. Removing data after one week can then be done very efficiently by just editing the `meta.json` and deleting the files associated to segment `D-7`.
|
*Example 2* You could also disable your merge policy and enforce daily segments. Removing data after one week can then be done very efficiently by just editing the `meta.json` and deleting the files associated with segment `D-7`.
|
||||||
|
|
||||||
## Merging
|
## Merging
|
||||||
|
|
||||||
|
|||||||
@@ -113,7 +113,7 @@ fn main() -> tantivy::Result<()> {
|
|||||||
// on its id.
|
// on its id.
|
||||||
//
|
//
|
||||||
// Note that `tantivy` does nothing to enforce the idea that
|
// Note that `tantivy` does nothing to enforce the idea that
|
||||||
// there is only one document associated to this id.
|
// there is only one document associated with this id.
|
||||||
//
|
//
|
||||||
// Also you might have noticed that we apply the delete before
|
// Also you might have noticed that we apply the delete before
|
||||||
// having committed. This does not matter really...
|
// having committed. This does not matter really...
|
||||||
|
|||||||
@@ -44,7 +44,7 @@ fn main() -> tantivy::Result<()> {
|
|||||||
// A segment contains different data structure.
|
// A segment contains different data structure.
|
||||||
// Inverted index stands for the combination of
|
// Inverted index stands for the combination of
|
||||||
// - the term dictionary
|
// - the term dictionary
|
||||||
// - the inverted lists associated to each terms and their positions
|
// - the inverted lists associated with each terms and their positions
|
||||||
let inverted_index = segment_reader.inverted_index(title)?;
|
let inverted_index = segment_reader.inverted_index(title)?;
|
||||||
|
|
||||||
// A `Term` is a text token associated with a field.
|
// A `Term` is a text token associated with a field.
|
||||||
@@ -105,7 +105,7 @@ fn main() -> tantivy::Result<()> {
|
|||||||
// A segment contains different data structure.
|
// A segment contains different data structure.
|
||||||
// Inverted index stands for the combination of
|
// Inverted index stands for the combination of
|
||||||
// - the term dictionary
|
// - the term dictionary
|
||||||
// - the inverted lists associated to each terms and their positions
|
// - the inverted lists associated with each terms and their positions
|
||||||
let inverted_index = segment_reader.inverted_index(title)?;
|
let inverted_index = segment_reader.inverted_index(title)?;
|
||||||
|
|
||||||
// This segment posting object is like a cursor over the documents matching the term.
|
// This segment posting object is like a cursor over the documents matching the term.
|
||||||
|
|||||||
@@ -18,6 +18,7 @@ fastdivide = "0.4"
|
|||||||
log = "0.4"
|
log = "0.4"
|
||||||
itertools = { version = "0.10.3" }
|
itertools = { version = "0.10.3" }
|
||||||
measure_time = { version="0.8.2", optional=true}
|
measure_time = { version="0.8.2", optional=true}
|
||||||
|
roaring = "0.10.1"
|
||||||
|
|
||||||
[dev-dependencies]
|
[dev-dependencies]
|
||||||
more-asserts = "0.3.0"
|
more-asserts = "0.3.0"
|
||||||
|
|||||||
@@ -44,6 +44,15 @@ mod tests {
|
|||||||
open(OwnedBytes::new(buffer)).unwrap()
|
open(OwnedBytes::new(buffer)).unwrap()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub fn serialize_and_load_dense<T: MonotonicallyMappableToU64 + Ord + Default>(
|
||||||
|
column: &[T],
|
||||||
|
fill_ratio: u32,
|
||||||
|
) -> Arc<dyn Column<T>> {
|
||||||
|
let mut buffer = Vec::new();
|
||||||
|
serialize(VecColumn::from(&column), &mut buffer, &ALL_CODEC_TYPES).unwrap();
|
||||||
|
open_dense(OwnedBytes::new(buffer), fill_ratio).unwrap()
|
||||||
|
}
|
||||||
|
|
||||||
#[bench]
|
#[bench]
|
||||||
fn bench_intfastfield_jumpy_veclookup(b: &mut Bencher) {
|
fn bench_intfastfield_jumpy_veclookup(b: &mut Bencher) {
|
||||||
let permutation = generate_permutation();
|
let permutation = generate_permutation();
|
||||||
@@ -183,6 +192,67 @@ mod tests {
|
|||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[bench]
|
||||||
|
fn bench_intfastfield_stride7_fflookup_sparse_roaring(b: &mut Bencher) {
|
||||||
|
let permutation = generate_permutation();
|
||||||
|
let n = permutation.len();
|
||||||
|
let column: Arc<dyn Column<u64>> = serialize_and_load(&permutation);
|
||||||
|
let column = SparseCodecRoaringBitmap::with_full(column);
|
||||||
|
b.iter(|| {
|
||||||
|
let mut a = 0u64;
|
||||||
|
for i in (0..n / 7).map(|val| val * 7) {
|
||||||
|
a += column.get_val(i as u64);
|
||||||
|
}
|
||||||
|
a
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
#[bench]
|
||||||
|
fn bench_intfastfield_stride7_fflookup_dense_bitmap_with_offset(b: &mut Bencher) {
|
||||||
|
let permutation = generate_permutation();
|
||||||
|
let n = permutation.len();
|
||||||
|
let column: Arc<dyn Column<u64>> = serialize_and_load_dense(&permutation, 1000);
|
||||||
|
b.iter(|| {
|
||||||
|
let mut a = 0u64;
|
||||||
|
for i in (0..n / 7).map(|val| val * 7) {
|
||||||
|
a += column.get_val(i as u64);
|
||||||
|
}
|
||||||
|
a
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
#[bench]
|
||||||
|
fn bench_intfastfield_stride7_fflookup_dense_bitmap_with_offset_70percent_dense(
|
||||||
|
b: &mut Bencher,
|
||||||
|
) {
|
||||||
|
let permutation = generate_permutation();
|
||||||
|
let n = permutation.len();
|
||||||
|
let column: Arc<dyn Column<u64>> = serialize_and_load_dense(&permutation, 700);
|
||||||
|
b.iter(|| {
|
||||||
|
let mut a = 0u64;
|
||||||
|
for i in (0..n / 7).map(|val| val * 7) {
|
||||||
|
a += column.get_val(i as u64);
|
||||||
|
}
|
||||||
|
a
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
#[bench]
|
||||||
|
fn bench_intfastfield_stride7_fflookup_dense_bitmap_with_offset_20percent_dense(
|
||||||
|
b: &mut Bencher,
|
||||||
|
) {
|
||||||
|
let permutation = generate_permutation();
|
||||||
|
let n = permutation.len();
|
||||||
|
let column: Arc<dyn Column<u64>> = serialize_and_load_dense(&permutation, 200);
|
||||||
|
b.iter(|| {
|
||||||
|
let mut a = 0u64;
|
||||||
|
for i in (0..n / 7).map(|val| val * 7) {
|
||||||
|
a += column.get_val(i as u64);
|
||||||
|
}
|
||||||
|
a
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
#[bench]
|
#[bench]
|
||||||
fn bench_intfastfield_scan_all_fflookup(b: &mut Bencher) {
|
fn bench_intfastfield_scan_all_fflookup(b: &mut Bencher) {
|
||||||
let permutation = generate_permutation();
|
let permutation = generate_permutation();
|
||||||
|
|||||||
@@ -75,7 +75,7 @@ impl FastFieldCodec for BitpackedCodec {
|
|||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
fn estimate(column: &impl Column) -> Option<f32> {
|
fn estimate(column: &dyn Column) -> Option<f32> {
|
||||||
let num_bits = compute_num_bits(column.max_value());
|
let num_bits = compute_num_bits(column.max_value());
|
||||||
let num_bits_uncompressed = 64;
|
let num_bits_uncompressed = 64;
|
||||||
Some(num_bits as f32 / num_bits_uncompressed as f32)
|
Some(num_bits as f32 / num_bits_uncompressed as f32)
|
||||||
|
|||||||
@@ -71,7 +71,7 @@ impl FastFieldCodec for BlockwiseLinearCodec {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Estimate first_chunk and extrapolate
|
// Estimate first_chunk and extrapolate
|
||||||
fn estimate(column: &impl crate::Column) -> Option<f32> {
|
fn estimate(column: &dyn crate::Column) -> Option<f32> {
|
||||||
if column.num_vals() < 10 * CHUNK_SIZE as u64 {
|
if column.num_vals() < 10 * CHUNK_SIZE as u64 {
|
||||||
return None;
|
return None;
|
||||||
}
|
}
|
||||||
@@ -100,7 +100,7 @@ impl FastFieldCodec for BlockwiseLinearCodec {
|
|||||||
Some(num_bits as f32 / num_bits_uncompressed as f32)
|
Some(num_bits as f32 / num_bits_uncompressed as f32)
|
||||||
}
|
}
|
||||||
|
|
||||||
fn serialize(column: &dyn crate::Column, wrt: &mut impl io::Write) -> io::Result<()> {
|
fn serialize(column: &dyn Column, wrt: &mut impl io::Write) -> io::Result<()> {
|
||||||
// The BitpackedReader assumes a normalized vector.
|
// The BitpackedReader assumes a normalized vector.
|
||||||
assert_eq!(column.min_value(), 0);
|
assert_eq!(column.min_value(), 0);
|
||||||
let mut buffer = Vec::with_capacity(CHUNK_SIZE);
|
let mut buffer = Vec::with_capacity(CHUNK_SIZE);
|
||||||
|
|||||||
@@ -4,20 +4,13 @@ use std::ops::RangeInclusive;
|
|||||||
use tantivy_bitpacker::minmax;
|
use tantivy_bitpacker::minmax;
|
||||||
|
|
||||||
pub trait Column<T: PartialOrd = u64>: Send + Sync {
|
pub trait Column<T: PartialOrd = u64>: Send + Sync {
|
||||||
/// Return a `ColumnReader`.
|
/// Return the value associated with the given idx.
|
||||||
fn reader(&self) -> Box<dyn ColumnReader<T> + '_> {
|
|
||||||
Box::new(ColumnReaderAdapter { column: self })
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Return the value associated to the given idx.
|
|
||||||
///
|
///
|
||||||
/// This accessor should return as fast as possible.
|
/// This accessor should return as fast as possible.
|
||||||
///
|
///
|
||||||
/// # Panics
|
/// # Panics
|
||||||
///
|
///
|
||||||
/// May panic if `idx` is greater than the column length.
|
/// May panic if `idx` is greater than the column length.
|
||||||
///
|
|
||||||
/// TODO remove to force people to use `.reader()`.
|
|
||||||
fn get_val(&self, idx: u64) -> T;
|
fn get_val(&self, idx: u64) -> T;
|
||||||
|
|
||||||
/// Fills an output buffer with the fast field values
|
/// Fills an output buffer with the fast field values
|
||||||
@@ -67,40 +60,11 @@ pub trait Column<T: PartialOrd = u64>: Send + Sync {
|
|||||||
fn num_vals(&self) -> u64;
|
fn num_vals(&self) -> u64;
|
||||||
|
|
||||||
/// Returns a iterator over the data
|
/// Returns a iterator over the data
|
||||||
///
|
fn iter<'a>(&'a self) -> Box<dyn Iterator<Item = T> + 'a> {
|
||||||
/// TODO get rid of `.iter()` and extend ColumnReader instead.
|
|
||||||
fn iter(&self) -> Box<dyn Iterator<Item = T> + '_> {
|
|
||||||
Box::new((0..self.num_vals()).map(|idx| self.get_val(idx)))
|
Box::new((0..self.num_vals()).map(|idx| self.get_val(idx)))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// `ColumnReader` makes it possible to read forward through a column.
|
|
||||||
///
|
|
||||||
/// TODO add methods to make it possible to scan the column and replace `.iter()`
|
|
||||||
pub trait ColumnReader<T = u64> {
|
|
||||||
fn seek(&mut self, idx: u64) -> T;
|
|
||||||
}
|
|
||||||
|
|
||||||
pub(crate) struct ColumnReaderAdapter<'a, C: ?Sized> {
|
|
||||||
column: &'a C,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl<'a, C: ?Sized> From<&'a C> for ColumnReaderAdapter<'a, C> {
|
|
||||||
fn from(column: &'a C) -> Self {
|
|
||||||
ColumnReaderAdapter { column }
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl<'a, T, C: ?Sized> ColumnReader<T> for ColumnReaderAdapter<'a, C>
|
|
||||||
where
|
|
||||||
C: Column<T>,
|
|
||||||
T: PartialOrd<T>,
|
|
||||||
{
|
|
||||||
fn seek(&mut self, idx: u64) -> T {
|
|
||||||
self.column.get_val(idx)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub struct VecColumn<'a, T = u64> {
|
pub struct VecColumn<'a, T = u64> {
|
||||||
values: &'a [T],
|
values: &'a [T],
|
||||||
min_value: T,
|
min_value: T,
|
||||||
@@ -124,11 +88,7 @@ impl<'a, C: Column<T>, T: Copy + PartialOrd> Column<T> for &'a C {
|
|||||||
(*self).num_vals()
|
(*self).num_vals()
|
||||||
}
|
}
|
||||||
|
|
||||||
fn reader(&self) -> Box<dyn ColumnReader<T> + '_> {
|
fn iter<'b>(&'b self) -> Box<dyn Iterator<Item = T> + 'b> {
|
||||||
(*self).reader()
|
|
||||||
}
|
|
||||||
|
|
||||||
fn iter(&self) -> Box<dyn Iterator<Item = T> + '_> {
|
|
||||||
(*self).iter()
|
(*self).iter()
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -233,36 +193,10 @@ where
|
|||||||
Box::new(self.from_column.iter().map(&self.monotonic_mapping))
|
Box::new(self.from_column.iter().map(&self.monotonic_mapping))
|
||||||
}
|
}
|
||||||
|
|
||||||
fn reader(&self) -> Box<dyn ColumnReader<Output> + '_> {
|
|
||||||
Box::new(MonotonicMappingColumnReader {
|
|
||||||
col_reader: ColumnReaderAdapter::from(&self.from_column),
|
|
||||||
monotonic_mapping: &self.monotonic_mapping,
|
|
||||||
intermdiary_type: PhantomData,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
// We voluntarily do not implement get_range as it yields a regression,
|
// We voluntarily do not implement get_range as it yields a regression,
|
||||||
// and we do not have any specialized implementation anyway.
|
// and we do not have any specialized implementation anyway.
|
||||||
}
|
}
|
||||||
|
|
||||||
struct MonotonicMappingColumnReader<'a, ColR, Transform, U> {
|
|
||||||
col_reader: ColR,
|
|
||||||
monotonic_mapping: &'a Transform,
|
|
||||||
intermdiary_type: PhantomData<U>,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl<'a, U, V, ColR, Transform> ColumnReader<V>
|
|
||||||
for MonotonicMappingColumnReader<'a, ColR, Transform, U>
|
|
||||||
where
|
|
||||||
ColR: ColumnReader<U> + 'a,
|
|
||||||
Transform: Fn(U) -> V,
|
|
||||||
{
|
|
||||||
fn seek(&mut self, idx: u64) -> V {
|
|
||||||
let intermediary_value = self.col_reader.seek(idx);
|
|
||||||
(*self.monotonic_mapping)(intermediary_value)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub struct IterColumn<T>(T);
|
pub struct IterColumn<T>(T);
|
||||||
|
|
||||||
impl<T> From<T> for IterColumn<T>
|
impl<T> From<T> for IterColumn<T>
|
||||||
|
|||||||
@@ -22,6 +22,7 @@ mod compact_space;
|
|||||||
mod line;
|
mod line;
|
||||||
mod linear;
|
mod linear;
|
||||||
mod monotonic_mapping;
|
mod monotonic_mapping;
|
||||||
|
mod sparse_codec_wrapper;
|
||||||
|
|
||||||
mod column;
|
mod column;
|
||||||
mod gcd;
|
mod gcd;
|
||||||
@@ -29,12 +30,14 @@ mod serialize;
|
|||||||
|
|
||||||
use self::bitpacked::BitpackedCodec;
|
use self::bitpacked::BitpackedCodec;
|
||||||
use self::blockwise_linear::BlockwiseLinearCodec;
|
use self::blockwise_linear::BlockwiseLinearCodec;
|
||||||
pub use self::column::{monotonic_map_column, Column, ColumnReader, VecColumn};
|
pub use self::column::{monotonic_map_column, Column, VecColumn};
|
||||||
use self::linear::LinearCodec;
|
use self::linear::LinearCodec;
|
||||||
pub use self::monotonic_mapping::MonotonicallyMappableToU64;
|
pub use self::monotonic_mapping::MonotonicallyMappableToU64;
|
||||||
pub use self::serialize::{
|
pub use self::serialize::{
|
||||||
estimate, serialize, serialize_and_load, serialize_u128, NormalizedHeader,
|
estimate, serialize, serialize_and_load, serialize_u128, NormalizedHeader,
|
||||||
};
|
};
|
||||||
|
pub use sparse_codec_wrapper::DenseCodec;
|
||||||
|
pub use sparse_codec_wrapper::SparseCodecRoaringBitmap;
|
||||||
|
|
||||||
#[derive(PartialEq, Eq, PartialOrd, Ord, Debug, Clone, Copy)]
|
#[derive(PartialEq, Eq, PartialOrd, Ord, Debug, Clone, Copy)]
|
||||||
#[repr(u8)]
|
#[repr(u8)]
|
||||||
@@ -76,6 +79,44 @@ impl FastFieldCodecType {
|
|||||||
pub fn open_u128(bytes: OwnedBytes) -> io::Result<Arc<dyn Column<u128>>> {
|
pub fn open_u128(bytes: OwnedBytes) -> io::Result<Arc<dyn Column<u128>>> {
|
||||||
Ok(Arc::new(CompactSpaceDecompressor::open(bytes)?))
|
Ok(Arc::new(CompactSpaceDecompressor::open(bytes)?))
|
||||||
}
|
}
|
||||||
|
//DenseCodec
|
||||||
|
//
|
||||||
|
/// Returns the correct codec reader wrapped in the `Arc` for the data.
|
||||||
|
pub fn open_dense<T: MonotonicallyMappableToU64>(
|
||||||
|
mut bytes: OwnedBytes,
|
||||||
|
fill_ratio: u32,
|
||||||
|
) -> io::Result<Arc<dyn Column<T>>> {
|
||||||
|
let header = Header::deserialize(&mut bytes)?;
|
||||||
|
match header.codec_type {
|
||||||
|
FastFieldCodecType::Bitpacked => {
|
||||||
|
open_specific_codec_dense::<BitpackedCodec, _>(bytes, &header, fill_ratio)
|
||||||
|
}
|
||||||
|
FastFieldCodecType::Linear => {
|
||||||
|
open_specific_codec_dense::<LinearCodec, _>(bytes, &header, fill_ratio)
|
||||||
|
}
|
||||||
|
FastFieldCodecType::BlockwiseLinear => {
|
||||||
|
open_specific_codec_dense::<BlockwiseLinearCodec, _>(bytes, &header, fill_ratio)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn open_specific_codec_dense<C: FastFieldCodec, Item: MonotonicallyMappableToU64>(
|
||||||
|
bytes: OwnedBytes,
|
||||||
|
header: &Header,
|
||||||
|
fill_ratio: u32,
|
||||||
|
) -> io::Result<Arc<dyn Column<Item>>> {
|
||||||
|
let normalized_header = header.normalized();
|
||||||
|
let reader = C::open_from_bytes(bytes, normalized_header)?;
|
||||||
|
let reader = DenseCodec::with_fill_ratio(reader, fill_ratio);
|
||||||
|
let min_value = header.min_value;
|
||||||
|
if let Some(gcd) = header.gcd {
|
||||||
|
let monotonic_mapping = move |val: u64| Item::from_u64(min_value + val * gcd.get());
|
||||||
|
Ok(Arc::new(monotonic_map_column(reader, monotonic_mapping)))
|
||||||
|
} else {
|
||||||
|
let monotonic_mapping = move |val: u64| Item::from_u64(min_value + val);
|
||||||
|
Ok(Arc::new(monotonic_map_column(reader, monotonic_mapping)))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
/// Returns the correct codec reader wrapped in the `Arc` for the data.
|
/// Returns the correct codec reader wrapped in the `Arc` for the data.
|
||||||
pub fn open<T: MonotonicallyMappableToU64>(
|
pub fn open<T: MonotonicallyMappableToU64>(
|
||||||
@@ -123,7 +164,7 @@ trait FastFieldCodec: 'static {
|
|||||||
///
|
///
|
||||||
/// The column iterator should be preferred over using column `get_val` method for
|
/// The column iterator should be preferred over using column `get_val` method for
|
||||||
/// performance reasons.
|
/// performance reasons.
|
||||||
fn serialize(column: &dyn Column<u64>, write: &mut impl Write) -> io::Result<()>;
|
fn serialize(column: &dyn Column, write: &mut impl Write) -> io::Result<()>;
|
||||||
|
|
||||||
/// Returns an estimate of the compression ratio.
|
/// Returns an estimate of the compression ratio.
|
||||||
/// If the codec is not applicable, returns `None`.
|
/// If the codec is not applicable, returns `None`.
|
||||||
@@ -132,7 +173,7 @@ trait FastFieldCodec: 'static {
|
|||||||
///
|
///
|
||||||
/// It could make sense to also return a value representing
|
/// It could make sense to also return a value representing
|
||||||
/// computational complexity.
|
/// computational complexity.
|
||||||
fn estimate(column: &impl Column) -> Option<f32>;
|
fn estimate(column: &dyn Column) -> Option<f32>;
|
||||||
}
|
}
|
||||||
|
|
||||||
pub const ALL_CODEC_TYPES: [FastFieldCodecType; 3] = [
|
pub const ALL_CODEC_TYPES: [FastFieldCodecType; 3] = [
|
||||||
@@ -440,6 +481,7 @@ mod bench {
|
|||||||
let data: Vec<_> = get_data();
|
let data: Vec<_> = get_data();
|
||||||
bench_get::<BitpackedCodec>(b, &data);
|
bench_get::<BitpackedCodec>(b, &data);
|
||||||
}
|
}
|
||||||
|
|
||||||
#[bench]
|
#[bench]
|
||||||
fn bench_fastfield_bitpack_get_dynamic(b: &mut Bencher) {
|
fn bench_fastfield_bitpack_get_dynamic(b: &mut Bencher) {
|
||||||
let data: Vec<_> = get_data();
|
let data: Vec<_> = get_data();
|
||||||
|
|||||||
@@ -69,23 +69,28 @@ impl Line {
|
|||||||
|
|
||||||
// Same as train, but the intercept is only estimated from provided sample positions
|
// Same as train, but the intercept is only estimated from provided sample positions
|
||||||
pub fn estimate(ys: &dyn Column, sample_positions: &[u64]) -> Self {
|
pub fn estimate(ys: &dyn Column, sample_positions: &[u64]) -> Self {
|
||||||
Self::train_from(ys, sample_positions.iter().cloned())
|
Self::train_from(
|
||||||
|
ys,
|
||||||
|
sample_positions
|
||||||
|
.iter()
|
||||||
|
.cloned()
|
||||||
|
.map(|pos| (pos, ys.get_val(pos))),
|
||||||
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Intercept is only computed from provided positions
|
// Intercept is only computed from provided positions
|
||||||
fn train_from(ys: &dyn Column, positions: impl Iterator<Item = u64>) -> Self {
|
fn train_from(ys: &dyn Column, positions_and_values: impl Iterator<Item = (u64, u64)>) -> Self {
|
||||||
let last_idx = if let Some(last_idx) = NonZeroU64::new(ys.num_vals() - 1) {
|
let num_vals = if let Some(num_vals) = NonZeroU64::new(ys.num_vals() - 1) {
|
||||||
last_idx
|
num_vals
|
||||||
} else {
|
} else {
|
||||||
return Line::default();
|
return Line::default();
|
||||||
};
|
};
|
||||||
|
|
||||||
let mut ys_reader = ys.reader();
|
let y0 = ys.get_val(0);
|
||||||
let y0 = ys_reader.seek(0);
|
let y1 = ys.get_val(num_vals.get());
|
||||||
let y1 = ys_reader.seek(last_idx.get());
|
|
||||||
|
|
||||||
// We first independently pick our slope.
|
// We first independently pick our slope.
|
||||||
let slope = compute_slope(y0, y1, last_idx);
|
let slope = compute_slope(y0, y1, num_vals);
|
||||||
|
|
||||||
// We picked our slope. Note that it does not have to be perfect.
|
// We picked our slope. Note that it does not have to be perfect.
|
||||||
// Now we need to compute the best intercept.
|
// Now we need to compute the best intercept.
|
||||||
@@ -115,12 +120,8 @@ impl Line {
|
|||||||
intercept: 0,
|
intercept: 0,
|
||||||
};
|
};
|
||||||
let heuristic_shift = y0.wrapping_sub(MID_POINT);
|
let heuristic_shift = y0.wrapping_sub(MID_POINT);
|
||||||
let mut ys_reader = ys.reader();
|
line.intercept = positions_and_values
|
||||||
line.intercept = positions
|
.map(|(pos, y)| y.wrapping_sub(line.eval(pos)))
|
||||||
.map(|pos| {
|
|
||||||
let y = ys_reader.seek(pos);
|
|
||||||
y.wrapping_sub(line.eval(pos))
|
|
||||||
})
|
|
||||||
.min_by_key(|&val| val.wrapping_sub(heuristic_shift))
|
.min_by_key(|&val| val.wrapping_sub(heuristic_shift))
|
||||||
.unwrap_or(0u64); //< Never happens.
|
.unwrap_or(0u64); //< Never happens.
|
||||||
line
|
line
|
||||||
@@ -137,7 +138,10 @@ impl Line {
|
|||||||
/// This function is only invariable by translation if all of the
|
/// This function is only invariable by translation if all of the
|
||||||
/// `ys` are packaged into half of the space. (See heuristic below)
|
/// `ys` are packaged into half of the space. (See heuristic below)
|
||||||
pub fn train(ys: &dyn Column) -> Self {
|
pub fn train(ys: &dyn Column) -> Self {
|
||||||
Self::train_from(ys, 0..ys.num_vals())
|
Self::train_from(
|
||||||
|
ys,
|
||||||
|
ys.iter().enumerate().map(|(pos, val)| (pos as u64, val)),
|
||||||
|
)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -121,7 +121,7 @@ impl FastFieldCodec for LinearCodec {
|
|||||||
/// where the local maxima for the deviation of the calculated value are and
|
/// where the local maxima for the deviation of the calculated value are and
|
||||||
/// the offset to shift all values to >=0 is also unknown.
|
/// the offset to shift all values to >=0 is also unknown.
|
||||||
#[allow(clippy::question_mark)]
|
#[allow(clippy::question_mark)]
|
||||||
fn estimate(column: &impl Column) -> Option<f32> {
|
fn estimate(column: &dyn Column) -> Option<f32> {
|
||||||
if column.num_vals() < 3 {
|
if column.num_vals() < 3 {
|
||||||
return None; // disable compressor for this case
|
return None; // disable compressor for this case
|
||||||
}
|
}
|
||||||
@@ -134,11 +134,10 @@ impl FastFieldCodec for LinearCodec {
|
|||||||
|
|
||||||
let line = Line::estimate(column, &sample_positions);
|
let line = Line::estimate(column, &sample_positions);
|
||||||
|
|
||||||
let mut column_reader = column.reader();
|
|
||||||
let estimated_bit_width = sample_positions
|
let estimated_bit_width = sample_positions
|
||||||
.into_iter()
|
.into_iter()
|
||||||
.map(|pos| {
|
.map(|pos| {
|
||||||
let actual_value = column_reader.seek(pos);
|
let actual_value = column.get_val(pos);
|
||||||
let interpolated_val = line.eval(pos as u64);
|
let interpolated_val = line.eval(pos as u64);
|
||||||
actual_value.wrapping_sub(interpolated_val)
|
actual_value.wrapping_sub(interpolated_val)
|
||||||
})
|
})
|
||||||
|
|||||||
@@ -36,11 +36,7 @@ impl MonotonicallyMappableToU64 for i64 {
|
|||||||
impl MonotonicallyMappableToU64 for bool {
|
impl MonotonicallyMappableToU64 for bool {
|
||||||
#[inline(always)]
|
#[inline(always)]
|
||||||
fn to_u64(self) -> u64 {
|
fn to_u64(self) -> u64 {
|
||||||
if self {
|
u64::from(self)
|
||||||
1
|
|
||||||
} else {
|
|
||||||
0
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
#[inline(always)]
|
#[inline(always)]
|
||||||
|
|||||||
42
fastfield_codecs/src/monotonic_mapping_u128.rs
Normal file
42
fastfield_codecs/src/monotonic_mapping_u128.rs
Normal file
@@ -0,0 +1,42 @@
|
|||||||
|
use std::net::{IpAddr, Ipv6Addr};
|
||||||
|
|
||||||
|
pub trait MonotonicallyMappableToU128: 'static + PartialOrd + Copy + Send + Sync {
|
||||||
|
/// Converts a value to u128.
|
||||||
|
///
|
||||||
|
/// Internally all fast field values are encoded as u64.
|
||||||
|
fn to_u128(self) -> u128;
|
||||||
|
|
||||||
|
/// Converts a value from u128
|
||||||
|
///
|
||||||
|
/// Internally all fast field values are encoded as u64.
|
||||||
|
/// **Note: To be used for converting encoded Term, Posting values.**
|
||||||
|
fn from_u128(val: u128) -> Self;
|
||||||
|
}
|
||||||
|
|
||||||
|
impl MonotonicallyMappableToU128 for u128 {
|
||||||
|
fn to_u128(self) -> u128 {
|
||||||
|
self
|
||||||
|
}
|
||||||
|
|
||||||
|
fn from_u128(val: u128) -> Self {
|
||||||
|
val
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl MonotonicallyMappableToU128 for IpAddr {
|
||||||
|
fn to_u128(self) -> u128 {
|
||||||
|
ip_to_u128(self)
|
||||||
|
}
|
||||||
|
|
||||||
|
fn from_u128(val: u128) -> Self {
|
||||||
|
IpAddr::from(val.to_be_bytes())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn ip_to_u128(ip_addr: IpAddr) -> u128 {
|
||||||
|
let ip_addr_v6: Ipv6Addr = match ip_addr {
|
||||||
|
IpAddr::V4(v4) => v4.to_ipv6_mapped(),
|
||||||
|
IpAddr::V6(v6) => v6,
|
||||||
|
};
|
||||||
|
u128::from_be_bytes(ip_addr_v6.octets())
|
||||||
|
}
|
||||||
157
fastfield_codecs/src/sparse_codec_wrapper.rs
Normal file
157
fastfield_codecs/src/sparse_codec_wrapper.rs
Normal file
@@ -0,0 +1,157 @@
|
|||||||
|
use std::sync::Arc;
|
||||||
|
|
||||||
|
use rand::{thread_rng, Rng};
|
||||||
|
use roaring::RoaringBitmap;
|
||||||
|
|
||||||
|
use crate::Column;
|
||||||
|
|
||||||
|
pub struct SparseCodecRoaringBitmap {
|
||||||
|
null: RoaringBitmap,
|
||||||
|
column: Arc<dyn Column<u64>>, // column: C,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl SparseCodecRoaringBitmap {
|
||||||
|
pub fn with_full(column: Arc<dyn Column<u64>>) -> Self {
|
||||||
|
let mut rb = RoaringBitmap::new();
|
||||||
|
rb.insert_range(0..column.num_vals() as u32 + 1);
|
||||||
|
Self { null: rb, column }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Column for SparseCodecRoaringBitmap {
|
||||||
|
fn get_val(&self, idx: u64) -> u64 {
|
||||||
|
let position_of_val = self.null.rank(idx as u32);
|
||||||
|
self.column.get_val(position_of_val) // TODO this does not handle null!
|
||||||
|
// self.null.select(num_vals)
|
||||||
|
}
|
||||||
|
|
||||||
|
fn min_value(&self) -> u64 {
|
||||||
|
todo!()
|
||||||
|
}
|
||||||
|
|
||||||
|
fn max_value(&self) -> u64 {
|
||||||
|
todo!()
|
||||||
|
}
|
||||||
|
|
||||||
|
fn num_vals(&self) -> u64 {
|
||||||
|
todo!()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub struct DenseCodec<C> {
|
||||||
|
// the bitmap blocks of length 64 bit each
|
||||||
|
blocks: Vec<u64>,
|
||||||
|
// the offset for each block
|
||||||
|
offsets: Vec<u32>,
|
||||||
|
column: C, // column: C,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<C: Column> DenseCodec<C> {
|
||||||
|
// fill ratio valid range 0..1000 1000 == all elements, 1 == every 1000th element
|
||||||
|
pub fn with_fill_ratio(column: C, fill_ratio: u32) -> Self {
|
||||||
|
let mut rng = thread_rng();
|
||||||
|
let num_blocks = (column.num_vals() as usize / 64) + 1;
|
||||||
|
let mut blocks = Vec::with_capacity(num_blocks);
|
||||||
|
let mut offsets = Vec::with_capacity(num_blocks);
|
||||||
|
// fill all blocks
|
||||||
|
let mut offset = 0;
|
||||||
|
for _block_num in 0..num_blocks {
|
||||||
|
let mut block = 0;
|
||||||
|
for n in 0..64 {
|
||||||
|
if rng.gen_range(0..=1000) <= fill_ratio {
|
||||||
|
set_bit_at(&mut block, n);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
blocks.push(block);
|
||||||
|
offsets.push(offset);
|
||||||
|
offset += block.count_ones();
|
||||||
|
}
|
||||||
|
|
||||||
|
Self {
|
||||||
|
blocks,
|
||||||
|
offsets,
|
||||||
|
column,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn with_full(column: C) -> Self {
|
||||||
|
let num_blocks = (column.num_vals() as usize / 64) + 1;
|
||||||
|
let mut blocks = Vec::with_capacity(num_blocks);
|
||||||
|
let mut offsets = Vec::with_capacity(num_blocks);
|
||||||
|
// fill all blocks
|
||||||
|
let mut offset = 0;
|
||||||
|
for _block_num in 0..num_blocks {
|
||||||
|
let block = u64::MAX;
|
||||||
|
blocks.push(block);
|
||||||
|
offsets.push(offset);
|
||||||
|
offset += block.count_ones();
|
||||||
|
}
|
||||||
|
|
||||||
|
Self {
|
||||||
|
blocks,
|
||||||
|
offsets,
|
||||||
|
column,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn gen_mask(msb: u64) -> u64 {
|
||||||
|
let src = 1 << msb;
|
||||||
|
src - 1
|
||||||
|
}
|
||||||
|
|
||||||
|
fn get_bit_at(input: u64, n: u64) -> bool {
|
||||||
|
input & (1 << n) != 0
|
||||||
|
}
|
||||||
|
|
||||||
|
fn set_bit_at(input: &mut u64, n: u64) {
|
||||||
|
*input |= 1 << n;
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<C: Column> Column for DenseCodec<C> {
|
||||||
|
fn get_val(&self, idx: u64) -> u64 {
|
||||||
|
let block_pos = idx / 64;
|
||||||
|
let pos_in_block = idx % 64;
|
||||||
|
let offset = self.offsets[block_pos as usize];
|
||||||
|
let bitvec = self.blocks[block_pos as usize];
|
||||||
|
let offset_in_block = (bitvec & gen_mask(pos_in_block)).count_ones();
|
||||||
|
let dense_idx = offset as u64 + offset_in_block as u64;
|
||||||
|
if get_bit_at(bitvec, pos_in_block) {
|
||||||
|
self.column.get_val(dense_idx)
|
||||||
|
} else {
|
||||||
|
0 // TODO null
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn min_value(&self) -> u64 {
|
||||||
|
todo!()
|
||||||
|
}
|
||||||
|
|
||||||
|
fn max_value(&self) -> u64 {
|
||||||
|
todo!()
|
||||||
|
}
|
||||||
|
|
||||||
|
fn num_vals(&self) -> u64 {
|
||||||
|
todo!()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
//use itertools::Itertools;
|
||||||
|
|
||||||
|
//use super::*;
|
||||||
|
//use crate::serialize_and_load;
|
||||||
|
|
||||||
|
//#[test]
|
||||||
|
//fn dense_test() {
|
||||||
|
//let data = (0..100u64).collect_vec();
|
||||||
|
//{
|
||||||
|
//let column = serialize_and_load(&data);
|
||||||
|
//let dense = DenseCodec::with_full(column);
|
||||||
|
//for i in 0..100 {
|
||||||
|
//dense.get_val(i);
|
||||||
|
//}
|
||||||
|
//}
|
||||||
|
//}
|
||||||
|
}
|
||||||
@@ -425,7 +425,7 @@ impl SegmentHistogramCollector {
|
|||||||
let bucket = &mut self.buckets[bucket_pos];
|
let bucket = &mut self.buckets[bucket_pos];
|
||||||
bucket.doc_count += 1;
|
bucket.doc_count += 1;
|
||||||
if let Some(sub_aggregation) = self.sub_aggregations.as_mut() {
|
if let Some(sub_aggregation) = self.sub_aggregations.as_mut() {
|
||||||
(&mut sub_aggregation[bucket_pos]).collect(doc, bucket_with_accessor)?;
|
sub_aggregation[bucket_pos].collect(doc, bucket_with_accessor)?;
|
||||||
}
|
}
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -38,7 +38,7 @@ pub trait CustomSegmentScorer<TScore>: 'static {
|
|||||||
pub trait CustomScorer<TScore>: Sync {
|
pub trait CustomScorer<TScore>: Sync {
|
||||||
/// Type of the associated [`CustomSegmentScorer`].
|
/// Type of the associated [`CustomSegmentScorer`].
|
||||||
type Child: CustomSegmentScorer<TScore>;
|
type Child: CustomSegmentScorer<TScore>;
|
||||||
/// Builds a child scorer for a specific segment. The child scorer is associated to
|
/// Builds a child scorer for a specific segment. The child scorer is associated with
|
||||||
/// a specific segment.
|
/// a specific segment.
|
||||||
fn segment_scorer(&self, segment_reader: &SegmentReader) -> crate::Result<Self::Child>;
|
fn segment_scorer(&self, segment_reader: &SegmentReader) -> crate::Result<Self::Child>;
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -91,7 +91,7 @@ fn facet_depth(facet_bytes: &[u8]) -> usize {
|
|||||||
/// let index = Index::create_in_ram(schema);
|
/// let index = Index::create_in_ram(schema);
|
||||||
/// {
|
/// {
|
||||||
/// let mut index_writer = index.writer(3_000_000)?;
|
/// let mut index_writer = index.writer(3_000_000)?;
|
||||||
/// // a document can be associated to any number of facets
|
/// // a document can be associated with any number of facets
|
||||||
/// index_writer.add_document(doc!(
|
/// index_writer.add_document(doc!(
|
||||||
/// title => "The Name of the Wind",
|
/// title => "The Name of the Wind",
|
||||||
/// facet => Facet::from("/lang/en"),
|
/// facet => Facet::from("/lang/en"),
|
||||||
|
|||||||
@@ -37,7 +37,7 @@ impl HistogramCollector {
|
|||||||
/// The scale/range of the histogram is not dynamic. It is required to
|
/// The scale/range of the histogram is not dynamic. It is required to
|
||||||
/// define it by supplying following parameter:
|
/// define it by supplying following parameter:
|
||||||
/// - `min_value`: the minimum value that can be recorded in the histogram.
|
/// - `min_value`: the minimum value that can be recorded in the histogram.
|
||||||
/// - `bucket_width`: the length of the interval that is associated to each buckets.
|
/// - `bucket_width`: the length of the interval that is associated with each buckets.
|
||||||
/// - `num_buckets`: The overall number of buckets.
|
/// - `num_buckets`: The overall number of buckets.
|
||||||
///
|
///
|
||||||
/// Together, this parameters define a partition of `[min_value, min_value + num_buckets *
|
/// Together, this parameters define a partition of `[min_value, min_value + num_buckets *
|
||||||
|
|||||||
@@ -142,7 +142,7 @@ pub trait Collector: Sync + Send {
|
|||||||
/// e.g. `usize` for the `Count` collector.
|
/// e.g. `usize` for the `Count` collector.
|
||||||
type Fruit: Fruit;
|
type Fruit: Fruit;
|
||||||
|
|
||||||
/// Type of the `SegmentCollector` associated to this collector.
|
/// Type of the `SegmentCollector` associated with this collector.
|
||||||
type Child: SegmentCollector;
|
type Child: SegmentCollector;
|
||||||
|
|
||||||
/// `set_segment` is called before beginning to enumerate
|
/// `set_segment` is called before beginning to enumerate
|
||||||
@@ -156,7 +156,7 @@ pub trait Collector: Sync + Send {
|
|||||||
/// Returns true iff the collector requires to compute scores for documents.
|
/// Returns true iff the collector requires to compute scores for documents.
|
||||||
fn requires_scoring(&self) -> bool;
|
fn requires_scoring(&self) -> bool;
|
||||||
|
|
||||||
/// Combines the fruit associated to the collection of each segments
|
/// Combines the fruit associated with the collection of each segments
|
||||||
/// into one fruit.
|
/// into one fruit.
|
||||||
fn merge_fruits(
|
fn merge_fruits(
|
||||||
&self,
|
&self,
|
||||||
|
|||||||
@@ -693,7 +693,7 @@ impl Collector for TopDocs {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Segment Collector associated to `TopDocs`.
|
/// Segment Collector associated with `TopDocs`.
|
||||||
pub struct TopScoreSegmentCollector(TopSegmentCollector<Score>);
|
pub struct TopScoreSegmentCollector(TopSegmentCollector<Score>);
|
||||||
|
|
||||||
impl SegmentCollector for TopScoreSegmentCollector {
|
impl SegmentCollector for TopScoreSegmentCollector {
|
||||||
|
|||||||
@@ -40,7 +40,7 @@ pub trait ScoreTweaker<TScore>: Sync {
|
|||||||
/// Type of the associated [`ScoreSegmentTweaker`].
|
/// Type of the associated [`ScoreSegmentTweaker`].
|
||||||
type Child: ScoreSegmentTweaker<TScore>;
|
type Child: ScoreSegmentTweaker<TScore>;
|
||||||
|
|
||||||
/// Builds a child tweaker for a specific segment. The child scorer is associated to
|
/// Builds a child tweaker for a specific segment. The child scorer is associated with
|
||||||
/// a specific segment.
|
/// a specific segment.
|
||||||
fn segment_tweaker(&self, segment_reader: &SegmentReader) -> Result<Self::Child>;
|
fn segment_tweaker(&self, segment_reader: &SegmentReader) -> Result<Self::Child>;
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -130,7 +130,7 @@ impl SegmentMeta {
|
|||||||
/// Returns the relative path of a component of our segment.
|
/// Returns the relative path of a component of our segment.
|
||||||
///
|
///
|
||||||
/// It just joins the segment id with the extension
|
/// It just joins the segment id with the extension
|
||||||
/// associated to a segment component.
|
/// associated with a segment component.
|
||||||
pub fn relative_path(&self, component: SegmentComponent) -> PathBuf {
|
pub fn relative_path(&self, component: SegmentComponent) -> PathBuf {
|
||||||
let mut path = self.id().uuid_string();
|
let mut path = self.id().uuid_string();
|
||||||
path.push_str(&*match component {
|
path.push_str(&*match component {
|
||||||
@@ -326,13 +326,13 @@ pub struct IndexMeta {
|
|||||||
/// `IndexSettings` to configure index options.
|
/// `IndexSettings` to configure index options.
|
||||||
#[serde(default)]
|
#[serde(default)]
|
||||||
pub index_settings: IndexSettings,
|
pub index_settings: IndexSettings,
|
||||||
/// List of `SegmentMeta` information associated to each finalized segment of the index.
|
/// List of `SegmentMeta` information associated with each finalized segment of the index.
|
||||||
pub segments: Vec<SegmentMeta>,
|
pub segments: Vec<SegmentMeta>,
|
||||||
/// Index `Schema`
|
/// Index `Schema`
|
||||||
pub schema: Schema,
|
pub schema: Schema,
|
||||||
/// Opstamp associated to the last `commit` operation.
|
/// Opstamp associated with the last `commit` operation.
|
||||||
pub opstamp: Opstamp,
|
pub opstamp: Opstamp,
|
||||||
/// Payload associated to the last commit.
|
/// Payload associated with the last commit.
|
||||||
///
|
///
|
||||||
/// Upon commit, clients can optionally add a small `String` payload to their commit
|
/// Upon commit, clients can optionally add a small `String` payload to their commit
|
||||||
/// to help identify this commit.
|
/// to help identify this commit.
|
||||||
|
|||||||
@@ -9,11 +9,11 @@ use crate::schema::{IndexRecordOption, Term};
|
|||||||
use crate::termdict::TermDictionary;
|
use crate::termdict::TermDictionary;
|
||||||
|
|
||||||
/// The inverted index reader is in charge of accessing
|
/// The inverted index reader is in charge of accessing
|
||||||
/// the inverted index associated to a specific field.
|
/// the inverted index associated with a specific field.
|
||||||
///
|
///
|
||||||
/// # Note
|
/// # Note
|
||||||
///
|
///
|
||||||
/// It is safe to delete the segment associated to
|
/// It is safe to delete the segment associated with
|
||||||
/// an `InvertedIndexReader`. As long as it is open,
|
/// an `InvertedIndexReader`. As long as it is open,
|
||||||
/// the `FileSlice` it is relying on should
|
/// the `FileSlice` it is relying on should
|
||||||
/// stay available.
|
/// stay available.
|
||||||
@@ -30,7 +30,7 @@ pub struct InvertedIndexReader {
|
|||||||
}
|
}
|
||||||
|
|
||||||
impl InvertedIndexReader {
|
impl InvertedIndexReader {
|
||||||
#[cfg_attr(feature = "cargo-clippy", allow(clippy::needless_pass_by_value))] // for symmetry
|
#[allow(clippy::needless_pass_by_value)] // for symmetry
|
||||||
pub(crate) fn new(
|
pub(crate) fn new(
|
||||||
termdict: TermDictionary,
|
termdict: TermDictionary,
|
||||||
postings_file_slice: FileSlice,
|
postings_file_slice: FileSlice,
|
||||||
|
|||||||
@@ -69,7 +69,7 @@ pub struct Searcher {
|
|||||||
}
|
}
|
||||||
|
|
||||||
impl Searcher {
|
impl Searcher {
|
||||||
/// Returns the `Index` associated to the `Searcher`
|
/// Returns the `Index` associated with the `Searcher`
|
||||||
pub fn index(&self) -> &Index {
|
pub fn index(&self) -> &Index {
|
||||||
&self.inner.index
|
&self.inner.index
|
||||||
}
|
}
|
||||||
@@ -108,7 +108,7 @@ impl Searcher {
|
|||||||
store_reader.get_async(doc_address.doc_id).await
|
store_reader.get_async(doc_address.doc_id).await
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Access the schema associated to the index of this searcher.
|
/// Access the schema associated with the index of this searcher.
|
||||||
pub fn schema(&self) -> &Schema {
|
pub fn schema(&self) -> &Schema {
|
||||||
&self.inner.schema
|
&self.inner.schema
|
||||||
}
|
}
|
||||||
@@ -161,11 +161,11 @@ impl Searcher {
|
|||||||
///
|
///
|
||||||
/// Search works as follows :
|
/// Search works as follows :
|
||||||
///
|
///
|
||||||
/// First the weight object associated to the query is created.
|
/// First the weight object associated with the query is created.
|
||||||
///
|
///
|
||||||
/// Then, the query loops over the segments and for each segment :
|
/// Then, the query loops over the segments and for each segment :
|
||||||
/// - setup the collector and informs it that the segment being processed has changed.
|
/// - setup the collector and informs it that the segment being processed has changed.
|
||||||
/// - creates a SegmentCollector for collecting documents associated to the segment
|
/// - creates a SegmentCollector for collecting documents associated with the segment
|
||||||
/// - creates a `Scorer` object associated for this segment
|
/// - creates a `Scorer` object associated for this segment
|
||||||
/// - iterate through the matched documents and push them to the segment collector.
|
/// - iterate through the matched documents and push them to the segment collector.
|
||||||
///
|
///
|
||||||
|
|||||||
@@ -70,7 +70,7 @@ impl Segment {
|
|||||||
/// Returns the relative path of a component of our segment.
|
/// Returns the relative path of a component of our segment.
|
||||||
///
|
///
|
||||||
/// It just joins the segment id with the extension
|
/// It just joins the segment id with the extension
|
||||||
/// associated to a segment component.
|
/// associated with a segment component.
|
||||||
pub fn relative_path(&self, component: SegmentComponent) -> PathBuf {
|
pub fn relative_path(&self, component: SegmentComponent) -> PathBuf {
|
||||||
self.meta.relative_path(component)
|
self.meta.relative_path(component)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -6,7 +6,7 @@ use std::slice;
|
|||||||
/// except the delete component that takes an `segment_uuid`.`delete_opstamp`.`component_extension`
|
/// except the delete component that takes an `segment_uuid`.`delete_opstamp`.`component_extension`
|
||||||
#[derive(Copy, Clone, Eq, PartialEq)]
|
#[derive(Copy, Clone, Eq, PartialEq)]
|
||||||
pub enum SegmentComponent {
|
pub enum SegmentComponent {
|
||||||
/// Postings (or inverted list). Sorted lists of document ids, associated to terms
|
/// Postings (or inverted list). Sorted lists of document ids, associated with terms
|
||||||
Postings,
|
Postings,
|
||||||
/// Positions of terms in each document.
|
/// Positions of terms in each document.
|
||||||
Positions,
|
Positions,
|
||||||
|
|||||||
@@ -57,7 +57,7 @@ impl SegmentId {
|
|||||||
/// Picking the first 8 chars is ok to identify
|
/// Picking the first 8 chars is ok to identify
|
||||||
/// segments in a display message (e.g. a5c4dfcb).
|
/// segments in a display message (e.g. a5c4dfcb).
|
||||||
pub fn short_uuid_string(&self) -> String {
|
pub fn short_uuid_string(&self) -> String {
|
||||||
(&self.0.as_simple().to_string()[..8]).to_string()
|
self.0.as_simple().to_string()[..8].to_string()
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Returns a segment uuid string.
|
/// Returns a segment uuid string.
|
||||||
|
|||||||
@@ -89,7 +89,7 @@ impl SegmentReader {
|
|||||||
&self.fast_fields_readers
|
&self.fast_fields_readers
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Accessor to the `FacetReader` associated to a given `Field`.
|
/// Accessor to the `FacetReader` associated with a given `Field`.
|
||||||
pub fn facet_reader(&self, field: Field) -> crate::Result<FacetReader> {
|
pub fn facet_reader(&self, field: Field) -> crate::Result<FacetReader> {
|
||||||
let field_entry = self.schema.get_field_entry(field);
|
let field_entry = self.schema.get_field_entry(field);
|
||||||
|
|
||||||
@@ -208,13 +208,13 @@ impl SegmentReader {
|
|||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Returns a field reader associated to the field given in argument.
|
/// Returns a field reader associated with the field given in argument.
|
||||||
/// If the field was not present in the index during indexing time,
|
/// If the field was not present in the index during indexing time,
|
||||||
/// the InvertedIndexReader is empty.
|
/// the InvertedIndexReader is empty.
|
||||||
///
|
///
|
||||||
/// The field reader is in charge of iterating through the
|
/// The field reader is in charge of iterating through the
|
||||||
/// term dictionary associated to a specific field,
|
/// term dictionary associated with a specific field,
|
||||||
/// and opening the posting list associated to any term.
|
/// and opening the posting list associated with any term.
|
||||||
///
|
///
|
||||||
/// If the field is not marked as index, a warn is logged and an empty `InvertedIndexReader`
|
/// If the field is not marked as index, a warn is logged and an empty `InvertedIndexReader`
|
||||||
/// is returned.
|
/// is returned.
|
||||||
@@ -241,7 +241,7 @@ impl SegmentReader {
|
|||||||
|
|
||||||
if postings_file_opt.is_none() || record_option_opt.is_none() {
|
if postings_file_opt.is_none() || record_option_opt.is_none() {
|
||||||
// no documents in the segment contained this field.
|
// no documents in the segment contained this field.
|
||||||
// As a result, no data is associated to the inverted index.
|
// As a result, no data is associated with the inverted index.
|
||||||
//
|
//
|
||||||
// Returns an empty inverted index.
|
// Returns an empty inverted index.
|
||||||
let record_option = record_option_opt.unwrap_or(IndexRecordOption::Basic);
|
let record_option = record_option_opt.unwrap_or(IndexRecordOption::Basic);
|
||||||
|
|||||||
@@ -154,14 +154,14 @@ impl CompositeFile {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Returns the `FileSlice` associated
|
/// Returns the `FileSlice` associated with
|
||||||
/// to a given `Field` and stored in a `CompositeFile`.
|
/// a given `Field` and stored in a `CompositeFile`.
|
||||||
pub fn open_read(&self, field: Field) -> Option<FileSlice> {
|
pub fn open_read(&self, field: Field) -> Option<FileSlice> {
|
||||||
self.open_read_with_idx(field, 0)
|
self.open_read_with_idx(field, 0)
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Returns the `FileSlice` associated
|
/// Returns the `FileSlice` associated with
|
||||||
/// to a given `Field` and stored in a `CompositeFile`.
|
/// a given `Field` and stored in a `CompositeFile`.
|
||||||
pub fn open_read_with_idx(&self, field: Field, idx: usize) -> Option<FileSlice> {
|
pub fn open_read_with_idx(&self, field: Field, idx: usize) -> Option<FileSlice> {
|
||||||
self.offsets_index
|
self.offsets_index
|
||||||
.get(&FileAddr { field, idx })
|
.get(&FileAddr { field, idx })
|
||||||
|
|||||||
@@ -39,7 +39,7 @@ impl RetryPolicy {
|
|||||||
|
|
||||||
/// The `DirectoryLock` is an object that represents a file lock.
|
/// The `DirectoryLock` is an object that represents a file lock.
|
||||||
///
|
///
|
||||||
/// It is associated to a lock file, that gets deleted on `Drop.`
|
/// It is associated with a lock file, that gets deleted on `Drop.`
|
||||||
pub struct DirectoryLock(Box<dyn Send + Sync + 'static>);
|
pub struct DirectoryLock(Box<dyn Send + Sync + 'static>);
|
||||||
|
|
||||||
struct DirectoryLockGuard {
|
struct DirectoryLockGuard {
|
||||||
|
|||||||
@@ -1,5 +1,5 @@
|
|||||||
use std::ops::{Deref, Range};
|
use std::ops::{Deref, Range};
|
||||||
use std::sync::{Arc, Weak};
|
use std::sync::Arc;
|
||||||
use std::{fmt, io};
|
use std::{fmt, io};
|
||||||
|
|
||||||
use async_trait::async_trait;
|
use async_trait::async_trait;
|
||||||
@@ -8,9 +8,6 @@ use stable_deref_trait::StableDeref;
|
|||||||
|
|
||||||
use crate::directory::OwnedBytes;
|
use crate::directory::OwnedBytes;
|
||||||
|
|
||||||
pub type ArcBytes = Arc<dyn Deref<Target = [u8]> + Send + Sync + 'static>;
|
|
||||||
pub type WeakArcBytes = Weak<dyn Deref<Target = [u8]> + Send + Sync + 'static>;
|
|
||||||
|
|
||||||
/// Objects that represents files sections in tantivy.
|
/// Objects that represents files sections in tantivy.
|
||||||
///
|
///
|
||||||
/// By contract, whatever happens to the directory file, as long as a FileHandle
|
/// By contract, whatever happens to the directory file, as long as a FileHandle
|
||||||
|
|||||||
@@ -9,7 +9,7 @@ use crc32fast::Hasher;
|
|||||||
|
|
||||||
use crate::directory::{WatchCallback, WatchCallbackList, WatchHandle};
|
use crate::directory::{WatchCallback, WatchCallbackList, WatchHandle};
|
||||||
|
|
||||||
pub const POLLING_INTERVAL: Duration = Duration::from_millis(if cfg!(test) { 1 } else { 500 });
|
const POLLING_INTERVAL: Duration = Duration::from_millis(if cfg!(test) { 1 } else { 500 });
|
||||||
|
|
||||||
// Watches a file and executes registered callbacks when the file is modified.
|
// Watches a file and executes registered callbacks when the file is modified.
|
||||||
pub struct FileWatcher {
|
pub struct FileWatcher {
|
||||||
|
|||||||
@@ -3,7 +3,7 @@ use std::fs::{self, File, OpenOptions};
|
|||||||
use std::io::{self, BufWriter, Read, Seek, Write};
|
use std::io::{self, BufWriter, Read, Seek, Write};
|
||||||
use std::ops::Deref;
|
use std::ops::Deref;
|
||||||
use std::path::{Path, PathBuf};
|
use std::path::{Path, PathBuf};
|
||||||
use std::sync::{Arc, RwLock};
|
use std::sync::{Arc, RwLock, Weak};
|
||||||
use std::{fmt, result};
|
use std::{fmt, result};
|
||||||
|
|
||||||
use fs2::FileExt;
|
use fs2::FileExt;
|
||||||
@@ -18,10 +18,13 @@ use crate::directory::error::{
|
|||||||
};
|
};
|
||||||
use crate::directory::file_watcher::FileWatcher;
|
use crate::directory::file_watcher::FileWatcher;
|
||||||
use crate::directory::{
|
use crate::directory::{
|
||||||
AntiCallToken, ArcBytes, Directory, DirectoryLock, FileHandle, Lock, OwnedBytes,
|
AntiCallToken, Directory, DirectoryLock, FileHandle, Lock, OwnedBytes, TerminatingWrite,
|
||||||
TerminatingWrite, WatchCallback, WatchHandle, WeakArcBytes, WritePtr,
|
WatchCallback, WatchHandle, WritePtr,
|
||||||
};
|
};
|
||||||
|
|
||||||
|
pub type ArcBytes = Arc<dyn Deref<Target = [u8]> + Send + Sync + 'static>;
|
||||||
|
pub type WeakArcBytes = Weak<dyn Deref<Target = [u8]> + Send + Sync + 'static>;
|
||||||
|
|
||||||
/// Create a default io error given a string.
|
/// Create a default io error given a string.
|
||||||
pub(crate) fn make_io_err(msg: String) -> io::Error {
|
pub(crate) fn make_io_err(msg: String) -> io::Error {
|
||||||
io::Error::new(io::ErrorKind::Other, msg)
|
io::Error::new(io::ErrorKind::Other, msg)
|
||||||
@@ -334,7 +337,7 @@ impl Directory for MmapDirectory {
|
|||||||
Ok(Arc::new(owned_bytes))
|
Ok(Arc::new(owned_bytes))
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Any entry associated to the path in the mmap will be
|
/// Any entry associated with the path in the mmap will be
|
||||||
/// removed before the file is deleted.
|
/// removed before the file is deleted.
|
||||||
fn delete(&self, path: &Path) -> result::Result<(), DeleteError> {
|
fn delete(&self, path: &Path) -> result::Result<(), DeleteError> {
|
||||||
let full_path = self.resolve_path(path);
|
let full_path = self.resolve_path(path);
|
||||||
@@ -472,6 +475,8 @@ mod tests {
|
|||||||
// There are more tests in directory/mod.rs
|
// There are more tests in directory/mod.rs
|
||||||
// The following tests are specific to the MmapDirectory
|
// The following tests are specific to the MmapDirectory
|
||||||
|
|
||||||
|
use std::time::Duration;
|
||||||
|
|
||||||
use common::HasLen;
|
use common::HasLen;
|
||||||
|
|
||||||
use super::*;
|
use super::*;
|
||||||
@@ -610,7 +615,14 @@ mod tests {
|
|||||||
mmap_directory.get_cache_info().mmapped.len()
|
mmap_directory.get_cache_info().mmapped.len()
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
assert!(mmap_directory.get_cache_info().mmapped.is_empty());
|
// This test failed on CI. The last Mmap is dropped from the merging thread so there might
|
||||||
Ok(())
|
// be a race condition indeed.
|
||||||
|
for _ in 0..10 {
|
||||||
|
if mmap_directory.get_cache_info().mmapped.is_empty() {
|
||||||
|
return Ok(());
|
||||||
|
}
|
||||||
|
std::thread::sleep(Duration::from_millis(200));
|
||||||
|
}
|
||||||
|
panic!("The cache still contains information. One of the Mmap has not been dropped.");
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -26,7 +26,6 @@ pub use ownedbytes::OwnedBytes;
|
|||||||
pub(crate) use self::composite_file::{CompositeFile, CompositeWrite};
|
pub(crate) use self::composite_file::{CompositeFile, CompositeWrite};
|
||||||
pub use self::directory::{Directory, DirectoryClone, DirectoryLock};
|
pub use self::directory::{Directory, DirectoryClone, DirectoryLock};
|
||||||
pub use self::directory_lock::{Lock, INDEX_WRITER_LOCK, META_LOCK};
|
pub use self::directory_lock::{Lock, INDEX_WRITER_LOCK, META_LOCK};
|
||||||
pub(crate) use self::file_slice::{ArcBytes, WeakArcBytes};
|
|
||||||
pub use self::file_slice::{FileHandle, FileSlice};
|
pub use self::file_slice::{FileHandle, FileSlice};
|
||||||
pub use self::ram_directory::RamDirectory;
|
pub use self::ram_directory::RamDirectory;
|
||||||
pub use self::watch_event_router::{WatchCallback, WatchCallbackList, WatchHandle};
|
pub use self::watch_event_router::{WatchCallback, WatchCallbackList, WatchHandle};
|
||||||
|
|||||||
@@ -136,6 +136,20 @@ impl RamDirectory {
|
|||||||
Self::default()
|
Self::default()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Deep clones the directory.
|
||||||
|
///
|
||||||
|
/// Ulterior writes on one of the copy
|
||||||
|
/// will not affect the other copy.
|
||||||
|
pub fn deep_clone(&self) -> RamDirectory {
|
||||||
|
let inner_clone = InnerDirectory {
|
||||||
|
fs: self.fs.read().unwrap().fs.clone(),
|
||||||
|
watch_router: Default::default(),
|
||||||
|
};
|
||||||
|
RamDirectory {
|
||||||
|
fs: Arc::new(RwLock::new(inner_clone)),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
/// Returns the sum of the size of the different files
|
/// Returns the sum of the size of the different files
|
||||||
/// in the [`RamDirectory`].
|
/// in the [`RamDirectory`].
|
||||||
pub fn total_mem_usage(&self) -> usize {
|
pub fn total_mem_usage(&self) -> usize {
|
||||||
@@ -256,4 +270,23 @@ mod tests {
|
|||||||
assert_eq!(directory_copy.atomic_read(path_atomic).unwrap(), msg_atomic);
|
assert_eq!(directory_copy.atomic_read(path_atomic).unwrap(), msg_atomic);
|
||||||
assert_eq!(directory_copy.atomic_read(path_seq).unwrap(), msg_seq);
|
assert_eq!(directory_copy.atomic_read(path_seq).unwrap(), msg_seq);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_ram_directory_deep_clone() {
|
||||||
|
let dir = RamDirectory::default();
|
||||||
|
let test = Path::new("test");
|
||||||
|
let test2 = Path::new("test2");
|
||||||
|
dir.atomic_write(test, b"firstwrite").unwrap();
|
||||||
|
let dir_clone = dir.deep_clone();
|
||||||
|
assert_eq!(
|
||||||
|
dir_clone.atomic_read(test).unwrap(),
|
||||||
|
dir.atomic_read(test).unwrap()
|
||||||
|
);
|
||||||
|
dir.atomic_write(test, b"original").unwrap();
|
||||||
|
dir_clone.atomic_write(test, b"clone").unwrap();
|
||||||
|
dir_clone.atomic_write(test2, b"clone2").unwrap();
|
||||||
|
assert_eq!(dir.atomic_read(test).unwrap(), b"original");
|
||||||
|
assert_eq!(&dir_clone.atomic_read(test).unwrap(), b"clone");
|
||||||
|
assert_eq!(&dir_clone.atomic_read(test2).unwrap(), b"clone2");
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,3 +1,4 @@
|
|||||||
|
use std::ops::Range;
|
||||||
use std::sync::Arc;
|
use std::sync::Arc;
|
||||||
|
|
||||||
use fastfield_codecs::Column;
|
use fastfield_codecs::Column;
|
||||||
@@ -31,36 +32,39 @@ impl BytesFastFieldReader {
|
|||||||
Ok(BytesFastFieldReader { idx_reader, values })
|
Ok(BytesFastFieldReader { idx_reader, values })
|
||||||
}
|
}
|
||||||
|
|
||||||
fn range(&self, doc: DocId) -> (usize, usize) {
|
fn range(&self, doc: DocId) -> Range<u64> {
|
||||||
let idx = doc as u64;
|
let idx = doc as u64;
|
||||||
let start = self.idx_reader.get_val(idx) as usize;
|
let start = self.idx_reader.get_val(idx);
|
||||||
let stop = self.idx_reader.get_val(idx + 1) as usize;
|
let end = self.idx_reader.get_val(idx + 1);
|
||||||
(start, stop)
|
start..end
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Returns the bytes associated to the given `doc`
|
/// Returns the bytes associated with the given `doc`
|
||||||
pub fn get_bytes(&self, doc: DocId) -> &[u8] {
|
pub fn get_bytes(&self, doc: DocId) -> &[u8] {
|
||||||
let (start, stop) = self.range(doc);
|
let range = self.range(doc);
|
||||||
&self.values.as_slice()[start..stop]
|
&self.values.as_slice()[range.start as usize..range.end as usize]
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Returns the length of the bytes associated to the given `doc`
|
/// Returns the length of the bytes associated with the given `doc`
|
||||||
pub fn num_bytes(&self, doc: DocId) -> usize {
|
pub fn num_bytes(&self, doc: DocId) -> u64 {
|
||||||
let (start, stop) = self.range(doc);
|
let range = self.range(doc);
|
||||||
stop - start
|
range.end - range.start
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Returns the overall number of bytes in this bytes fast field.
|
/// Returns the overall number of bytes in this bytes fast field.
|
||||||
pub fn total_num_bytes(&self) -> usize {
|
pub fn total_num_bytes(&self) -> u64 {
|
||||||
self.values.len()
|
self.values.len() as u64
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl MultiValueLength for BytesFastFieldReader {
|
impl MultiValueLength for BytesFastFieldReader {
|
||||||
|
fn get_range(&self, doc_id: DocId) -> std::ops::Range<u64> {
|
||||||
|
self.range(doc_id)
|
||||||
|
}
|
||||||
fn get_len(&self, doc_id: DocId) -> u64 {
|
fn get_len(&self, doc_id: DocId) -> u64 {
|
||||||
self.num_bytes(doc_id) as u64
|
self.num_bytes(doc_id)
|
||||||
}
|
}
|
||||||
fn get_total_len(&self) -> u64 {
|
fn get_total_len(&self) -> u64 {
|
||||||
self.total_num_bytes() as u64
|
self.total_num_bytes()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -24,7 +24,7 @@ use crate::DocId;
|
|||||||
///
|
///
|
||||||
/// Once acquired, writing is done by calling
|
/// Once acquired, writing is done by calling
|
||||||
/// [`.add_document_val(&[u8])`](BytesFastFieldWriter::add_document_val)
|
/// [`.add_document_val(&[u8])`](BytesFastFieldWriter::add_document_val)
|
||||||
/// once per document, even if there are no bytes associated to it.
|
/// once per document, even if there are no bytes associated with it.
|
||||||
pub struct BytesFastFieldWriter {
|
pub struct BytesFastFieldWriter {
|
||||||
field: Field,
|
field: Field,
|
||||||
vals: Vec<u8>,
|
vals: Vec<u8>,
|
||||||
@@ -45,7 +45,7 @@ impl BytesFastFieldWriter {
|
|||||||
pub fn mem_usage(&self) -> usize {
|
pub fn mem_usage(&self) -> usize {
|
||||||
self.vals.capacity() + self.doc_index.capacity() * std::mem::size_of::<u64>()
|
self.vals.capacity() + self.doc_index.capacity() * std::mem::size_of::<u64>()
|
||||||
}
|
}
|
||||||
/// Access the field associated to the `BytesFastFieldWriter`
|
/// Access the field associated with the `BytesFastFieldWriter`
|
||||||
pub fn field(&self) -> Field {
|
pub fn field(&self) -> Field {
|
||||||
self.field
|
self.field
|
||||||
}
|
}
|
||||||
@@ -67,7 +67,7 @@ impl BytesFastFieldWriter {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Register the bytes associated to a document.
|
/// Register the bytes associated with a document.
|
||||||
///
|
///
|
||||||
/// The method returns the `DocId` of the document that was
|
/// The method returns the `DocId` of the document that was
|
||||||
/// just written.
|
/// just written.
|
||||||
|
|||||||
@@ -7,7 +7,7 @@ use crate::termdict::{TermDictionary, TermOrdinal};
|
|||||||
use crate::DocId;
|
use crate::DocId;
|
||||||
|
|
||||||
/// The facet reader makes it possible to access the list of
|
/// The facet reader makes it possible to access the list of
|
||||||
/// facets associated to a given document in a specific
|
/// facets associated with a given document in a specific
|
||||||
/// segment.
|
/// segment.
|
||||||
///
|
///
|
||||||
/// Rather than manipulating `Facet` object directly, the API
|
/// Rather than manipulating `Facet` object directly, the API
|
||||||
@@ -58,7 +58,7 @@ impl FacetReader {
|
|||||||
&self.term_dict
|
&self.term_dict
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Given a term ordinal returns the term associated to it.
|
/// Given a term ordinal returns the term associated with it.
|
||||||
pub fn facet_from_ord(
|
pub fn facet_from_ord(
|
||||||
&mut self,
|
&mut self,
|
||||||
facet_ord: TermOrdinal,
|
facet_ord: TermOrdinal,
|
||||||
@@ -74,7 +74,7 @@ impl FacetReader {
|
|||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Return the list of facet ordinals associated to a document.
|
/// Return the list of facet ordinals associated with a document.
|
||||||
pub fn facet_ords(&self, doc: DocId, output: &mut Vec<u64>) {
|
pub fn facet_ords(&self, doc: DocId, output: &mut Vec<u64>) {
|
||||||
self.term_ords.get_vals(doc, output);
|
self.term_ords.get_vals(doc, output);
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -47,7 +47,9 @@ mod writer;
|
|||||||
/// Trait for `BytesFastFieldReader` and `MultiValuedFastFieldReader` to return the length of data
|
/// Trait for `BytesFastFieldReader` and `MultiValuedFastFieldReader` to return the length of data
|
||||||
/// for a doc_id
|
/// for a doc_id
|
||||||
pub trait MultiValueLength {
|
pub trait MultiValueLength {
|
||||||
/// returns the num of values associated to a doc_id
|
/// returns the positions for a docid
|
||||||
|
fn get_range(&self, doc_id: DocId) -> std::ops::Range<u64>;
|
||||||
|
/// returns the num of values associated with a doc_id
|
||||||
fn get_len(&self, doc_id: DocId) -> u64;
|
fn get_len(&self, doc_id: DocId) -> u64;
|
||||||
/// returns the sum of num values for all doc_ids
|
/// returns the sum of num values for all doc_ids
|
||||||
fn get_total_len(&self) -> u64;
|
fn get_total_len(&self) -> u64;
|
||||||
|
|||||||
@@ -1,10 +1,9 @@
|
|||||||
mod multivalue_start_index;
|
|
||||||
mod reader;
|
mod reader;
|
||||||
mod writer;
|
mod writer;
|
||||||
|
|
||||||
pub(crate) use self::multivalue_start_index::MultivalueStartIndex;
|
|
||||||
pub use self::reader::MultiValuedFastFieldReader;
|
pub use self::reader::MultiValuedFastFieldReader;
|
||||||
pub use self::writer::MultiValuedFastFieldWriter;
|
pub use self::writer::MultiValuedFastFieldWriter;
|
||||||
|
pub(crate) use self::writer::MultivalueStartIndex;
|
||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod tests {
|
mod tests {
|
||||||
@@ -403,6 +402,74 @@ mod bench {
|
|||||||
use crate::schema::{Cardinality, NumericOptions, Schema};
|
use crate::schema::{Cardinality, NumericOptions, Schema};
|
||||||
use crate::Document;
|
use crate::Document;
|
||||||
|
|
||||||
|
fn bench_multi_value_ff_merge_opt(
|
||||||
|
num_docs: usize,
|
||||||
|
segments_every_n_docs: usize,
|
||||||
|
merge_policy: impl crate::indexer::MergePolicy + 'static,
|
||||||
|
) {
|
||||||
|
let mut builder = crate::schema::SchemaBuilder::new();
|
||||||
|
|
||||||
|
let fast_multi =
|
||||||
|
crate::schema::NumericOptions::default().set_fast(Cardinality::MultiValues);
|
||||||
|
let multi_field = builder.add_f64_field("f64s", fast_multi);
|
||||||
|
|
||||||
|
let index = crate::Index::create_in_ram(builder.build());
|
||||||
|
|
||||||
|
let mut writer = index.writer_for_tests().unwrap();
|
||||||
|
writer.set_merge_policy(Box::new(merge_policy));
|
||||||
|
|
||||||
|
for i in 0..num_docs {
|
||||||
|
let mut doc = crate::Document::new();
|
||||||
|
doc.add_f64(multi_field, 0.24);
|
||||||
|
doc.add_f64(multi_field, 0.27);
|
||||||
|
doc.add_f64(multi_field, 0.37);
|
||||||
|
if i % 3 == 0 {
|
||||||
|
doc.add_f64(multi_field, 0.44);
|
||||||
|
}
|
||||||
|
|
||||||
|
writer.add_document(doc).unwrap();
|
||||||
|
if i % segments_every_n_docs == 0 {
|
||||||
|
writer.commit().unwrap();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
{
|
||||||
|
writer.wait_merging_threads().unwrap();
|
||||||
|
let mut writer = index.writer_for_tests().unwrap();
|
||||||
|
let segment_ids = index.searchable_segment_ids().unwrap();
|
||||||
|
writer.merge(&segment_ids).wait().unwrap();
|
||||||
|
}
|
||||||
|
|
||||||
|
// If a merging thread fails, we should end up with more
|
||||||
|
// than one segment here
|
||||||
|
assert_eq!(1, index.searchable_segments().unwrap().len());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[bench]
|
||||||
|
fn bench_multi_value_ff_merge_many_segments(b: &mut Bencher) {
|
||||||
|
let num_docs = 100_000;
|
||||||
|
b.iter(|| {
|
||||||
|
bench_multi_value_ff_merge_opt(num_docs, 1_000, crate::indexer::NoMergePolicy);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
#[bench]
|
||||||
|
fn bench_multi_value_ff_merge_many_segments_log_merge(b: &mut Bencher) {
|
||||||
|
let num_docs = 100_000;
|
||||||
|
b.iter(|| {
|
||||||
|
let merge_policy = crate::indexer::LogMergePolicy::default();
|
||||||
|
bench_multi_value_ff_merge_opt(num_docs, 1_000, merge_policy);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
#[bench]
|
||||||
|
fn bench_multi_value_ff_merge_few_segments(b: &mut Bencher) {
|
||||||
|
let num_docs = 100_000;
|
||||||
|
b.iter(|| {
|
||||||
|
bench_multi_value_ff_merge_opt(num_docs, 33_000, crate::indexer::NoMergePolicy);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
fn multi_values(num_docs: usize, vals_per_doc: usize) -> Vec<Vec<u64>> {
|
fn multi_values(num_docs: usize, vals_per_doc: usize) -> Vec<Vec<u64>> {
|
||||||
let mut vals = vec![];
|
let mut vals = vec![];
|
||||||
for _i in 0..num_docs {
|
for _i in 0..num_docs {
|
||||||
|
|||||||
@@ -1,171 +0,0 @@
|
|||||||
use fastfield_codecs::{Column, ColumnReader};
|
|
||||||
|
|
||||||
use crate::indexer::doc_id_mapping::DocIdMapping;
|
|
||||||
|
|
||||||
pub(crate) struct MultivalueStartIndex<'a, C: Column> {
|
|
||||||
column: &'a C,
|
|
||||||
doc_id_map: &'a DocIdMapping,
|
|
||||||
min_value: u64,
|
|
||||||
max_value: u64,
|
|
||||||
}
|
|
||||||
|
|
||||||
struct MultivalueStartIndexReader<'a, C: Column> {
|
|
||||||
seek_head: MultivalueStartIndexIter<'a, C>,
|
|
||||||
seek_next_id: u64,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl<'a, C: Column> MultivalueStartIndexReader<'a, C> {
|
|
||||||
fn new(column: &'a C, doc_id_map: &'a DocIdMapping) -> Self {
|
|
||||||
Self {
|
|
||||||
seek_head: MultivalueStartIndexIter {
|
|
||||||
column,
|
|
||||||
doc_id_map,
|
|
||||||
new_doc_id: 0,
|
|
||||||
offset: 0u64,
|
|
||||||
},
|
|
||||||
seek_next_id: 0u64,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn reset(&mut self) {
|
|
||||||
self.seek_next_id = 0;
|
|
||||||
self.seek_head.new_doc_id = 0;
|
|
||||||
self.seek_head.offset = 0;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl<'a, C: Column> ColumnReader for MultivalueStartIndexReader<'a, C> {
|
|
||||||
fn seek(&mut self, idx: u64) -> u64 {
|
|
||||||
if self.seek_next_id > idx {
|
|
||||||
self.reset();
|
|
||||||
}
|
|
||||||
let to_skip = idx - self.seek_next_id;
|
|
||||||
self.seek_next_id = idx + 1;
|
|
||||||
self.seek_head.nth(to_skip as usize).unwrap()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl<'a, C: Column> MultivalueStartIndex<'a, C> {
|
|
||||||
pub fn new(column: &'a C, doc_id_map: &'a DocIdMapping) -> Self {
|
|
||||||
assert_eq!(column.num_vals(), doc_id_map.num_old_doc_ids() as u64 + 1);
|
|
||||||
let iter = MultivalueStartIndexIter::new(column, doc_id_map);
|
|
||||||
let (min_value, max_value) = tantivy_bitpacker::minmax(iter).unwrap_or((0, 0));
|
|
||||||
MultivalueStartIndex {
|
|
||||||
column,
|
|
||||||
doc_id_map,
|
|
||||||
min_value,
|
|
||||||
max_value,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn specialized_reader(&self) -> MultivalueStartIndexReader<'a, C> {
|
|
||||||
MultivalueStartIndexReader::new(self.column, self.doc_id_map)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
impl<'a, C: Column> Column for MultivalueStartIndex<'a, C> {
|
|
||||||
fn reader(&self) -> Box<dyn ColumnReader + '_> {
|
|
||||||
Box::new(self.specialized_reader())
|
|
||||||
}
|
|
||||||
|
|
||||||
fn get_val(&self, idx: u64) -> u64 {
|
|
||||||
let mut reader = self.specialized_reader();
|
|
||||||
reader.seek(idx)
|
|
||||||
}
|
|
||||||
|
|
||||||
fn min_value(&self) -> u64 {
|
|
||||||
self.min_value
|
|
||||||
}
|
|
||||||
|
|
||||||
fn max_value(&self) -> u64 {
|
|
||||||
self.max_value
|
|
||||||
}
|
|
||||||
|
|
||||||
fn num_vals(&self) -> u64 {
|
|
||||||
(self.doc_id_map.num_new_doc_ids() + 1) as u64
|
|
||||||
}
|
|
||||||
|
|
||||||
fn iter<'b>(&'b self) -> Box<dyn Iterator<Item = u64> + 'b> {
|
|
||||||
Box::new(MultivalueStartIndexIter::new(self.column, self.doc_id_map))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
struct MultivalueStartIndexIter<'a, C: Column> {
|
|
||||||
column: &'a C,
|
|
||||||
doc_id_map: &'a DocIdMapping,
|
|
||||||
new_doc_id: usize,
|
|
||||||
offset: u64,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl<'a, C: Column> MultivalueStartIndexIter<'a, C> {
|
|
||||||
fn new(column: &'a C, doc_id_map: &'a DocIdMapping) -> Self {
|
|
||||||
Self {
|
|
||||||
column,
|
|
||||||
doc_id_map,
|
|
||||||
new_doc_id: 0,
|
|
||||||
offset: 0,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl<'a, C: Column> Iterator for MultivalueStartIndexIter<'a, C> {
|
|
||||||
type Item = u64;
|
|
||||||
|
|
||||||
fn next(&mut self) -> Option<Self::Item> {
|
|
||||||
if self.new_doc_id > self.doc_id_map.num_new_doc_ids() {
|
|
||||||
return None;
|
|
||||||
}
|
|
||||||
let new_doc_id = self.new_doc_id;
|
|
||||||
self.new_doc_id += 1;
|
|
||||||
let start_offset = self.offset;
|
|
||||||
if new_doc_id < self.doc_id_map.num_new_doc_ids() {
|
|
||||||
let old_doc = self.doc_id_map.get_old_doc_id(new_doc_id as u32) as u64;
|
|
||||||
let num_vals_for_doc = self.column.get_val(old_doc + 1) - self.column.get_val(old_doc);
|
|
||||||
self.offset += num_vals_for_doc;
|
|
||||||
}
|
|
||||||
Some(start_offset)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[cfg(test)]
|
|
||||||
mod tests {
|
|
||||||
use fastfield_codecs::VecColumn;
|
|
||||||
|
|
||||||
use super::*;
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_multivalue_start_index() {
|
|
||||||
let doc_id_mapping = DocIdMapping::from_new_id_to_old_id(vec![4, 1, 2]);
|
|
||||||
assert_eq!(doc_id_mapping.num_old_doc_ids(), 5);
|
|
||||||
let col = VecColumn::from(&[0u64, 3, 5, 10, 12, 16][..]);
|
|
||||||
let multivalue_start_index = MultivalueStartIndex::new(
|
|
||||||
&col, // 3, 2, 5, 2, 4
|
|
||||||
&doc_id_mapping,
|
|
||||||
);
|
|
||||||
assert_eq!(multivalue_start_index.num_vals(), 4);
|
|
||||||
assert_eq!(
|
|
||||||
multivalue_start_index.iter().collect::<Vec<u64>>(),
|
|
||||||
vec![0, 4, 6, 11]
|
|
||||||
); // 4, 2, 5
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_multivalue_get_vals() {
|
|
||||||
let doc_id_mapping =
|
|
||||||
DocIdMapping::from_new_id_to_old_id(vec![0, 1, 2, 3, 4, 5, 6, 7, 8, 9]);
|
|
||||||
assert_eq!(doc_id_mapping.num_old_doc_ids(), 10);
|
|
||||||
let col = VecColumn::from(&[0, 1, 1, 2, 3, 5, 8, 13, 21, 34, 55][..]);
|
|
||||||
let multivalue_start_index = MultivalueStartIndex::new(&col, &doc_id_mapping);
|
|
||||||
assert_eq!(
|
|
||||||
multivalue_start_index.iter().collect::<Vec<u64>>(),
|
|
||||||
vec![0, 1, 1, 2, 3, 5, 8, 13, 21, 34, 55]
|
|
||||||
);
|
|
||||||
assert_eq!(multivalue_start_index.num_vals(), 11);
|
|
||||||
let mut multivalue_start_index_reader = multivalue_start_index.reader();
|
|
||||||
assert_eq!(multivalue_start_index_reader.seek(3), 2);
|
|
||||||
assert_eq!(multivalue_start_index_reader.seek(5), 5);
|
|
||||||
assert_eq!(multivalue_start_index_reader.seek(8), 21);
|
|
||||||
assert_eq!(multivalue_start_index_reader.seek(4), 3);
|
|
||||||
assert_eq!(multivalue_start_index_reader.seek(0), 0);
|
|
||||||
assert_eq!(multivalue_start_index_reader.seek(10), 55);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -30,8 +30,8 @@ impl<Item: FastValue> MultiValuedFastFieldReader<Item> {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Returns `[start, end)`, such that the values associated
|
/// Returns `[start, end)`, such that the values associated with
|
||||||
/// to the given document are `start..end`.
|
/// the given document are `start..end`.
|
||||||
#[inline]
|
#[inline]
|
||||||
fn range(&self, doc: DocId) -> Range<u64> {
|
fn range(&self, doc: DocId) -> Range<u64> {
|
||||||
let idx = doc as u64;
|
let idx = doc as u64;
|
||||||
@@ -40,7 +40,7 @@ impl<Item: FastValue> MultiValuedFastFieldReader<Item> {
|
|||||||
start..end
|
start..end
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Returns the array of values associated to the given `doc`.
|
/// Returns the array of values associated with the given `doc`.
|
||||||
#[inline]
|
#[inline]
|
||||||
fn get_vals_for_range(&self, range: Range<u64>, vals: &mut Vec<Item>) {
|
fn get_vals_for_range(&self, range: Range<u64>, vals: &mut Vec<Item>) {
|
||||||
let len = (range.end - range.start) as usize;
|
let len = (range.end - range.start) as usize;
|
||||||
@@ -48,7 +48,7 @@ impl<Item: FastValue> MultiValuedFastFieldReader<Item> {
|
|||||||
self.vals_reader.get_range(range.start, &mut vals[..]);
|
self.vals_reader.get_range(range.start, &mut vals[..]);
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Returns the array of values associated to the given `doc`.
|
/// Returns the array of values associated with the given `doc`.
|
||||||
#[inline]
|
#[inline]
|
||||||
pub fn get_vals(&self, doc: DocId, vals: &mut Vec<Item>) {
|
pub fn get_vals(&self, doc: DocId, vals: &mut Vec<Item>) {
|
||||||
let range = self.range(doc);
|
let range = self.range(doc);
|
||||||
@@ -88,6 +88,9 @@ impl<Item: FastValue> MultiValuedFastFieldReader<Item> {
|
|||||||
}
|
}
|
||||||
|
|
||||||
impl<Item: FastValue> MultiValueLength for MultiValuedFastFieldReader<Item> {
|
impl<Item: FastValue> MultiValueLength for MultiValuedFastFieldReader<Item> {
|
||||||
|
fn get_range(&self, doc_id: DocId) -> Range<u64> {
|
||||||
|
self.range(doc_id)
|
||||||
|
}
|
||||||
fn get_len(&self, doc_id: DocId) -> u64 {
|
fn get_len(&self, doc_id: DocId) -> u64 {
|
||||||
self.num_vals(doc_id) as u64
|
self.num_vals(doc_id) as u64
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,11 +1,10 @@
|
|||||||
use std::io;
|
use std::io;
|
||||||
|
use std::sync::Mutex;
|
||||||
|
|
||||||
use fastfield_codecs::{MonotonicallyMappableToU64, VecColumn};
|
use fastfield_codecs::{Column, MonotonicallyMappableToU64, VecColumn};
|
||||||
use fnv::FnvHashMap;
|
use fnv::FnvHashMap;
|
||||||
|
|
||||||
use crate::fastfield::{
|
use crate::fastfield::{value_to_u64, CompositeFastFieldSerializer, FastFieldType};
|
||||||
value_to_u64, CompositeFastFieldSerializer, FastFieldType, MultivalueStartIndex,
|
|
||||||
};
|
|
||||||
use crate::indexer::doc_id_mapping::DocIdMapping;
|
use crate::indexer::doc_id_mapping::DocIdMapping;
|
||||||
use crate::postings::UnorderedTermId;
|
use crate::postings::UnorderedTermId;
|
||||||
use crate::schema::{Document, Field, Value};
|
use crate::schema::{Document, Field, Value};
|
||||||
@@ -63,7 +62,7 @@ impl MultiValuedFastFieldWriter {
|
|||||||
+ self.doc_index.capacity() * std::mem::size_of::<u64>()
|
+ self.doc_index.capacity() * std::mem::size_of::<u64>()
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Access the field associated to the `MultiValuedFastFieldWriter`
|
/// Access the field associated with the `MultiValuedFastFieldWriter`
|
||||||
pub fn field(&self) -> Field {
|
pub fn field(&self) -> Field {
|
||||||
self.field
|
self.field
|
||||||
}
|
}
|
||||||
@@ -201,3 +200,155 @@ impl MultiValuedFastFieldWriter {
|
|||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub(crate) struct MultivalueStartIndex<'a, C: Column> {
|
||||||
|
column: &'a C,
|
||||||
|
doc_id_map: &'a DocIdMapping,
|
||||||
|
min_max_opt: Mutex<Option<(u64, u64)>>,
|
||||||
|
random_seeker: Mutex<MultivalueStartIndexRandomSeeker<'a, C>>,
|
||||||
|
}
|
||||||
|
|
||||||
|
struct MultivalueStartIndexRandomSeeker<'a, C: Column> {
|
||||||
|
seek_head: MultivalueStartIndexIter<'a, C>,
|
||||||
|
seek_next_id: u64,
|
||||||
|
}
|
||||||
|
impl<'a, C: Column> MultivalueStartIndexRandomSeeker<'a, C> {
|
||||||
|
fn new(column: &'a C, doc_id_map: &'a DocIdMapping) -> Self {
|
||||||
|
Self {
|
||||||
|
seek_head: MultivalueStartIndexIter {
|
||||||
|
column,
|
||||||
|
doc_id_map,
|
||||||
|
new_doc_id: 0,
|
||||||
|
offset: 0u64,
|
||||||
|
},
|
||||||
|
seek_next_id: 0u64,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<'a, C: Column> MultivalueStartIndex<'a, C> {
|
||||||
|
pub fn new(column: &'a C, doc_id_map: &'a DocIdMapping) -> Self {
|
||||||
|
assert_eq!(column.num_vals(), doc_id_map.num_old_doc_ids() as u64 + 1);
|
||||||
|
MultivalueStartIndex {
|
||||||
|
column,
|
||||||
|
doc_id_map,
|
||||||
|
min_max_opt: Mutex::default(),
|
||||||
|
random_seeker: Mutex::new(MultivalueStartIndexRandomSeeker::new(column, doc_id_map)),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn minmax(&self) -> (u64, u64) {
|
||||||
|
if let Some((min, max)) = *self.min_max_opt.lock().unwrap() {
|
||||||
|
return (min, max);
|
||||||
|
}
|
||||||
|
let (min, max) = tantivy_bitpacker::minmax(self.iter()).unwrap_or((0u64, 0u64));
|
||||||
|
*self.min_max_opt.lock().unwrap() = Some((min, max));
|
||||||
|
(min, max)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
impl<'a, C: Column> Column for MultivalueStartIndex<'a, C> {
|
||||||
|
fn get_val(&self, idx: u64) -> u64 {
|
||||||
|
let mut random_seeker_lock = self.random_seeker.lock().unwrap();
|
||||||
|
if random_seeker_lock.seek_next_id > idx {
|
||||||
|
*random_seeker_lock =
|
||||||
|
MultivalueStartIndexRandomSeeker::new(self.column, self.doc_id_map);
|
||||||
|
}
|
||||||
|
let to_skip = idx - random_seeker_lock.seek_next_id;
|
||||||
|
random_seeker_lock.seek_next_id = idx + 1;
|
||||||
|
random_seeker_lock.seek_head.nth(to_skip as usize).unwrap()
|
||||||
|
}
|
||||||
|
|
||||||
|
fn min_value(&self) -> u64 {
|
||||||
|
self.minmax().0
|
||||||
|
}
|
||||||
|
|
||||||
|
fn max_value(&self) -> u64 {
|
||||||
|
self.minmax().1
|
||||||
|
}
|
||||||
|
|
||||||
|
fn num_vals(&self) -> u64 {
|
||||||
|
(self.doc_id_map.num_new_doc_ids() + 1) as u64
|
||||||
|
}
|
||||||
|
|
||||||
|
fn iter<'b>(&'b self) -> Box<dyn Iterator<Item = u64> + 'b> {
|
||||||
|
Box::new(MultivalueStartIndexIter::new(self.column, self.doc_id_map))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
struct MultivalueStartIndexIter<'a, C: Column> {
|
||||||
|
pub column: &'a C,
|
||||||
|
pub doc_id_map: &'a DocIdMapping,
|
||||||
|
pub new_doc_id: usize,
|
||||||
|
pub offset: u64,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<'a, C: Column> MultivalueStartIndexIter<'a, C> {
|
||||||
|
fn new(column: &'a C, doc_id_map: &'a DocIdMapping) -> Self {
|
||||||
|
Self {
|
||||||
|
column,
|
||||||
|
doc_id_map,
|
||||||
|
new_doc_id: 0,
|
||||||
|
offset: 0,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<'a, C: Column> Iterator for MultivalueStartIndexIter<'a, C> {
|
||||||
|
type Item = u64;
|
||||||
|
|
||||||
|
fn next(&mut self) -> Option<Self::Item> {
|
||||||
|
if self.new_doc_id > self.doc_id_map.num_new_doc_ids() {
|
||||||
|
return None;
|
||||||
|
}
|
||||||
|
let new_doc_id = self.new_doc_id;
|
||||||
|
self.new_doc_id += 1;
|
||||||
|
let start_offset = self.offset;
|
||||||
|
if new_doc_id < self.doc_id_map.num_new_doc_ids() {
|
||||||
|
let old_doc = self.doc_id_map.get_old_doc_id(new_doc_id as u32) as u64;
|
||||||
|
let num_vals_for_doc = self.column.get_val(old_doc + 1) - self.column.get_val(old_doc);
|
||||||
|
self.offset += num_vals_for_doc;
|
||||||
|
}
|
||||||
|
Some(start_offset)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use super::*;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_multivalue_start_index() {
|
||||||
|
let doc_id_mapping = DocIdMapping::from_new_id_to_old_id(vec![4, 1, 2]);
|
||||||
|
assert_eq!(doc_id_mapping.num_old_doc_ids(), 5);
|
||||||
|
let col = VecColumn::from(&[0u64, 3, 5, 10, 12, 16][..]);
|
||||||
|
let multivalue_start_index = MultivalueStartIndex::new(
|
||||||
|
&col, // 3, 2, 5, 2, 4
|
||||||
|
&doc_id_mapping,
|
||||||
|
);
|
||||||
|
assert_eq!(multivalue_start_index.num_vals(), 4);
|
||||||
|
assert_eq!(
|
||||||
|
multivalue_start_index.iter().collect::<Vec<u64>>(),
|
||||||
|
vec![0, 4, 6, 11]
|
||||||
|
); // 4, 2, 5
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_multivalue_get_vals() {
|
||||||
|
let doc_id_mapping =
|
||||||
|
DocIdMapping::from_new_id_to_old_id(vec![0, 1, 2, 3, 4, 5, 6, 7, 8, 9]);
|
||||||
|
assert_eq!(doc_id_mapping.num_old_doc_ids(), 10);
|
||||||
|
let col = VecColumn::from(&[0, 1, 1, 2, 3, 5, 8, 13, 21, 34, 55][..]);
|
||||||
|
let multivalue_start_index = MultivalueStartIndex::new(&col, &doc_id_mapping);
|
||||||
|
assert_eq!(
|
||||||
|
multivalue_start_index.iter().collect::<Vec<u64>>(),
|
||||||
|
vec![0, 1, 1, 2, 3, 5, 8, 13, 21, 34, 55]
|
||||||
|
);
|
||||||
|
assert_eq!(multivalue_start_index.num_vals(), 11);
|
||||||
|
assert_eq!(multivalue_start_index.get_val(3), 2);
|
||||||
|
assert_eq!(multivalue_start_index.get_val(5), 5);
|
||||||
|
assert_eq!(multivalue_start_index.get_val(8), 21);
|
||||||
|
assert_eq!(multivalue_start_index.get_val(4), 3);
|
||||||
|
assert_eq!(multivalue_start_index.get_val(0), 0);
|
||||||
|
assert_eq!(multivalue_start_index.get_val(10), 55);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
@@ -135,7 +135,7 @@ impl FastFieldReaders {
|
|||||||
Ok(MultiValuedFastFieldReader::open(idx_reader, vals_reader))
|
Ok(MultiValuedFastFieldReader::open(idx_reader, vals_reader))
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Returns the `u64` fast field reader reader associated to `field`.
|
/// Returns the `u64` fast field reader reader associated with `field`.
|
||||||
///
|
///
|
||||||
/// If `field` is not a u64 fast field, this method returns an Error.
|
/// If `field` is not a u64 fast field, this method returns an Error.
|
||||||
pub fn u64(&self, field: Field) -> crate::Result<Arc<dyn Column<u64>>> {
|
pub fn u64(&self, field: Field) -> crate::Result<Arc<dyn Column<u64>>> {
|
||||||
@@ -143,16 +143,16 @@ impl FastFieldReaders {
|
|||||||
self.typed_fast_field_reader(field)
|
self.typed_fast_field_reader(field)
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Returns the `u64` fast field reader reader associated to `field`, regardless of whether the
|
/// Returns the `u64` fast field reader reader associated with `field`, regardless of whether
|
||||||
/// given field is effectively of type `u64` or not.
|
/// the given field is effectively of type `u64` or not.
|
||||||
///
|
///
|
||||||
/// If not, the fastfield reader will returns the u64-value associated to the original
|
/// If not, the fastfield reader will returns the u64-value associated with the original
|
||||||
/// FastValue.
|
/// FastValue.
|
||||||
pub fn u64_lenient(&self, field: Field) -> crate::Result<Arc<dyn Column<u64>>> {
|
pub fn u64_lenient(&self, field: Field) -> crate::Result<Arc<dyn Column<u64>>> {
|
||||||
self.typed_fast_field_reader(field)
|
self.typed_fast_field_reader(field)
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Returns the `i64` fast field reader reader associated to `field`.
|
/// Returns the `i64` fast field reader reader associated with `field`.
|
||||||
///
|
///
|
||||||
/// If `field` is not a i64 fast field, this method returns an Error.
|
/// If `field` is not a i64 fast field, this method returns an Error.
|
||||||
pub fn i64(&self, field: Field) -> crate::Result<Arc<dyn Column<i64>>> {
|
pub fn i64(&self, field: Field) -> crate::Result<Arc<dyn Column<i64>>> {
|
||||||
@@ -160,7 +160,7 @@ impl FastFieldReaders {
|
|||||||
self.typed_fast_field_reader(field)
|
self.typed_fast_field_reader(field)
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Returns the `date` fast field reader reader associated to `field`.
|
/// Returns the `date` fast field reader reader associated with `field`.
|
||||||
///
|
///
|
||||||
/// If `field` is not a date fast field, this method returns an Error.
|
/// If `field` is not a date fast field, this method returns an Error.
|
||||||
pub fn date(&self, field: Field) -> crate::Result<Arc<dyn Column<DateTime>>> {
|
pub fn date(&self, field: Field) -> crate::Result<Arc<dyn Column<DateTime>>> {
|
||||||
@@ -168,7 +168,7 @@ impl FastFieldReaders {
|
|||||||
self.typed_fast_field_reader(field)
|
self.typed_fast_field_reader(field)
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Returns the `f64` fast field reader reader associated to `field`.
|
/// Returns the `f64` fast field reader reader associated with `field`.
|
||||||
///
|
///
|
||||||
/// If `field` is not a f64 fast field, this method returns an Error.
|
/// If `field` is not a f64 fast field, this method returns an Error.
|
||||||
pub fn f64(&self, field: Field) -> crate::Result<Arc<dyn Column<f64>>> {
|
pub fn f64(&self, field: Field) -> crate::Result<Arc<dyn Column<f64>>> {
|
||||||
@@ -176,7 +176,7 @@ impl FastFieldReaders {
|
|||||||
self.typed_fast_field_reader(field)
|
self.typed_fast_field_reader(field)
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Returns the `bool` fast field reader reader associated to `field`.
|
/// Returns the `bool` fast field reader reader associated with `field`.
|
||||||
///
|
///
|
||||||
/// If `field` is not a bool fast field, this method returns an Error.
|
/// If `field` is not a bool fast field, this method returns an Error.
|
||||||
pub fn bool(&self, field: Field) -> crate::Result<Arc<dyn Column<bool>>> {
|
pub fn bool(&self, field: Field) -> crate::Result<Arc<dyn Column<bool>>> {
|
||||||
@@ -184,7 +184,7 @@ impl FastFieldReaders {
|
|||||||
self.typed_fast_field_reader(field)
|
self.typed_fast_field_reader(field)
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Returns a `u64s` multi-valued fast field reader reader associated to `field`.
|
/// Returns a `u64s` multi-valued fast field reader reader associated with `field`.
|
||||||
///
|
///
|
||||||
/// If `field` is not a u64 multi-valued fast field, this method returns an Error.
|
/// If `field` is not a u64 multi-valued fast field, this method returns an Error.
|
||||||
pub fn u64s(&self, field: Field) -> crate::Result<MultiValuedFastFieldReader<u64>> {
|
pub fn u64s(&self, field: Field) -> crate::Result<MultiValuedFastFieldReader<u64>> {
|
||||||
@@ -192,15 +192,15 @@ impl FastFieldReaders {
|
|||||||
self.typed_fast_field_multi_reader(field)
|
self.typed_fast_field_multi_reader(field)
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Returns a `u64s` multi-valued fast field reader reader associated to `field`, regardless of
|
/// Returns a `u64s` multi-valued fast field reader reader associated with `field`, regardless
|
||||||
/// whether the given field is effectively of type `u64` or not.
|
/// of whether the given field is effectively of type `u64` or not.
|
||||||
///
|
///
|
||||||
/// If `field` is not a u64 multi-valued fast field, this method returns an Error.
|
/// If `field` is not a u64 multi-valued fast field, this method returns an Error.
|
||||||
pub fn u64s_lenient(&self, field: Field) -> crate::Result<MultiValuedFastFieldReader<u64>> {
|
pub fn u64s_lenient(&self, field: Field) -> crate::Result<MultiValuedFastFieldReader<u64>> {
|
||||||
self.typed_fast_field_multi_reader(field)
|
self.typed_fast_field_multi_reader(field)
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Returns a `i64s` multi-valued fast field reader reader associated to `field`.
|
/// Returns a `i64s` multi-valued fast field reader reader associated with `field`.
|
||||||
///
|
///
|
||||||
/// If `field` is not a i64 multi-valued fast field, this method returns an Error.
|
/// If `field` is not a i64 multi-valued fast field, this method returns an Error.
|
||||||
pub fn i64s(&self, field: Field) -> crate::Result<MultiValuedFastFieldReader<i64>> {
|
pub fn i64s(&self, field: Field) -> crate::Result<MultiValuedFastFieldReader<i64>> {
|
||||||
@@ -208,7 +208,7 @@ impl FastFieldReaders {
|
|||||||
self.typed_fast_field_multi_reader(field)
|
self.typed_fast_field_multi_reader(field)
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Returns a `f64s` multi-valued fast field reader reader associated to `field`.
|
/// Returns a `f64s` multi-valued fast field reader reader associated with `field`.
|
||||||
///
|
///
|
||||||
/// If `field` is not a f64 multi-valued fast field, this method returns an Error.
|
/// If `field` is not a f64 multi-valued fast field, this method returns an Error.
|
||||||
pub fn f64s(&self, field: Field) -> crate::Result<MultiValuedFastFieldReader<f64>> {
|
pub fn f64s(&self, field: Field) -> crate::Result<MultiValuedFastFieldReader<f64>> {
|
||||||
@@ -216,7 +216,7 @@ impl FastFieldReaders {
|
|||||||
self.typed_fast_field_multi_reader(field)
|
self.typed_fast_field_multi_reader(field)
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Returns a `bools` multi-valued fast field reader reader associated to `field`.
|
/// Returns a `bools` multi-valued fast field reader reader associated with `field`.
|
||||||
///
|
///
|
||||||
/// If `field` is not a bool multi-valued fast field, this method returns an Error.
|
/// If `field` is not a bool multi-valued fast field, this method returns an Error.
|
||||||
pub fn bools(&self, field: Field) -> crate::Result<MultiValuedFastFieldReader<bool>> {
|
pub fn bools(&self, field: Field) -> crate::Result<MultiValuedFastFieldReader<bool>> {
|
||||||
@@ -224,7 +224,7 @@ impl FastFieldReaders {
|
|||||||
self.typed_fast_field_multi_reader(field)
|
self.typed_fast_field_multi_reader(field)
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Returns a `time::OffsetDateTime` multi-valued fast field reader reader associated to
|
/// Returns a `time::OffsetDateTime` multi-valued fast field reader reader associated with
|
||||||
/// `field`.
|
/// `field`.
|
||||||
///
|
///
|
||||||
/// If `field` is not a `time::OffsetDateTime` multi-valued fast field, this method returns an
|
/// If `field` is not a `time::OffsetDateTime` multi-valued fast field, this method returns an
|
||||||
@@ -234,7 +234,7 @@ impl FastFieldReaders {
|
|||||||
self.typed_fast_field_multi_reader(field)
|
self.typed_fast_field_multi_reader(field)
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Returns the `bytes` fast field reader associated to `field`.
|
/// Returns the `bytes` fast field reader associated with `field`.
|
||||||
///
|
///
|
||||||
/// If `field` is not a bytes fast field, returns an Error.
|
/// If `field` is not a bytes fast field, returns an Error.
|
||||||
pub fn bytes(&self, field: Field) -> crate::Result<BytesFastFieldReader> {
|
pub fn bytes(&self, field: Field) -> crate::Result<BytesFastFieldReader> {
|
||||||
|
|||||||
@@ -131,7 +131,7 @@ impl FastFieldsWriter {
|
|||||||
.sum::<usize>()
|
.sum::<usize>()
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Get the `FastFieldWriter` associated to a field.
|
/// Get the `FastFieldWriter` associated with a field.
|
||||||
pub fn get_term_id_writer(&self, field: Field) -> Option<&MultiValuedFastFieldWriter> {
|
pub fn get_term_id_writer(&self, field: Field) -> Option<&MultiValuedFastFieldWriter> {
|
||||||
// TODO optimize
|
// TODO optimize
|
||||||
self.term_id_writers
|
self.term_id_writers
|
||||||
@@ -139,7 +139,7 @@ impl FastFieldsWriter {
|
|||||||
.find(|field_writer| field_writer.field() == field)
|
.find(|field_writer| field_writer.field() == field)
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Get the `FastFieldWriter` associated to a field.
|
/// Get the `FastFieldWriter` associated with a field.
|
||||||
pub fn get_field_writer(&self, field: Field) -> Option<&IntFastFieldWriter> {
|
pub fn get_field_writer(&self, field: Field) -> Option<&IntFastFieldWriter> {
|
||||||
// TODO optimize
|
// TODO optimize
|
||||||
self.single_value_writers
|
self.single_value_writers
|
||||||
@@ -147,7 +147,7 @@ impl FastFieldsWriter {
|
|||||||
.find(|field_writer| field_writer.field() == field)
|
.find(|field_writer| field_writer.field() == field)
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Get the `FastFieldWriter` associated to a field.
|
/// Get the `FastFieldWriter` associated with a field.
|
||||||
pub fn get_field_writer_mut(&mut self, field: Field) -> Option<&mut IntFastFieldWriter> {
|
pub fn get_field_writer_mut(&mut self, field: Field) -> Option<&mut IntFastFieldWriter> {
|
||||||
// TODO optimize
|
// TODO optimize
|
||||||
self.single_value_writers
|
self.single_value_writers
|
||||||
@@ -155,7 +155,7 @@ impl FastFieldsWriter {
|
|||||||
.find(|field_writer| field_writer.field() == field)
|
.find(|field_writer| field_writer.field() == field)
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Get the `FastFieldWriter` associated to a field.
|
/// Get the `FastFieldWriter` associated with a field.
|
||||||
pub fn get_term_id_writer_mut(
|
pub fn get_term_id_writer_mut(
|
||||||
&mut self,
|
&mut self,
|
||||||
field: Field,
|
field: Field,
|
||||||
@@ -294,7 +294,7 @@ impl IntFastFieldWriter {
|
|||||||
/// Records a new value.
|
/// Records a new value.
|
||||||
///
|
///
|
||||||
/// The n-th value being recorded is implicitly
|
/// The n-th value being recorded is implicitly
|
||||||
/// associated to the document with the `DocId` n.
|
/// associated with the document with the `DocId` n.
|
||||||
/// (Well, `n-1` actually because of 0-indexing)
|
/// (Well, `n-1` actually because of 0-indexing)
|
||||||
pub fn add_val(&mut self, val: u64) {
|
pub fn add_val(&mut self, val: u64) {
|
||||||
self.vals.add(val);
|
self.vals.add(val);
|
||||||
@@ -313,7 +313,7 @@ impl IntFastFieldWriter {
|
|||||||
/// (or use the default value) and records it.
|
/// (or use the default value) and records it.
|
||||||
///
|
///
|
||||||
///
|
///
|
||||||
/// Extract the value associated to the fast field for
|
/// Extract the value associated with the fast field for
|
||||||
/// this document.
|
/// this document.
|
||||||
///
|
///
|
||||||
/// i64 and f64 are remapped to u64 using the logic
|
/// i64 and f64 are remapped to u64 using the logic
|
||||||
@@ -383,7 +383,7 @@ struct WriterFastFieldAccessProvider<'map, 'bitp> {
|
|||||||
}
|
}
|
||||||
|
|
||||||
impl<'map, 'bitp> Column for WriterFastFieldAccessProvider<'map, 'bitp> {
|
impl<'map, 'bitp> Column for WriterFastFieldAccessProvider<'map, 'bitp> {
|
||||||
/// Return the value associated to the given doc.
|
/// Return the value associated with the given doc.
|
||||||
///
|
///
|
||||||
/// Whenever possible use the Iterator passed to the fastfield creation instead, for performance
|
/// Whenever possible use the Iterator passed to the fastfield creation instead, for performance
|
||||||
/// reasons.
|
/// reasons.
|
||||||
|
|||||||
@@ -1,4 +1,4 @@
|
|||||||
//! The fieldnorm represents the length associated to
|
//! The fieldnorm represents the length associated with
|
||||||
//! a given Field of a given document.
|
//! a given Field of a given document.
|
||||||
//!
|
//!
|
||||||
//! This metric is important to compute the score of a
|
//! This metric is important to compute the score of a
|
||||||
|
|||||||
@@ -47,9 +47,9 @@ impl FieldNormReaders {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Reads the fieldnorm associated to a document.
|
/// Reads the fieldnorm associated with a document.
|
||||||
///
|
///
|
||||||
/// The [fieldnorm](FieldNormReader::fieldnorm) represents the length associated to
|
/// The [fieldnorm](FieldNormReader::fieldnorm) represents the length associated with
|
||||||
/// a given Field of a given document.
|
/// a given Field of a given document.
|
||||||
#[derive(Clone)]
|
#[derive(Clone)]
|
||||||
pub struct FieldNormReader(ReaderImplEnum);
|
pub struct FieldNormReader(ReaderImplEnum);
|
||||||
@@ -104,7 +104,7 @@ impl FieldNormReader {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Returns the `fieldnorm` associated to a doc id.
|
/// Returns the `fieldnorm` associated with a doc id.
|
||||||
/// The fieldnorm is a value approximating the number
|
/// The fieldnorm is a value approximating the number
|
||||||
/// of tokens in a given field of the `doc_id`.
|
/// of tokens in a given field of the `doc_id`.
|
||||||
///
|
///
|
||||||
@@ -123,7 +123,7 @@ impl FieldNormReader {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Returns the `fieldnorm_id` associated to a document.
|
/// Returns the `fieldnorm_id` associated with a document.
|
||||||
#[inline]
|
#[inline]
|
||||||
pub fn fieldnorm_id(&self, doc_id: DocId) -> u8 {
|
pub fn fieldnorm_id(&self, doc_id: DocId) -> u8 {
|
||||||
match &self.0 {
|
match &self.0 {
|
||||||
|
|||||||
@@ -188,7 +188,7 @@ impl DeleteCursor {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[cfg_attr(feature = "cargo-clippy", allow(clippy::wrong_self_convention))]
|
#[allow(clippy::wrong_self_convention)]
|
||||||
fn is_behind_opstamp(&mut self, target_opstamp: Opstamp) -> bool {
|
fn is_behind_opstamp(&mut self, target_opstamp: Opstamp) -> bool {
|
||||||
self.get()
|
self.get()
|
||||||
.map(|operation| operation.opstamp < target_opstamp)
|
.map(|operation| operation.opstamp < target_opstamp)
|
||||||
@@ -246,18 +246,27 @@ impl DeleteCursor {
|
|||||||
mod tests {
|
mod tests {
|
||||||
|
|
||||||
use super::{DeleteOperation, DeleteQueue};
|
use super::{DeleteOperation, DeleteQueue};
|
||||||
use crate::schema::{Field, Term};
|
use crate::query::{Explanation, Scorer, Weight};
|
||||||
|
use crate::{DocId, Score, SegmentReader};
|
||||||
|
|
||||||
|
struct DummyWeight;
|
||||||
|
impl Weight for DummyWeight {
|
||||||
|
fn scorer(&self, _reader: &SegmentReader, _boost: Score) -> crate::Result<Box<dyn Scorer>> {
|
||||||
|
Err(crate::TantivyError::InternalError("dummy impl".to_owned()))
|
||||||
|
}
|
||||||
|
|
||||||
|
fn explain(&self, _reader: &SegmentReader, _doc: DocId) -> crate::Result<Explanation> {
|
||||||
|
Err(crate::TantivyError::InternalError("dummy impl".to_owned()))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_deletequeue() {
|
fn test_deletequeue() {
|
||||||
let delete_queue = DeleteQueue::new();
|
let delete_queue = DeleteQueue::new();
|
||||||
|
|
||||||
let make_op = |i: usize| {
|
let make_op = |i: usize| DeleteOperation {
|
||||||
let field = Field::from_field_id(1u32);
|
opstamp: i as u64,
|
||||||
DeleteOperation {
|
target: Box::new(DummyWeight),
|
||||||
opstamp: i as u64,
|
|
||||||
term: Term::from_field_u64(field, i as u64),
|
|
||||||
}
|
|
||||||
};
|
};
|
||||||
|
|
||||||
delete_queue.push(make_op(1));
|
delete_queue.push(make_op(1));
|
||||||
|
|||||||
@@ -24,7 +24,7 @@ impl SegmentDocIdMapping {
|
|||||||
|
|
||||||
/// Returns an iterator over the old document addresses, ordered by the new document ids.
|
/// Returns an iterator over the old document addresses, ordered by the new document ids.
|
||||||
///
|
///
|
||||||
/// In the returned `DocAddress`, the `segment_ord` is the ordinal of targetted segment
|
/// In the returned `DocAddress`, the `segment_ord` is the ordinal of targeted segment
|
||||||
/// in the list of merged segments.
|
/// in the list of merged segments.
|
||||||
pub(crate) fn iter_old_doc_addrs(&self) -> impl Iterator<Item = DocAddress> + '_ {
|
pub(crate) fn iter_old_doc_addrs(&self) -> impl Iterator<Item = DocAddress> + '_ {
|
||||||
self.new_doc_id_to_old_doc_addr.iter().copied()
|
self.new_doc_id_to_old_doc_addr.iter().copied()
|
||||||
|
|||||||
@@ -11,7 +11,6 @@ use super::segment_updater::SegmentUpdater;
|
|||||||
use super::{AddBatch, AddBatchReceiver, AddBatchSender, PreparedCommit};
|
use super::{AddBatch, AddBatchReceiver, AddBatchSender, PreparedCommit};
|
||||||
use crate::core::{Index, Segment, SegmentComponent, SegmentId, SegmentMeta, SegmentReader};
|
use crate::core::{Index, Segment, SegmentComponent, SegmentId, SegmentMeta, SegmentReader};
|
||||||
use crate::directory::{DirectoryLock, GarbageCollectionResult, TerminatingWrite};
|
use crate::directory::{DirectoryLock, GarbageCollectionResult, TerminatingWrite};
|
||||||
use crate::docset::{DocSet, TERMINATED};
|
|
||||||
use crate::error::TantivyError;
|
use crate::error::TantivyError;
|
||||||
use crate::fastfield::write_alive_bitset;
|
use crate::fastfield::write_alive_bitset;
|
||||||
use crate::indexer::delete_queue::{DeleteCursor, DeleteQueue};
|
use crate::indexer::delete_queue::{DeleteCursor, DeleteQueue};
|
||||||
@@ -20,8 +19,9 @@ use crate::indexer::index_writer_status::IndexWriterStatus;
|
|||||||
use crate::indexer::operation::DeleteOperation;
|
use crate::indexer::operation::DeleteOperation;
|
||||||
use crate::indexer::stamper::Stamper;
|
use crate::indexer::stamper::Stamper;
|
||||||
use crate::indexer::{MergePolicy, SegmentEntry, SegmentWriter};
|
use crate::indexer::{MergePolicy, SegmentEntry, SegmentWriter};
|
||||||
|
use crate::query::{Query, TermQuery};
|
||||||
use crate::schema::{Document, IndexRecordOption, Term};
|
use crate::schema::{Document, IndexRecordOption, Term};
|
||||||
use crate::{FutureResult, Opstamp};
|
use crate::{FutureResult, IndexReader, Opstamp};
|
||||||
|
|
||||||
// Size of the margin for the `memory_arena`. A segment is closed when the remaining memory
|
// Size of the margin for the `memory_arena`. A segment is closed when the remaining memory
|
||||||
// in the `memory_arena` goes below MARGIN_IN_BYTES.
|
// in the `memory_arena` goes below MARGIN_IN_BYTES.
|
||||||
@@ -57,6 +57,7 @@ pub struct IndexWriter {
|
|||||||
_directory_lock: Option<DirectoryLock>,
|
_directory_lock: Option<DirectoryLock>,
|
||||||
|
|
||||||
index: Index,
|
index: Index,
|
||||||
|
index_reader: IndexReader,
|
||||||
|
|
||||||
memory_arena_in_bytes_per_thread: usize,
|
memory_arena_in_bytes_per_thread: usize,
|
||||||
|
|
||||||
@@ -92,19 +93,14 @@ fn compute_deleted_bitset(
|
|||||||
|
|
||||||
// A delete operation should only affect
|
// A delete operation should only affect
|
||||||
// document that were inserted before it.
|
// document that were inserted before it.
|
||||||
let inverted_index = segment_reader.inverted_index(delete_op.term.field())?;
|
delete_op
|
||||||
if let Some(mut docset) =
|
.target
|
||||||
inverted_index.read_postings(&delete_op.term, IndexRecordOption::Basic)?
|
.for_each(segment_reader, &mut |doc_matching_delete_query, _| {
|
||||||
{
|
if doc_opstamps.is_deleted(doc_matching_delete_query, delete_op.opstamp) {
|
||||||
let mut doc_matching_deleted_term = docset.doc();
|
alive_bitset.remove(doc_matching_delete_query);
|
||||||
while doc_matching_deleted_term != TERMINATED {
|
|
||||||
if doc_opstamps.is_deleted(doc_matching_deleted_term, delete_op.opstamp) {
|
|
||||||
alive_bitset.remove(doc_matching_deleted_term);
|
|
||||||
might_have_changed = true;
|
might_have_changed = true;
|
||||||
}
|
}
|
||||||
doc_matching_deleted_term = docset.advance();
|
})?;
|
||||||
}
|
|
||||||
}
|
|
||||||
delete_cursor.advance();
|
delete_cursor.advance();
|
||||||
}
|
}
|
||||||
Ok(might_have_changed)
|
Ok(might_have_changed)
|
||||||
@@ -302,6 +298,7 @@ impl IndexWriter {
|
|||||||
|
|
||||||
memory_arena_in_bytes_per_thread,
|
memory_arena_in_bytes_per_thread,
|
||||||
index: index.clone(),
|
index: index.clone(),
|
||||||
|
index_reader: index.reader()?,
|
||||||
|
|
||||||
index_writer_status: IndexWriterStatus::from(document_receiver),
|
index_writer_status: IndexWriterStatus::from(document_receiver),
|
||||||
operation_sender: document_sender,
|
operation_sender: document_sender,
|
||||||
@@ -373,9 +370,9 @@ impl IndexWriter {
|
|||||||
/// This method is useful only for users trying to do complex
|
/// This method is useful only for users trying to do complex
|
||||||
/// operations, like converting an index format to another.
|
/// operations, like converting an index format to another.
|
||||||
///
|
///
|
||||||
/// It is safe to start writing file associated to the new `Segment`.
|
/// It is safe to start writing file associated with the new `Segment`.
|
||||||
/// These will not be garbage collected as long as an instance object of
|
/// These will not be garbage collected as long as an instance object of
|
||||||
/// `SegmentMeta` object associated to the new `Segment` is "alive".
|
/// `SegmentMeta` object associated with the new `Segment` is "alive".
|
||||||
pub fn new_segment(&self) -> Segment {
|
pub fn new_segment(&self) -> Segment {
|
||||||
self.index.new_segment()
|
self.index.new_segment()
|
||||||
}
|
}
|
||||||
@@ -666,10 +663,33 @@ impl IndexWriter {
|
|||||||
/// Like adds, the deletion itself will be visible
|
/// Like adds, the deletion itself will be visible
|
||||||
/// only after calling `commit()`.
|
/// only after calling `commit()`.
|
||||||
pub fn delete_term(&self, term: Term) -> Opstamp {
|
pub fn delete_term(&self, term: Term) -> Opstamp {
|
||||||
|
let query = TermQuery::new(term, IndexRecordOption::Basic);
|
||||||
|
// For backward compatibility, if Term is invalid for the index, do nothing but return an
|
||||||
|
// Opstamp
|
||||||
|
self.delete_query(Box::new(query))
|
||||||
|
.unwrap_or_else(|_| self.stamper.stamp())
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Delete all documents matching a given query.
|
||||||
|
/// Returns an `Err` if the query can't be executed.
|
||||||
|
///
|
||||||
|
/// Delete operation only affects documents that
|
||||||
|
/// were added in previous commits, and documents
|
||||||
|
/// that were added previously in the same commit.
|
||||||
|
///
|
||||||
|
/// Like adds, the deletion itself will be visible
|
||||||
|
/// only after calling `commit()`.
|
||||||
|
#[doc(hidden)]
|
||||||
|
pub fn delete_query(&self, query: Box<dyn Query>) -> crate::Result<Opstamp> {
|
||||||
|
let weight = query.weight(&self.index_reader.searcher(), false)?;
|
||||||
|
|
||||||
let opstamp = self.stamper.stamp();
|
let opstamp = self.stamper.stamp();
|
||||||
let delete_operation = DeleteOperation { opstamp, term };
|
let delete_operation = DeleteOperation {
|
||||||
|
opstamp,
|
||||||
|
target: weight,
|
||||||
|
};
|
||||||
self.delete_queue.push(delete_operation);
|
self.delete_queue.push(delete_operation);
|
||||||
opstamp
|
Ok(opstamp)
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Returns the opstamp of the last successful commit.
|
/// Returns the opstamp of the last successful commit.
|
||||||
@@ -738,10 +758,17 @@ impl IndexWriter {
|
|||||||
let (batch_opstamp, stamps) = self.get_batch_opstamps(count);
|
let (batch_opstamp, stamps) = self.get_batch_opstamps(count);
|
||||||
|
|
||||||
let mut adds = AddBatch::default();
|
let mut adds = AddBatch::default();
|
||||||
|
|
||||||
for (user_op, opstamp) in user_operations_it.zip(stamps) {
|
for (user_op, opstamp) in user_operations_it.zip(stamps) {
|
||||||
match user_op {
|
match user_op {
|
||||||
UserOperation::Delete(term) => {
|
UserOperation::Delete(term) => {
|
||||||
let delete_operation = DeleteOperation { opstamp, term };
|
let query = TermQuery::new(term, IndexRecordOption::Basic);
|
||||||
|
let weight = query.weight(&self.index_reader.searcher(), false)?;
|
||||||
|
|
||||||
|
let delete_operation = DeleteOperation {
|
||||||
|
opstamp,
|
||||||
|
target: weight,
|
||||||
|
};
|
||||||
self.delete_queue.push(delete_operation);
|
self.delete_queue.push(delete_operation);
|
||||||
}
|
}
|
||||||
UserOperation::Add(document) => {
|
UserOperation::Add(document) => {
|
||||||
@@ -786,7 +813,7 @@ mod tests {
|
|||||||
use crate::directory::error::LockError;
|
use crate::directory::error::LockError;
|
||||||
use crate::error::*;
|
use crate::error::*;
|
||||||
use crate::indexer::NoMergePolicy;
|
use crate::indexer::NoMergePolicy;
|
||||||
use crate::query::{QueryParser, TermQuery};
|
use crate::query::{BooleanQuery, Occur, Query, QueryParser, TermQuery};
|
||||||
use crate::schema::{
|
use crate::schema::{
|
||||||
self, Cardinality, Facet, FacetOptions, IndexRecordOption, NumericOptions,
|
self, Cardinality, Facet, FacetOptions, IndexRecordOption, NumericOptions,
|
||||||
TextFieldIndexing, TextOptions, FAST, INDEXED, STORED, STRING, TEXT,
|
TextFieldIndexing, TextOptions, FAST, INDEXED, STORED, STRING, TEXT,
|
||||||
@@ -1418,10 +1445,72 @@ mod tests {
|
|||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_delete_query_with_sort_by_field() -> crate::Result<()> {
|
||||||
|
let mut schema_builder = schema::Schema::builder();
|
||||||
|
let id_field =
|
||||||
|
schema_builder.add_u64_field("id", schema::INDEXED | schema::STORED | schema::FAST);
|
||||||
|
let schema = schema_builder.build();
|
||||||
|
|
||||||
|
let settings = IndexSettings {
|
||||||
|
sort_by_field: Some(IndexSortByField {
|
||||||
|
field: "id".to_string(),
|
||||||
|
order: Order::Desc,
|
||||||
|
}),
|
||||||
|
..Default::default()
|
||||||
|
};
|
||||||
|
|
||||||
|
let index = Index::builder()
|
||||||
|
.schema(schema)
|
||||||
|
.settings(settings)
|
||||||
|
.create_in_ram()?;
|
||||||
|
let index_reader = index.reader()?;
|
||||||
|
let mut index_writer = index.writer_for_tests()?;
|
||||||
|
|
||||||
|
// create and delete docs in same commit
|
||||||
|
for id in 0u64..5u64 {
|
||||||
|
index_writer.add_document(doc!(id_field => id))?;
|
||||||
|
}
|
||||||
|
for id in 1u64..4u64 {
|
||||||
|
let term = Term::from_field_u64(id_field, id);
|
||||||
|
let not_term = Term::from_field_u64(id_field, 2);
|
||||||
|
let term = Box::new(TermQuery::new(term, Default::default()));
|
||||||
|
let not_term = Box::new(TermQuery::new(not_term, Default::default()));
|
||||||
|
|
||||||
|
let query: BooleanQuery = vec![
|
||||||
|
(Occur::Must, term as Box<dyn Query>),
|
||||||
|
(Occur::MustNot, not_term as Box<dyn Query>),
|
||||||
|
]
|
||||||
|
.into();
|
||||||
|
|
||||||
|
index_writer.delete_query(Box::new(query))?;
|
||||||
|
}
|
||||||
|
for id in 5u64..10u64 {
|
||||||
|
index_writer.add_document(doc!(id_field => id))?;
|
||||||
|
}
|
||||||
|
index_writer.commit()?;
|
||||||
|
index_reader.reload()?;
|
||||||
|
|
||||||
|
let searcher = index_reader.searcher();
|
||||||
|
assert_eq!(searcher.segment_readers().len(), 1);
|
||||||
|
|
||||||
|
let segment_reader = searcher.segment_reader(0);
|
||||||
|
assert_eq!(segment_reader.num_docs(), 8);
|
||||||
|
assert_eq!(segment_reader.max_doc(), 10);
|
||||||
|
let fast_field_reader = segment_reader.fast_fields().u64(id_field)?;
|
||||||
|
let in_order_alive_ids: Vec<u64> = segment_reader
|
||||||
|
.doc_ids_alive()
|
||||||
|
.map(|doc| fast_field_reader.get_val(doc as u64))
|
||||||
|
.collect();
|
||||||
|
assert_eq!(&in_order_alive_ids[..], &[9, 8, 7, 6, 5, 4, 2, 0]);
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
#[derive(Debug, Clone, Copy)]
|
#[derive(Debug, Clone, Copy)]
|
||||||
enum IndexingOp {
|
enum IndexingOp {
|
||||||
AddDoc { id: u64 },
|
AddDoc { id: u64 },
|
||||||
DeleteDoc { id: u64 },
|
DeleteDoc { id: u64 },
|
||||||
|
DeleteDocQuery { id: u64 },
|
||||||
Commit,
|
Commit,
|
||||||
Merge,
|
Merge,
|
||||||
}
|
}
|
||||||
@@ -1429,6 +1518,7 @@ mod tests {
|
|||||||
fn balanced_operation_strategy() -> impl Strategy<Value = IndexingOp> {
|
fn balanced_operation_strategy() -> impl Strategy<Value = IndexingOp> {
|
||||||
prop_oneof![
|
prop_oneof![
|
||||||
(0u64..20u64).prop_map(|id| IndexingOp::DeleteDoc { id }),
|
(0u64..20u64).prop_map(|id| IndexingOp::DeleteDoc { id }),
|
||||||
|
(0u64..20u64).prop_map(|id| IndexingOp::DeleteDocQuery { id }),
|
||||||
(0u64..20u64).prop_map(|id| IndexingOp::AddDoc { id }),
|
(0u64..20u64).prop_map(|id| IndexingOp::AddDoc { id }),
|
||||||
(0u64..1u64).prop_map(|_| IndexingOp::Commit),
|
(0u64..1u64).prop_map(|_| IndexingOp::Commit),
|
||||||
(0u64..1u64).prop_map(|_| IndexingOp::Merge),
|
(0u64..1u64).prop_map(|_| IndexingOp::Merge),
|
||||||
@@ -1437,7 +1527,8 @@ mod tests {
|
|||||||
|
|
||||||
fn adding_operation_strategy() -> impl Strategy<Value = IndexingOp> {
|
fn adding_operation_strategy() -> impl Strategy<Value = IndexingOp> {
|
||||||
prop_oneof![
|
prop_oneof![
|
||||||
10 => (0u64..100u64).prop_map(|id| IndexingOp::DeleteDoc { id }),
|
5 => (0u64..100u64).prop_map(|id| IndexingOp::DeleteDoc { id }),
|
||||||
|
5 => (0u64..100u64).prop_map(|id| IndexingOp::DeleteDocQuery { id }),
|
||||||
50 => (0u64..100u64).prop_map(|id| IndexingOp::AddDoc { id }),
|
50 => (0u64..100u64).prop_map(|id| IndexingOp::AddDoc { id }),
|
||||||
2 => (0u64..1u64).prop_map(|_| IndexingOp::Commit),
|
2 => (0u64..1u64).prop_map(|_| IndexingOp::Commit),
|
||||||
1 => (0u64..1u64).prop_map(|_| IndexingOp::Merge),
|
1 => (0u64..1u64).prop_map(|_| IndexingOp::Merge),
|
||||||
@@ -1457,6 +1548,10 @@ mod tests {
|
|||||||
existing_ids.remove(&id);
|
existing_ids.remove(&id);
|
||||||
deleted_ids.insert(id);
|
deleted_ids.insert(id);
|
||||||
}
|
}
|
||||||
|
IndexingOp::DeleteDocQuery { id } => {
|
||||||
|
existing_ids.remove(&id);
|
||||||
|
deleted_ids.insert(id);
|
||||||
|
}
|
||||||
_ => {}
|
_ => {}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -1539,6 +1634,11 @@ mod tests {
|
|||||||
IndexingOp::DeleteDoc { id } => {
|
IndexingOp::DeleteDoc { id } => {
|
||||||
index_writer.delete_term(Term::from_field_u64(id_field, id));
|
index_writer.delete_term(Term::from_field_u64(id_field, id));
|
||||||
}
|
}
|
||||||
|
IndexingOp::DeleteDocQuery { id } => {
|
||||||
|
let term = Term::from_field_u64(id_field, id);
|
||||||
|
let query = TermQuery::new(term, Default::default());
|
||||||
|
index_writer.delete_query(Box::new(query))?;
|
||||||
|
}
|
||||||
IndexingOp::Commit => {
|
IndexingOp::Commit => {
|
||||||
index_writer.commit()?;
|
index_writer.commit()?;
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,20 +1,11 @@
|
|||||||
|
use crate::query::Weight;
|
||||||
use crate::schema::{Document, Term};
|
use crate::schema::{Document, Term};
|
||||||
use crate::Opstamp;
|
use crate::Opstamp;
|
||||||
|
|
||||||
/// Timestamped Delete operation.
|
/// Timestamped Delete operation.
|
||||||
#[derive(Clone, Eq, PartialEq, Debug)]
|
|
||||||
pub struct DeleteOperation {
|
pub struct DeleteOperation {
|
||||||
pub opstamp: Opstamp,
|
pub opstamp: Opstamp,
|
||||||
pub term: Term,
|
pub target: Box<dyn Weight>,
|
||||||
}
|
|
||||||
|
|
||||||
impl Default for DeleteOperation {
|
|
||||||
fn default() -> Self {
|
|
||||||
DeleteOperation {
|
|
||||||
opstamp: 0u64,
|
|
||||||
term: Term::new(),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Timestamped Add operation.
|
/// Timestamped Add operation.
|
||||||
|
|||||||
@@ -17,7 +17,7 @@ impl<'a> PreparedCommit<'a> {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Returns the opstamp associated to the prepared commit.
|
/// Returns the opstamp associated with the prepared commit.
|
||||||
pub fn opstamp(&self) -> Opstamp {
|
pub fn opstamp(&self) -> Opstamp {
|
||||||
self.opstamp
|
self.opstamp
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -72,13 +72,9 @@ impl<'a> SortedDocIdMultiValueColumn<'a> {
|
|||||||
impl<'a> Column for SortedDocIdMultiValueColumn<'a> {
|
impl<'a> Column for SortedDocIdMultiValueColumn<'a> {
|
||||||
fn get_val(&self, pos: u64) -> u64 {
|
fn get_val(&self, pos: u64) -> u64 {
|
||||||
// use the offsets index to find the doc_id which will contain the position.
|
// use the offsets index to find the doc_id which will contain the position.
|
||||||
// the offsets are strictly increasing so we can do a simple search on it.
|
// the offsets are strictly increasing so we can do a binary search on it.
|
||||||
let new_doc_id: DocId = self
|
|
||||||
.offsets
|
let new_doc_id: DocId = self.offsets.partition_point(|&offset| offset <= pos) as DocId - 1; // Offsets start at 0, so -1 is safe
|
||||||
.iter()
|
|
||||||
.position(|&offset| offset > pos)
|
|
||||||
.expect("pos is out of bounds") as DocId
|
|
||||||
- 1u32;
|
|
||||||
|
|
||||||
// now we need to find the position of `pos` in the multivalued bucket
|
// now we need to find the position of `pos` in the multivalued bucket
|
||||||
let num_pos_covered_until_now = self.offsets[new_doc_id as usize];
|
let num_pos_covered_until_now = self.offsets[new_doc_id as usize];
|
||||||
|
|||||||
@@ -31,7 +31,7 @@ pub use self::term_info::TermInfo;
|
|||||||
|
|
||||||
pub(crate) type UnorderedTermId = u64;
|
pub(crate) type UnorderedTermId = u64;
|
||||||
|
|
||||||
#[cfg_attr(feature = "cargo-clippy", allow(clippy::enum_variant_names))]
|
#[allow(clippy::enum_variant_names)]
|
||||||
#[derive(Debug, PartialEq, Clone, Copy, Eq)]
|
#[derive(Debug, PartialEq, Clone, Copy, Eq)]
|
||||||
pub(crate) enum FreqReadingOption {
|
pub(crate) enum FreqReadingOption {
|
||||||
NoFreq,
|
NoFreq,
|
||||||
|
|||||||
@@ -50,7 +50,7 @@ impl<'a> Iterator for VInt32Reader<'a> {
|
|||||||
/// Recorder is in charge of recording relevant information about
|
/// Recorder is in charge of recording relevant information about
|
||||||
/// the presence of a term in a document.
|
/// the presence of a term in a document.
|
||||||
///
|
///
|
||||||
/// Depending on the `TextIndexingOptions` associated to the
|
/// Depending on the `TextIndexingOptions` associated with the
|
||||||
/// field, the recorder may records
|
/// field, the recorder may records
|
||||||
/// * the document frequency
|
/// * the document frequency
|
||||||
/// * the document id
|
/// * the document id
|
||||||
|
|||||||
@@ -7,7 +7,7 @@ use crate::postings::compression::COMPRESSION_BLOCK_SIZE;
|
|||||||
use crate::postings::{branchless_binary_search, BlockSegmentPostings, Postings};
|
use crate::postings::{branchless_binary_search, BlockSegmentPostings, Postings};
|
||||||
use crate::{DocId, TERMINATED};
|
use crate::{DocId, TERMINATED};
|
||||||
|
|
||||||
/// `SegmentPostings` represents the inverted list or postings associated to
|
/// `SegmentPostings` represents the inverted list or postings associated with
|
||||||
/// a term in a `Segment`.
|
/// a term in a `Segment`.
|
||||||
///
|
///
|
||||||
/// As we iterate through the `SegmentPostings`, the frequencies are optionally decoded.
|
/// As we iterate through the `SegmentPostings`, the frequencies are optionally decoded.
|
||||||
@@ -216,7 +216,7 @@ impl HasLen for SegmentPostings {
|
|||||||
}
|
}
|
||||||
|
|
||||||
impl Postings for SegmentPostings {
|
impl Postings for SegmentPostings {
|
||||||
/// Returns the frequency associated to the current document.
|
/// Returns the frequency associated with the current document.
|
||||||
/// If the schema is set up so that no frequency have been encoded,
|
/// If the schema is set up so that no frequency have been encoded,
|
||||||
/// this method should always return 1.
|
/// this method should always return 1.
|
||||||
///
|
///
|
||||||
|
|||||||
@@ -4,7 +4,7 @@ use std::ops::Range;
|
|||||||
|
|
||||||
use common::{BinarySerializable, FixedSize};
|
use common::{BinarySerializable, FixedSize};
|
||||||
|
|
||||||
/// `TermInfo` wraps the metadata associated to a Term.
|
/// `TermInfo` wraps the metadata associated with a Term.
|
||||||
/// It is segment-local.
|
/// It is segment-local.
|
||||||
#[derive(Debug, Default, Eq, PartialEq, Clone)]
|
#[derive(Debug, Default, Eq, PartialEq, Clone)]
|
||||||
pub struct TermInfo {
|
pub struct TermInfo {
|
||||||
|
|||||||
@@ -17,7 +17,7 @@ impl Query for AllQuery {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Weight associated to the `AllQuery` query.
|
/// Weight associated with the `AllQuery` query.
|
||||||
pub struct AllWeight;
|
pub struct AllWeight;
|
||||||
|
|
||||||
impl Weight for AllWeight {
|
impl Weight for AllWeight {
|
||||||
@@ -37,7 +37,7 @@ impl Weight for AllWeight {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Scorer associated to the `AllQuery` query.
|
/// Scorer associated with the `AllQuery` query.
|
||||||
pub struct AllScorer {
|
pub struct AllScorer {
|
||||||
doc: DocId,
|
doc: DocId,
|
||||||
max_doc: DocId,
|
max_doc: DocId,
|
||||||
|
|||||||
@@ -14,7 +14,7 @@ use crate::DocId;
|
|||||||
/// when the bitset is sparse
|
/// when the bitset is sparse
|
||||||
pub struct BitSetDocSet {
|
pub struct BitSetDocSet {
|
||||||
docs: BitSet,
|
docs: BitSet,
|
||||||
cursor_bucket: u32, //< index associated to the current tiny bitset
|
cursor_bucket: u32, //< index associated with the current tiny bitset
|
||||||
cursor_tinybitset: TinySet,
|
cursor_tinybitset: TinySet,
|
||||||
doc: u32,
|
doc: u32,
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -39,7 +39,7 @@ impl Explanation {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Returns the value associated to the current node.
|
/// Returns the value associated with the current node.
|
||||||
pub fn value(&self) -> Score {
|
pub fn value(&self) -> Score {
|
||||||
self.value
|
self.value
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -5,7 +5,7 @@ use crate::{DocId, Score};
|
|||||||
|
|
||||||
/// Returns the intersection scorer.
|
/// Returns the intersection scorer.
|
||||||
///
|
///
|
||||||
/// The score associated to the documents is the sum of the
|
/// The score associated with the documents is the sum of the
|
||||||
/// score of the `Scorer`s given in argument.
|
/// score of the `Scorer`s given in argument.
|
||||||
///
|
///
|
||||||
/// For better performance, the function uses a
|
/// For better performance, the function uses a
|
||||||
|
|||||||
@@ -119,7 +119,7 @@ impl PhraseQuery {
|
|||||||
}
|
}
|
||||||
|
|
||||||
impl Query for PhraseQuery {
|
impl Query for PhraseQuery {
|
||||||
/// Create the weight associated to a query.
|
/// Create the weight associated with a query.
|
||||||
///
|
///
|
||||||
/// See [`Weight`].
|
/// See [`Weight`].
|
||||||
fn weight(&self, searcher: &Searcher, scoring_enabled: bool) -> crate::Result<Box<dyn Weight>> {
|
fn weight(&self, searcher: &Searcher, scoring_enabled: bool) -> crate::Result<Box<dyn Weight>> {
|
||||||
|
|||||||
@@ -42,7 +42,7 @@ use crate::{DocAddress, Term};
|
|||||||
/// [`Scorer`]: crate::query::Scorer
|
/// [`Scorer`]: crate::query::Scorer
|
||||||
/// [`SegmentReader`]: crate::SegmentReader
|
/// [`SegmentReader`]: crate::SegmentReader
|
||||||
pub trait Query: QueryClone + Send + Sync + downcast_rs::Downcast + fmt::Debug {
|
pub trait Query: QueryClone + Send + Sync + downcast_rs::Downcast + fmt::Debug {
|
||||||
/// Create the weight associated to a query.
|
/// Create the weight associated with a query.
|
||||||
///
|
///
|
||||||
/// If scoring is not required, setting `scoring_enabled` to `false`
|
/// If scoring is not required, setting `scoring_enabled` to `false`
|
||||||
/// can increase performances.
|
/// can increase performances.
|
||||||
@@ -67,7 +67,7 @@ pub trait Query: QueryClone + Send + Sync + downcast_rs::Downcast + fmt::Debug {
|
|||||||
Ok(result)
|
Ok(result)
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Extract all of the terms associated to the query and pass them to the
|
/// Extract all of the terms associated with the query and pass them to the
|
||||||
/// given closure.
|
/// given closure.
|
||||||
///
|
///
|
||||||
/// Each term is associated with a boolean indicating whether
|
/// Each term is associated with a boolean indicating whether
|
||||||
|
|||||||
@@ -610,7 +610,7 @@ impl QueryParser {
|
|||||||
if let Some((field, path)) = self.split_full_path(full_path) {
|
if let Some((field, path)) = self.split_full_path(full_path) {
|
||||||
return Ok(vec![(field, path, literal.phrase.as_str())]);
|
return Ok(vec![(field, path, literal.phrase.as_str())]);
|
||||||
}
|
}
|
||||||
// We need to add terms associated to json default fields.
|
// We need to add terms associated with json default fields.
|
||||||
let triplets: Vec<(Field, &str, &str)> = self
|
let triplets: Vec<(Field, &str, &str)> = self
|
||||||
.default_indexed_json_fields()
|
.default_indexed_json_fields()
|
||||||
.map(|json_field| (json_field, full_path.as_str(), literal.phrase.as_str()))
|
.map(|json_field| (json_field, full_path.as_str(), literal.phrase.as_str()))
|
||||||
|
|||||||
@@ -83,7 +83,7 @@ impl BytesOptions {
|
|||||||
///
|
///
|
||||||
/// Fast fields are designed for random access.
|
/// Fast fields are designed for random access.
|
||||||
/// Access time are similar to a random lookup in an array.
|
/// Access time are similar to a random lookup in an array.
|
||||||
/// If more than one value is associated to a fast field, only the last one is
|
/// If more than one value is associated with a fast field, only the last one is
|
||||||
/// kept.
|
/// kept.
|
||||||
#[must_use]
|
#[must_use]
|
||||||
pub fn set_fast(mut self) -> BytesOptions {
|
pub fn set_fast(mut self) -> BytesOptions {
|
||||||
|
|||||||
@@ -104,7 +104,7 @@ impl DateOptions {
|
|||||||
///
|
///
|
||||||
/// Fast fields are designed for random access.
|
/// Fast fields are designed for random access.
|
||||||
/// Access time are similar to a random lookup in an array.
|
/// Access time are similar to a random lookup in an array.
|
||||||
/// If more than one value is associated to a fast field, only the last one is
|
/// If more than one value is associated with a fast field, only the last one is
|
||||||
/// kept.
|
/// kept.
|
||||||
#[must_use]
|
#[must_use]
|
||||||
pub fn set_fast(mut self, cardinality: Cardinality) -> DateOptions {
|
pub fn set_fast(mut self, cardinality: Cardinality) -> DateOptions {
|
||||||
|
|||||||
@@ -35,7 +35,7 @@ pub enum FacetParseError {
|
|||||||
/// For instance, an e-commerce website could
|
/// For instance, an e-commerce website could
|
||||||
/// have a `Facet` for `/electronics/tv_and_video/led_tv`.
|
/// have a `Facet` for `/electronics/tv_and_video/led_tv`.
|
||||||
///
|
///
|
||||||
/// A document can be associated to any number of facets.
|
/// A document can be associated with any number of facets.
|
||||||
/// The hierarchy implicitly imply that a document
|
/// The hierarchy implicitly imply that a document
|
||||||
/// belonging to a facet also belongs to the ancestor of
|
/// belonging to a facet also belongs to the ancestor of
|
||||||
/// its facet. In the example above, `/electronics/tv_and_video/`
|
/// its facet. In the example above, `/electronics/tv_and_video/`
|
||||||
|
|||||||
109
src/schema/ip_options.rs
Normal file
109
src/schema/ip_options.rs
Normal file
@@ -0,0 +1,109 @@
|
|||||||
|
use std::ops::BitOr;
|
||||||
|
|
||||||
|
use serde::{Deserialize, Serialize};
|
||||||
|
|
||||||
|
use super::flags::{FastFlag, IndexedFlag, SchemaFlagList, StoredFlag};
|
||||||
|
use super::Cardinality;
|
||||||
|
|
||||||
|
/// Define how an ip field should be handled by tantivy.
|
||||||
|
#[derive(Clone, Debug, PartialEq, Eq, Serialize, Deserialize, Default)]
|
||||||
|
pub struct IpOptions {
|
||||||
|
#[serde(skip_serializing_if = "Option::is_none")]
|
||||||
|
fast: Option<Cardinality>,
|
||||||
|
stored: bool,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl IpOptions {
|
||||||
|
/// Returns true iff the value is a fast field.
|
||||||
|
pub fn is_fast(&self) -> bool {
|
||||||
|
self.fast.is_some()
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Returns `true` if the json object should be stored.
|
||||||
|
pub fn is_stored(&self) -> bool {
|
||||||
|
self.stored
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Returns the cardinality of the fastfield.
|
||||||
|
///
|
||||||
|
/// If the field has not been declared as a fastfield, then
|
||||||
|
/// the method returns None.
|
||||||
|
pub fn get_fastfield_cardinality(&self) -> Option<Cardinality> {
|
||||||
|
self.fast
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Sets the field as stored
|
||||||
|
#[must_use]
|
||||||
|
pub fn set_stored(mut self) -> Self {
|
||||||
|
self.stored = true;
|
||||||
|
self
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Set the field as a fast field.
|
||||||
|
///
|
||||||
|
/// Fast fields are designed for random access.
|
||||||
|
/// Access time are similar to a random lookup in an array.
|
||||||
|
/// If more than one value is associated with a fast field, only the last one is
|
||||||
|
/// kept.
|
||||||
|
#[must_use]
|
||||||
|
pub fn set_fast(mut self, cardinality: Cardinality) -> Self {
|
||||||
|
self.fast = Some(cardinality);
|
||||||
|
self
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl From<()> for IpOptions {
|
||||||
|
fn from(_: ()) -> IpOptions {
|
||||||
|
IpOptions::default()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl From<FastFlag> for IpOptions {
|
||||||
|
fn from(_: FastFlag) -> Self {
|
||||||
|
IpOptions {
|
||||||
|
stored: false,
|
||||||
|
fast: Some(Cardinality::SingleValue),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl From<StoredFlag> for IpOptions {
|
||||||
|
fn from(_: StoredFlag) -> Self {
|
||||||
|
IpOptions {
|
||||||
|
stored: true,
|
||||||
|
fast: None,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl From<IndexedFlag> for IpOptions {
|
||||||
|
fn from(_: IndexedFlag) -> Self {
|
||||||
|
IpOptions {
|
||||||
|
stored: false,
|
||||||
|
fast: None,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<T: Into<IpOptions>> BitOr<T> for IpOptions {
|
||||||
|
type Output = IpOptions;
|
||||||
|
|
||||||
|
fn bitor(self, other: T) -> IpOptions {
|
||||||
|
let other = other.into();
|
||||||
|
IpOptions {
|
||||||
|
stored: self.stored | other.stored,
|
||||||
|
fast: self.fast.or(other.fast),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<Head, Tail> From<SchemaFlagList<Head, Tail>> for IpOptions
|
||||||
|
where
|
||||||
|
Head: Clone,
|
||||||
|
Tail: Clone,
|
||||||
|
Self: BitOr<Output = Self> + From<Head> + From<Tail>,
|
||||||
|
{
|
||||||
|
fn from(head_tail: SchemaFlagList<Head, Tail>) -> Self {
|
||||||
|
Self::from(head_tail.head) | Self::from(head_tail.tail)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -120,6 +120,7 @@ mod date_time_options;
|
|||||||
mod field;
|
mod field;
|
||||||
mod flags;
|
mod flags;
|
||||||
mod index_record_option;
|
mod index_record_option;
|
||||||
|
mod ip_options;
|
||||||
mod json_object_options;
|
mod json_object_options;
|
||||||
mod named_field_document;
|
mod named_field_document;
|
||||||
mod numeric_options;
|
mod numeric_options;
|
||||||
@@ -138,6 +139,7 @@ pub use self::field_type::{FieldType, Type};
|
|||||||
pub use self::field_value::FieldValue;
|
pub use self::field_value::FieldValue;
|
||||||
pub use self::flags::{FAST, INDEXED, STORED};
|
pub use self::flags::{FAST, INDEXED, STORED};
|
||||||
pub use self::index_record_option::IndexRecordOption;
|
pub use self::index_record_option::IndexRecordOption;
|
||||||
|
pub use self::ip_options::IpOptions;
|
||||||
pub use self::json_object_options::JsonObjectOptions;
|
pub use self::json_object_options::JsonObjectOptions;
|
||||||
pub use self::named_field_document::NamedFieldDocument;
|
pub use self::named_field_document::NamedFieldDocument;
|
||||||
pub use self::numeric_options::NumericOptions;
|
pub use self::numeric_options::NumericOptions;
|
||||||
|
|||||||
@@ -7,10 +7,10 @@ use crate::schema::flags::{FastFlag, IndexedFlag, SchemaFlagList, StoredFlag};
|
|||||||
/// Express whether a field is single-value or multi-valued.
|
/// Express whether a field is single-value or multi-valued.
|
||||||
#[derive(Clone, Copy, PartialEq, Eq, Debug, Serialize, Deserialize)]
|
#[derive(Clone, Copy, PartialEq, Eq, Debug, Serialize, Deserialize)]
|
||||||
pub enum Cardinality {
|
pub enum Cardinality {
|
||||||
/// The document must have exactly one value associated to the document.
|
/// The document must have exactly one value associated with the document.
|
||||||
#[serde(rename = "single")]
|
#[serde(rename = "single")]
|
||||||
SingleValue,
|
SingleValue,
|
||||||
/// The document can have any number of values associated to the document.
|
/// The document can have any number of values associated with the document.
|
||||||
/// This is more memory and CPU expensive than the `SingleValue` solution.
|
/// This is more memory and CPU expensive than the `SingleValue` solution.
|
||||||
#[serde(rename = "multi")]
|
#[serde(rename = "multi")]
|
||||||
MultiValues,
|
MultiValues,
|
||||||
@@ -124,7 +124,7 @@ impl NumericOptions {
|
|||||||
///
|
///
|
||||||
/// Fast fields are designed for random access.
|
/// Fast fields are designed for random access.
|
||||||
/// Access time are similar to a random lookup in an array.
|
/// Access time are similar to a random lookup in an array.
|
||||||
/// If more than one value is associated to a fast field, only the last one is
|
/// If more than one value is associated with a fast field, only the last one is
|
||||||
/// kept.
|
/// kept.
|
||||||
#[must_use]
|
#[must_use]
|
||||||
pub fn set_fast(mut self, cardinality: Cardinality) -> NumericOptions {
|
pub fn set_fast(mut self, cardinality: Cardinality) -> NumericOptions {
|
||||||
|
|||||||
@@ -258,7 +258,7 @@ impl Eq for InnerSchema {}
|
|||||||
pub struct Schema(Arc<InnerSchema>);
|
pub struct Schema(Arc<InnerSchema>);
|
||||||
|
|
||||||
impl Schema {
|
impl Schema {
|
||||||
/// Return the `FieldEntry` associated to a `Field`.
|
/// Return the `FieldEntry` associated with a `Field`.
|
||||||
pub fn get_field_entry(&self, field: Field) -> &FieldEntry {
|
pub fn get_field_entry(&self, field: Field) -> &FieldEntry {
|
||||||
&self.0.fields[field.field_id() as usize]
|
&self.0.fields[field.field_id() as usize]
|
||||||
}
|
}
|
||||||
@@ -422,12 +422,8 @@ pub enum DocParsingError {
|
|||||||
impl DocParsingError {
|
impl DocParsingError {
|
||||||
/// Builds a NotJson DocParsingError
|
/// Builds a NotJson DocParsingError
|
||||||
fn invalid_json(invalid_json: &str) -> Self {
|
fn invalid_json(invalid_json: &str) -> Self {
|
||||||
let sample_json: String = if invalid_json.len() < 20 {
|
let sample = invalid_json.chars().take(20).collect();
|
||||||
invalid_json.to_string()
|
DocParsingError::InvalidJson(sample)
|
||||||
} else {
|
|
||||||
format!("{:?}...", &invalid_json[0..20])
|
|
||||||
};
|
|
||||||
DocParsingError::InvalidJson(sample_json)
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -793,6 +789,11 @@ mod tests {
|
|||||||
))
|
))
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
{
|
||||||
|
// Short JSON, under the 20 char take.
|
||||||
|
let json_err = schema.parse_document(r#"{"count": 50,}"#);
|
||||||
|
assert_matches!(json_err, Err(InvalidJson(_)));
|
||||||
|
}
|
||||||
{
|
{
|
||||||
let json_err = schema.parse_document(
|
let json_err = schema.parse_document(
|
||||||
r#"{
|
r#"{
|
||||||
|
|||||||
@@ -161,7 +161,7 @@ impl TextFieldIndexing {
|
|||||||
self
|
self
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Returns the indexing options associated to this field.
|
/// Returns the indexing options associated with this field.
|
||||||
///
|
///
|
||||||
/// See [`IndexRecordOption`] for more detail.
|
/// See [`IndexRecordOption`] for more detail.
|
||||||
pub fn index_option(&self) -> IndexRecordOption {
|
pub fn index_option(&self) -> IndexRecordOption {
|
||||||
|
|||||||
@@ -342,7 +342,7 @@ impl SnippetGenerator {
|
|||||||
|
|
||||||
/// Generates a snippet for the given `Document`.
|
/// Generates a snippet for the given `Document`.
|
||||||
///
|
///
|
||||||
/// This method extract the text associated to the `SnippetGenerator`'s field
|
/// This method extract the text associated with the `SnippetGenerator`'s field
|
||||||
/// and computes a snippet.
|
/// and computes a snippet.
|
||||||
pub fn snippet_from_doc(&self, doc: &Document) -> Snippet {
|
pub fn snippet_from_doc(&self, doc: &Document) -> Snippet {
|
||||||
let text: String = doc
|
let text: String = doc
|
||||||
|
|||||||
@@ -136,7 +136,7 @@ where A: Automaton
|
|||||||
}
|
}
|
||||||
|
|
||||||
/// Return the next `(key, value)` pair.
|
/// Return the next `(key, value)` pair.
|
||||||
#[cfg_attr(feature = "cargo-clippy", allow(clippy::should_implement_trait))]
|
#[allow(clippy::should_implement_trait)]
|
||||||
pub fn next(&mut self) -> Option<(&[u8], &TermInfo)> {
|
pub fn next(&mut self) -> Option<(&[u8], &TermInfo)> {
|
||||||
if self.advance() {
|
if self.advance() {
|
||||||
Some((self.key(), self.value()))
|
Some((self.key(), self.value()))
|
||||||
|
|||||||
@@ -138,12 +138,12 @@ impl TermDictionary {
|
|||||||
self.term_info_store.num_terms()
|
self.term_info_store.num_terms()
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Returns the ordinal associated to a given term.
|
/// Returns the ordinal associated with a given term.
|
||||||
pub fn term_ord<K: AsRef<[u8]>>(&self, key: K) -> io::Result<Option<TermOrdinal>> {
|
pub fn term_ord<K: AsRef<[u8]>>(&self, key: K) -> io::Result<Option<TermOrdinal>> {
|
||||||
Ok(self.fst_index.get(key))
|
Ok(self.fst_index.get(key))
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Stores the term associated to a given term ordinal in
|
/// Stores the term associated with a given term ordinal in
|
||||||
/// a `bytes` buffer.
|
/// a `bytes` buffer.
|
||||||
///
|
///
|
||||||
/// Term ordinals are defined as the position of the term in
|
/// Term ordinals are defined as the position of the term in
|
||||||
|
|||||||
@@ -179,7 +179,7 @@ where
|
|||||||
}
|
}
|
||||||
|
|
||||||
/// Return the next `(key, value)` pair.
|
/// Return the next `(key, value)` pair.
|
||||||
#[cfg_attr(feature = "cargo-clippy", allow(clippy::should_implement_trait))]
|
#[allow(clippy::should_implement_trait)]
|
||||||
pub fn next(&mut self) -> Option<(&[u8], &TermInfo)> {
|
pub fn next(&mut self) -> Option<(&[u8], &TermInfo)> {
|
||||||
if self.advance() {
|
if self.advance() {
|
||||||
Some((self.key(), self.value()))
|
Some((self.key(), self.value()))
|
||||||
|
|||||||
@@ -52,7 +52,7 @@ impl<W: io::Write> TermDictionaryBuilder<W> {
|
|||||||
/// to insert_key and insert_value.
|
/// to insert_key and insert_value.
|
||||||
///
|
///
|
||||||
/// Prefer using `.insert(key, value)`
|
/// Prefer using `.insert(key, value)`
|
||||||
#[allow(clippy::clippy::clippy::unnecessary_wraps)]
|
#[allow(clippy::unnecessary_wraps)]
|
||||||
pub(crate) fn insert_key(&mut self, key: &[u8]) -> io::Result<()> {
|
pub(crate) fn insert_key(&mut self, key: &[u8]) -> io::Result<()> {
|
||||||
self.sstable_writer.write_key(key);
|
self.sstable_writer.write_key(key);
|
||||||
Ok(())
|
Ok(())
|
||||||
@@ -153,7 +153,7 @@ impl TermDictionary {
|
|||||||
self.num_terms as usize
|
self.num_terms as usize
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Returns the ordinal associated to a given term.
|
/// Returns the ordinal associated with a given term.
|
||||||
pub fn term_ord<K: AsRef<[u8]>>(&self, key: K) -> io::Result<Option<TermOrdinal>> {
|
pub fn term_ord<K: AsRef<[u8]>>(&self, key: K) -> io::Result<Option<TermOrdinal>> {
|
||||||
let mut term_ord = 0u64;
|
let mut term_ord = 0u64;
|
||||||
let key_bytes = key.as_ref();
|
let key_bytes = key.as_ref();
|
||||||
@@ -167,7 +167,7 @@ impl TermDictionary {
|
|||||||
Ok(None)
|
Ok(None)
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Returns the term associated to a given term ordinal.
|
/// Returns the term associated with a given term ordinal.
|
||||||
///
|
///
|
||||||
/// Term ordinals are defined as the position of the term in
|
/// Term ordinals are defined as the position of the term in
|
||||||
/// the sorted list of terms.
|
/// the sorted list of terms.
|
||||||
|
|||||||
Reference in New Issue
Block a user