mirror of
https://github.com/quickwit-oss/tantivy.git
synced 2026-01-06 09:12:55 +00:00
Compare commits
11 Commits
ownedbytes
...
check_doc_
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
cf1460d296 | ||
|
|
047da20b5b | ||
|
|
1417eaf3a7 | ||
|
|
4f8493d2de | ||
|
|
8861366137 | ||
|
|
0e9fced336 | ||
|
|
b257b960b3 | ||
|
|
4708171a32 | ||
|
|
b493743f8d | ||
|
|
d2955a3fd2 | ||
|
|
17d5869ad6 |
4
.github/workflows/coverage.yml
vendored
4
.github/workflows/coverage.yml
vendored
@@ -15,11 +15,11 @@ jobs:
|
|||||||
steps:
|
steps:
|
||||||
- uses: actions/checkout@v4
|
- uses: actions/checkout@v4
|
||||||
- name: Install Rust
|
- name: Install Rust
|
||||||
run: rustup toolchain install nightly-2023-09-10 --profile minimal --component llvm-tools-preview
|
run: rustup toolchain install nightly-2024-04-10 --profile minimal --component llvm-tools-preview
|
||||||
- uses: Swatinem/rust-cache@v2
|
- uses: Swatinem/rust-cache@v2
|
||||||
- uses: taiki-e/install-action@cargo-llvm-cov
|
- uses: taiki-e/install-action@cargo-llvm-cov
|
||||||
- name: Generate code coverage
|
- name: Generate code coverage
|
||||||
run: cargo +nightly-2023-09-10 llvm-cov --all-features --workspace --doctests --lcov --output-path lcov.info
|
run: cargo +nightly-2024-04-10 llvm-cov --all-features --workspace --doctests --lcov --output-path lcov.info
|
||||||
- name: Upload coverage to Codecov
|
- name: Upload coverage to Codecov
|
||||||
uses: codecov/codecov-action@v3
|
uses: codecov/codecov-action@v3
|
||||||
continue-on-error: true
|
continue-on-error: true
|
||||||
|
|||||||
@@ -1,4 +1,4 @@
|
|||||||
Tantivy 0.22 (unreleased)
|
Tantivy 0.22
|
||||||
================================
|
================================
|
||||||
|
|
||||||
Tantivy 0.22 will be able to read indices created with Tantivy 0.21.
|
Tantivy 0.22 will be able to read indices created with Tantivy 0.21.
|
||||||
|
|||||||
@@ -62,6 +62,7 @@ tokenizer-api = { version= "0.3", path="./tokenizer-api", package="tantivy-token
|
|||||||
sketches-ddsketch = { version = "0.2.1", features = ["use_serde"] }
|
sketches-ddsketch = { version = "0.2.1", features = ["use_serde"] }
|
||||||
futures-util = { version = "0.3.28", optional = true }
|
futures-util = { version = "0.3.28", optional = true }
|
||||||
fnv = "1.0.7"
|
fnv = "1.0.7"
|
||||||
|
mediumvec = "1.3.0"
|
||||||
|
|
||||||
[target.'cfg(windows)'.dependencies]
|
[target.'cfg(windows)'.dependencies]
|
||||||
winapi = "0.3.9"
|
winapi = "0.3.9"
|
||||||
@@ -81,6 +82,7 @@ time = { version = "0.3.10", features = ["serde-well-known", "macros"] }
|
|||||||
postcard = { version = "1.0.4", features = [
|
postcard = { version = "1.0.4", features = [
|
||||||
"use-std",
|
"use-std",
|
||||||
], default-features = false }
|
], default-features = false }
|
||||||
|
peakmem-alloc = "0.3.0"
|
||||||
|
|
||||||
[target.'cfg(not(windows))'.dev-dependencies]
|
[target.'cfg(not(windows))'.dev-dependencies]
|
||||||
criterion = { version = "0.5", default-features = false }
|
criterion = { version = "0.5", default-features = false }
|
||||||
|
|||||||
@@ -1,5 +1,5 @@
|
|||||||
use std::io::Write;
|
use std::io::Write;
|
||||||
use std::{fmt, io, u64};
|
use std::{fmt, io};
|
||||||
|
|
||||||
use ownedbytes::OwnedBytes;
|
use ownedbytes::OwnedBytes;
|
||||||
|
|
||||||
|
|||||||
@@ -5,6 +5,12 @@ pub const JSON_PATH_SEGMENT_SEP: u8 = 1u8;
|
|||||||
pub const JSON_PATH_SEGMENT_SEP_STR: &str =
|
pub const JSON_PATH_SEGMENT_SEP_STR: &str =
|
||||||
unsafe { std::str::from_utf8_unchecked(&[JSON_PATH_SEGMENT_SEP]) };
|
unsafe { std::str::from_utf8_unchecked(&[JSON_PATH_SEGMENT_SEP]) };
|
||||||
|
|
||||||
|
/// Separates the json path and the value in
|
||||||
|
/// a JSON term binary representation.
|
||||||
|
pub const JSON_END_OF_PATH: u8 = 0u8;
|
||||||
|
pub const JSON_END_OF_PATH_STR: &str =
|
||||||
|
unsafe { std::str::from_utf8_unchecked(&[JSON_END_OF_PATH]) };
|
||||||
|
|
||||||
/// Create a new JsonPathWriter, that creates flattened json paths for tantivy.
|
/// Create a new JsonPathWriter, that creates flattened json paths for tantivy.
|
||||||
#[derive(Clone, Debug, Default)]
|
#[derive(Clone, Debug, Default)]
|
||||||
pub struct JsonPathWriter {
|
pub struct JsonPathWriter {
|
||||||
@@ -14,6 +20,14 @@ pub struct JsonPathWriter {
|
|||||||
}
|
}
|
||||||
|
|
||||||
impl JsonPathWriter {
|
impl JsonPathWriter {
|
||||||
|
pub fn with_expand_dots(expand_dots: bool) -> Self {
|
||||||
|
JsonPathWriter {
|
||||||
|
path: String::new(),
|
||||||
|
indices: Vec::new(),
|
||||||
|
expand_dots,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
pub fn new() -> Self {
|
pub fn new() -> Self {
|
||||||
JsonPathWriter {
|
JsonPathWriter {
|
||||||
path: String::new(),
|
path: String::new(),
|
||||||
@@ -39,8 +53,8 @@ impl JsonPathWriter {
|
|||||||
pub fn push(&mut self, segment: &str) {
|
pub fn push(&mut self, segment: &str) {
|
||||||
let len_path = self.path.len();
|
let len_path = self.path.len();
|
||||||
self.indices.push(len_path);
|
self.indices.push(len_path);
|
||||||
if !self.path.is_empty() {
|
if self.indices.len() > 1 {
|
||||||
self.path.push_str(JSON_PATH_SEGMENT_SEP_STR);
|
self.path.push(JSON_PATH_SEGMENT_SEP as char);
|
||||||
}
|
}
|
||||||
self.path.push_str(segment);
|
self.path.push_str(segment);
|
||||||
if self.expand_dots {
|
if self.expand_dots {
|
||||||
@@ -55,6 +69,12 @@ impl JsonPathWriter {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Set the end of JSON path marker.
|
||||||
|
#[inline]
|
||||||
|
pub fn set_end(&mut self) {
|
||||||
|
self.path.push_str(JSON_END_OF_PATH_STR);
|
||||||
|
}
|
||||||
|
|
||||||
/// Remove the last segment. Does nothing if the path is empty.
|
/// Remove the last segment. Does nothing if the path is empty.
|
||||||
#[inline]
|
#[inline]
|
||||||
pub fn pop(&mut self) {
|
pub fn pop(&mut self) {
|
||||||
@@ -91,6 +111,7 @@ mod tests {
|
|||||||
#[test]
|
#[test]
|
||||||
fn json_path_writer_test() {
|
fn json_path_writer_test() {
|
||||||
let mut writer = JsonPathWriter::new();
|
let mut writer = JsonPathWriter::new();
|
||||||
|
writer.set_expand_dots(false);
|
||||||
|
|
||||||
writer.push("root");
|
writer.push("root");
|
||||||
assert_eq!(writer.as_str(), "root");
|
assert_eq!(writer.as_str(), "root");
|
||||||
@@ -109,4 +130,15 @@ mod tests {
|
|||||||
writer.push("k8s.node.id");
|
writer.push("k8s.node.id");
|
||||||
assert_eq!(writer.as_str(), "root\u{1}k8s\u{1}node\u{1}id");
|
assert_eq!(writer.as_str(), "root\u{1}k8s\u{1}node\u{1}id");
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_json_path_expand_dots_enabled_pop_segment() {
|
||||||
|
let mut json_writer = JsonPathWriter::with_expand_dots(true);
|
||||||
|
json_writer.push("hello");
|
||||||
|
assert_eq!(json_writer.as_str(), "hello");
|
||||||
|
json_writer.push("color.hue");
|
||||||
|
assert_eq!(json_writer.as_str(), "hello\x01color\x01hue");
|
||||||
|
json_writer.pop();
|
||||||
|
assert_eq!(json_writer.as_str(), "hello");
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -9,7 +9,7 @@ mod byte_count;
|
|||||||
mod datetime;
|
mod datetime;
|
||||||
pub mod file_slice;
|
pub mod file_slice;
|
||||||
mod group_by;
|
mod group_by;
|
||||||
mod json_path_writer;
|
pub mod json_path_writer;
|
||||||
mod serialize;
|
mod serialize;
|
||||||
mod vint;
|
mod vint;
|
||||||
mod writer;
|
mod writer;
|
||||||
|
|||||||
335
examples/doc_mem.rs
Normal file
335
examples/doc_mem.rs
Normal file
@@ -0,0 +1,335 @@
|
|||||||
|
#![allow(unused_imports)]
|
||||||
|
#![allow(dead_code)]
|
||||||
|
use std::alloc::System;
|
||||||
|
use std::env::args;
|
||||||
|
use std::net::Ipv6Addr;
|
||||||
|
|
||||||
|
use columnar::{MonotonicallyMappableToU128, MonotonicallyMappableToU64};
|
||||||
|
use common::{BinarySerializable, CountingWriter, DateTime, FixedSize};
|
||||||
|
use peakmem_alloc::*;
|
||||||
|
use tantivy::schema::{Field, FieldValue, OwnedValue, FAST, INDEXED, STRING, TEXT};
|
||||||
|
use tantivy::tokenizer::PreTokenizedString;
|
||||||
|
use tantivy::{doc, TantivyDocument};
|
||||||
|
|
||||||
|
const GH_LOGS: &str = include_str!("../benches/gh.json");
|
||||||
|
const HDFS_LOGS: &str = include_str!("../benches/hdfs.json");
|
||||||
|
|
||||||
|
#[global_allocator]
|
||||||
|
static GLOBAL: &PeakMemAlloc<System> = &INSTRUMENTED_SYSTEM;
|
||||||
|
|
||||||
|
fn main() {
|
||||||
|
dbg!(std::mem::size_of::<TantivyDocument>());
|
||||||
|
dbg!(std::mem::size_of::<DocContainerRef>());
|
||||||
|
dbg!(std::mem::size_of::<OwnedValue>());
|
||||||
|
dbg!(std::mem::size_of::<OwnedValueMedVec>());
|
||||||
|
dbg!(std::mem::size_of::<ValueContainerRef>());
|
||||||
|
dbg!(std::mem::size_of::<mediumvec::vec32::Vec32::<u8>>());
|
||||||
|
|
||||||
|
let filter = args().nth(1);
|
||||||
|
measure_fn(
|
||||||
|
test_hdfs::<TantivyDocument>,
|
||||||
|
"hdfs TantivyDocument",
|
||||||
|
&filter,
|
||||||
|
);
|
||||||
|
measure_fn(
|
||||||
|
test_hdfs::<TantivyDocumentMedVec>,
|
||||||
|
"hdfs TantivyDocumentMedVec",
|
||||||
|
&filter,
|
||||||
|
);
|
||||||
|
measure_fn(
|
||||||
|
test_hdfs::<DocContainerRef>,
|
||||||
|
"hdfs DocContainerRef",
|
||||||
|
&filter,
|
||||||
|
);
|
||||||
|
measure_fn(test_gh::<TantivyDocument>, "gh TantivyDocument", &filter);
|
||||||
|
measure_fn(
|
||||||
|
test_gh::<TantivyDocumentMedVec>,
|
||||||
|
"gh TantivyDocumentMedVec",
|
||||||
|
&filter,
|
||||||
|
);
|
||||||
|
measure_fn(test_gh::<DocContainerRef>, "gh DocContainerRef", &filter);
|
||||||
|
}
|
||||||
|
fn measure_fn<F: FnOnce()>(f: F, name: &str, filter: &Option<std::string::String>) {
|
||||||
|
if let Some(filter) = filter {
|
||||||
|
if !name.contains(filter) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
GLOBAL.reset_peak_memory();
|
||||||
|
f();
|
||||||
|
println!("Peak Memory {} : {:#?}", GLOBAL.get_peak_memory(), name);
|
||||||
|
}
|
||||||
|
fn test_hdfs<T: From<TantivyDocument>>() {
|
||||||
|
let schema = {
|
||||||
|
let mut schema_builder = tantivy::schema::SchemaBuilder::new();
|
||||||
|
schema_builder.add_u64_field("timestamp", INDEXED);
|
||||||
|
schema_builder.add_text_field("body", TEXT);
|
||||||
|
schema_builder.add_text_field("severity", STRING);
|
||||||
|
schema_builder.build()
|
||||||
|
};
|
||||||
|
let mut docs: Vec<T> = Vec::with_capacity(HDFS_LOGS.lines().count());
|
||||||
|
for doc_json in HDFS_LOGS.lines() {
|
||||||
|
let doc = TantivyDocument::parse_json(&schema, doc_json)
|
||||||
|
.unwrap()
|
||||||
|
.into();
|
||||||
|
docs.push(doc);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn test_gh<T: From<TantivyDocument>>() {
|
||||||
|
let schema = {
|
||||||
|
let mut schema_builder = tantivy::schema::SchemaBuilder::new();
|
||||||
|
schema_builder.add_json_field("json", FAST);
|
||||||
|
schema_builder.build()
|
||||||
|
};
|
||||||
|
let mut docs: Vec<T> = Vec::with_capacity(GH_LOGS.lines().count());
|
||||||
|
for doc_json in GH_LOGS.lines() {
|
||||||
|
let json_field = schema.get_field("json").unwrap();
|
||||||
|
|
||||||
|
let json_val: serde_json::Map<String, serde_json::Value> =
|
||||||
|
serde_json::from_str(doc_json).unwrap();
|
||||||
|
let doc = tantivy::doc!(json_field=>json_val).into();
|
||||||
|
docs.push(doc);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Clone, Debug, Default)]
|
||||||
|
#[allow(dead_code)]
|
||||||
|
pub struct TantivyDocumentMedVec {
|
||||||
|
field_values: mediumvec::Vec32<FieldValueMedVec>,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Clone, PartialEq)]
|
||||||
|
pub struct FieldValueMedVec {
|
||||||
|
pub field: Field,
|
||||||
|
pub value: OwnedValueMedVec,
|
||||||
|
}
|
||||||
|
|
||||||
|
/// This is a owned variant of `Value`, that can be passed around without lifetimes.
|
||||||
|
/// Represents the value of a any field.
|
||||||
|
/// It is an enum over all over all of the possible field type.
|
||||||
|
#[derive(Debug, Clone, PartialEq)]
|
||||||
|
pub enum OwnedValueMedVec {
|
||||||
|
/// A null value.
|
||||||
|
Null,
|
||||||
|
/// The str type is used for any text information.
|
||||||
|
Str(mediumvec::vec32::Vec32<u8>),
|
||||||
|
/// Unsigned 64-bits Integer `u64`
|
||||||
|
U64(u64),
|
||||||
|
/// Signed 64-bits Integer `i64`
|
||||||
|
I64(i64),
|
||||||
|
/// 64-bits Float `f64`
|
||||||
|
F64(f64),
|
||||||
|
/// Bool value
|
||||||
|
Bool(bool),
|
||||||
|
/// Date/time with nanoseconds precision
|
||||||
|
Date(DateTime),
|
||||||
|
Array(mediumvec::vec32::Vec32<Self>),
|
||||||
|
/// Dynamic object value.
|
||||||
|
Object(mediumvec::vec32::Vec32<(String, Self)>),
|
||||||
|
/// IpV6 Address. Internally there is no IpV4, it needs to be converted to `Ipv6Addr`.
|
||||||
|
IpAddr(Ipv6Addr),
|
||||||
|
/// Pre-tokenized str type,
|
||||||
|
PreTokStr(Box<PreTokenizedString>),
|
||||||
|
/// Arbitrarily sized byte array
|
||||||
|
Bytes(mediumvec::vec32::Vec32<u8>),
|
||||||
|
}
|
||||||
|
|
||||||
|
impl From<TantivyDocument> for TantivyDocumentMedVec {
|
||||||
|
fn from(doc: TantivyDocument) -> Self {
|
||||||
|
let field_values = doc
|
||||||
|
.into_iter()
|
||||||
|
.map(|fv| FieldValueMedVec {
|
||||||
|
field: fv.field,
|
||||||
|
value: fv.value.into(),
|
||||||
|
})
|
||||||
|
.collect();
|
||||||
|
TantivyDocumentMedVec { field_values }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
impl From<OwnedValue> for OwnedValueMedVec {
|
||||||
|
fn from(value: OwnedValue) -> Self {
|
||||||
|
match value {
|
||||||
|
OwnedValue::Null => OwnedValueMedVec::Null,
|
||||||
|
OwnedValue::Str(s) => {
|
||||||
|
let bytes = s.into_bytes();
|
||||||
|
let vec = mediumvec::vec32::Vec32::from_vec(bytes);
|
||||||
|
OwnedValueMedVec::Str(vec)
|
||||||
|
}
|
||||||
|
OwnedValue::U64(u) => OwnedValueMedVec::U64(u),
|
||||||
|
OwnedValue::I64(i) => OwnedValueMedVec::I64(i),
|
||||||
|
OwnedValue::F64(f) => OwnedValueMedVec::F64(f),
|
||||||
|
OwnedValue::Bool(b) => OwnedValueMedVec::Bool(b),
|
||||||
|
OwnedValue::Date(d) => OwnedValueMedVec::Date(d),
|
||||||
|
OwnedValue::Array(arr) => {
|
||||||
|
let arr = arr.into_iter().map(|v| v.into()).collect();
|
||||||
|
OwnedValueMedVec::Array(arr)
|
||||||
|
}
|
||||||
|
OwnedValue::Object(obj) => {
|
||||||
|
let obj = obj.into_iter().map(|(k, v)| (k, v.into())).collect();
|
||||||
|
OwnedValueMedVec::Object(obj)
|
||||||
|
}
|
||||||
|
OwnedValue::IpAddr(ip) => OwnedValueMedVec::IpAddr(ip),
|
||||||
|
_ => panic!("Unsupported value type {:?}", value),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[repr(packed)]
|
||||||
|
pub struct FieldValueContainerRef {
|
||||||
|
pub field: u16,
|
||||||
|
pub value: ValueContainerRef,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[repr(packed)]
|
||||||
|
struct DocContainerRef {
|
||||||
|
container: OwnedValueRefContainer,
|
||||||
|
field_values: mediumvec::Vec32<FieldValueContainerRef>,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Default)]
|
||||||
|
struct OwnedValueRefContainer {
|
||||||
|
nodes: mediumvec::Vec32<ValueContainerRef>,
|
||||||
|
node_data: mediumvec::Vec32<u8>,
|
||||||
|
}
|
||||||
|
impl OwnedValueRefContainer {
|
||||||
|
fn shrink_to_fit(&mut self) {
|
||||||
|
self.nodes.shrink_to_fit();
|
||||||
|
self.node_data.shrink_to_fit();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl From<TantivyDocument> for DocContainerRef {
|
||||||
|
fn from(doc: TantivyDocument) -> Self {
|
||||||
|
let mut container = OwnedValueRefContainer::default();
|
||||||
|
let field_values = doc
|
||||||
|
.into_iter()
|
||||||
|
.map(|fv| FieldValueContainerRef {
|
||||||
|
field: fv.field.field_id().try_into().unwrap(),
|
||||||
|
value: container.add_value(fv.value),
|
||||||
|
})
|
||||||
|
.collect();
|
||||||
|
container.shrink_to_fit();
|
||||||
|
Self {
|
||||||
|
field_values,
|
||||||
|
container,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// References to positions in two array, one for the OwnedValueRef and the other for the encoded
|
||||||
|
// bytes
|
||||||
|
#[derive(Debug, Clone, PartialEq)]
|
||||||
|
pub enum ValueContainerRef {
|
||||||
|
/// A null value.
|
||||||
|
Null,
|
||||||
|
/// The str type is used for any text information.
|
||||||
|
Str(u32),
|
||||||
|
/// Unsigned 64-bits Integer `u64`
|
||||||
|
U64(u32), // position of the serialized 8 bytes in the data array
|
||||||
|
/// Signed 64-bits Integer `i64`
|
||||||
|
I64(u32), // position of the serialized 8 bytes in the data array
|
||||||
|
/// 64-bits Float `f64`
|
||||||
|
F64(u32), // position of the serialized 8 bytes in the data array
|
||||||
|
/// Bool value
|
||||||
|
Bool(bool), // inlined bool
|
||||||
|
/// Date/time with nanoseconds precision
|
||||||
|
Date(u32), // position of the serialized 8 byte in the data array
|
||||||
|
Array(NodeAddress),
|
||||||
|
/// Dynamic object value.
|
||||||
|
Object(NodeAddress),
|
||||||
|
/// IpV6 Address. Internally there is no IpV4, it needs to be converted to `Ipv6Addr`.
|
||||||
|
IpAddr(u32), // position of the serialized 16 bytes in the data array
|
||||||
|
/// Arbitrarily sized byte array
|
||||||
|
Bytes(u32),
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Clone, PartialEq)]
|
||||||
|
pub struct NodeAddress {
|
||||||
|
pos: u32,
|
||||||
|
num_nodes: u32,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl OwnedValueRefContainer {
|
||||||
|
pub fn add_value(&mut self, value: OwnedValue) -> ValueContainerRef {
|
||||||
|
match value {
|
||||||
|
OwnedValue::Null => ValueContainerRef::Null,
|
||||||
|
OwnedValue::U64(num) => ValueContainerRef::U64(write_into(&mut self.node_data, num)),
|
||||||
|
OwnedValue::I64(num) => ValueContainerRef::I64(write_into(&mut self.node_data, num)),
|
||||||
|
OwnedValue::F64(num) => ValueContainerRef::F64(write_into(&mut self.node_data, num)),
|
||||||
|
OwnedValue::Bool(b) => ValueContainerRef::Bool(b),
|
||||||
|
OwnedValue::Date(date) => ValueContainerRef::Date(write_into(
|
||||||
|
&mut self.node_data,
|
||||||
|
date.into_timestamp_nanos(),
|
||||||
|
)),
|
||||||
|
OwnedValue::Str(bytes) => {
|
||||||
|
ValueContainerRef::Str(write_into(&mut self.node_data, bytes))
|
||||||
|
}
|
||||||
|
OwnedValue::Bytes(bytes) => {
|
||||||
|
ValueContainerRef::Bytes(write_into(&mut self.node_data, bytes))
|
||||||
|
}
|
||||||
|
OwnedValue::Array(elements) => {
|
||||||
|
let pos = self.nodes.len() as u32;
|
||||||
|
let len = elements.len() as u32;
|
||||||
|
for elem in elements {
|
||||||
|
let ref_elem = self.add_value(elem);
|
||||||
|
self.nodes.push(ref_elem);
|
||||||
|
}
|
||||||
|
ValueContainerRef::Array(NodeAddress {
|
||||||
|
pos,
|
||||||
|
num_nodes: len,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
OwnedValue::Object(entries) => {
|
||||||
|
let pos = self.nodes.len() as u32;
|
||||||
|
let len = entries.len() as u32;
|
||||||
|
for (key, value) in entries {
|
||||||
|
let ref_key = self.add_value(OwnedValue::Str(key));
|
||||||
|
let ref_value = self.add_value(value);
|
||||||
|
self.nodes.push(ref_key);
|
||||||
|
self.nodes.push(ref_value);
|
||||||
|
}
|
||||||
|
ValueContainerRef::Object(NodeAddress {
|
||||||
|
pos,
|
||||||
|
num_nodes: len,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
OwnedValue::IpAddr(num) => {
|
||||||
|
ValueContainerRef::IpAddr(write_into(&mut self.node_data, num.to_u128()))
|
||||||
|
}
|
||||||
|
OwnedValue::PreTokStr(_) => todo!(),
|
||||||
|
OwnedValue::Facet(_) => todo!(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn write_into<T: BinarySerializable>(data: &mut mediumvec::Vec32<u8>, value: T) -> u32 {
|
||||||
|
let pos = data.len() as u32;
|
||||||
|
data.as_vec(|vec| value.serialize(vec).unwrap());
|
||||||
|
pos
|
||||||
|
}
|
||||||
|
|
||||||
|
fn write_into_2<T: BinarySerializable>(data: &mut mediumvec::Vec32<u8>, value: T) -> NodeAddress {
|
||||||
|
let pos = data.len() as u32;
|
||||||
|
let mut len = 0;
|
||||||
|
data.as_vec(|vec| {
|
||||||
|
let mut wrt = CountingWriter::wrap(vec);
|
||||||
|
value.serialize(&mut wrt).unwrap();
|
||||||
|
len = wrt.written_bytes() as u32;
|
||||||
|
});
|
||||||
|
NodeAddress {
|
||||||
|
pos,
|
||||||
|
num_nodes: len,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// impl From<ContainerDocRef> for TantivyDocument {
|
||||||
|
// fn from(doc: ContainerDocRef) -> Self {
|
||||||
|
// let mut doc2 = TantivyDocument::new();
|
||||||
|
// for fv in doc.field_values {
|
||||||
|
// let field = Field::from_field_id(fv.field as u32);
|
||||||
|
// let value = doc.container.get_value(fv.value);
|
||||||
|
// doc2.add(FieldValue::new(field, value));
|
||||||
|
//}
|
||||||
|
// doc2
|
||||||
|
//}
|
||||||
@@ -218,27 +218,14 @@ fn term_or_phrase_infallible(inp: &str) -> JResult<&str, Option<UserInputLeaf>>
|
|||||||
}
|
}
|
||||||
|
|
||||||
fn term_group(inp: &str) -> IResult<&str, UserInputAst> {
|
fn term_group(inp: &str) -> IResult<&str, UserInputAst> {
|
||||||
let occur_symbol = alt((
|
|
||||||
value(Occur::MustNot, char('-')),
|
|
||||||
value(Occur::Must, char('+')),
|
|
||||||
));
|
|
||||||
|
|
||||||
map(
|
map(
|
||||||
tuple((
|
tuple((
|
||||||
terminated(field_name, multispace0),
|
terminated(field_name, multispace0),
|
||||||
delimited(
|
delimited(tuple((char('('), multispace0)), ast, char(')')),
|
||||||
tuple((char('('), multispace0)),
|
|
||||||
separated_list0(multispace1, tuple((opt(occur_symbol), term_or_phrase))),
|
|
||||||
char(')'),
|
|
||||||
),
|
|
||||||
)),
|
)),
|
||||||
|(field_name, terms)| {
|
|(field_name, mut ast)| {
|
||||||
UserInputAst::Clause(
|
ast.set_default_field(field_name);
|
||||||
terms
|
ast
|
||||||
.into_iter()
|
|
||||||
.map(|(occur, leaf)| (occur, leaf.set_field(Some(field_name.clone())).into()))
|
|
||||||
.collect(),
|
|
||||||
)
|
|
||||||
},
|
},
|
||||||
)(inp)
|
)(inp)
|
||||||
}
|
}
|
||||||
@@ -258,46 +245,18 @@ fn term_group_precond(inp: &str) -> IResult<&str, (), ()> {
|
|||||||
}
|
}
|
||||||
|
|
||||||
fn term_group_infallible(inp: &str) -> JResult<&str, UserInputAst> {
|
fn term_group_infallible(inp: &str) -> JResult<&str, UserInputAst> {
|
||||||
let (mut inp, (field_name, _, _, _)) =
|
let (inp, (field_name, _, _, _)) =
|
||||||
tuple((field_name, multispace0, char('('), multispace0))(inp).expect("precondition failed");
|
tuple((field_name, multispace0, char('('), multispace0))(inp).expect("precondition failed");
|
||||||
|
|
||||||
let mut terms = Vec::new();
|
let res = delimited_infallible(
|
||||||
let mut errs = Vec::new();
|
nothing,
|
||||||
|
map(ast_infallible, |(mut ast, errors)| {
|
||||||
let mut first_round = true;
|
ast.set_default_field(field_name.to_string());
|
||||||
loop {
|
(ast, errors)
|
||||||
let mut space_error = if first_round {
|
}),
|
||||||
first_round = false;
|
opt_i_err(char(')'), "expected ')'"),
|
||||||
Vec::new()
|
)(inp);
|
||||||
} else {
|
res
|
||||||
let (rest, (_, err)) = space1_infallible(inp)?;
|
|
||||||
inp = rest;
|
|
||||||
err
|
|
||||||
};
|
|
||||||
if inp.is_empty() {
|
|
||||||
errs.push(LenientErrorInternal {
|
|
||||||
pos: inp.len(),
|
|
||||||
message: "missing )".to_string(),
|
|
||||||
});
|
|
||||||
break Ok((inp, (UserInputAst::Clause(terms), errs)));
|
|
||||||
}
|
|
||||||
if let Some(inp) = inp.strip_prefix(')') {
|
|
||||||
break Ok((inp, (UserInputAst::Clause(terms), errs)));
|
|
||||||
}
|
|
||||||
// only append missing space error if we did not reach the end of group
|
|
||||||
errs.append(&mut space_error);
|
|
||||||
|
|
||||||
// here we do the assumption term_or_phrase_infallible always consume something if the
|
|
||||||
// first byte is not `)` or ' '. If it did not, we would end up looping.
|
|
||||||
|
|
||||||
let (rest, ((occur, leaf), mut err)) =
|
|
||||||
tuple_infallible((occur_symbol, term_or_phrase_infallible))(inp)?;
|
|
||||||
errs.append(&mut err);
|
|
||||||
if let Some(leaf) = leaf {
|
|
||||||
terms.push((occur, leaf.set_field(Some(field_name.clone())).into()));
|
|
||||||
}
|
|
||||||
inp = rest;
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
fn exists(inp: &str) -> IResult<&str, UserInputLeaf> {
|
fn exists(inp: &str) -> IResult<&str, UserInputLeaf> {
|
||||||
@@ -1468,8 +1427,18 @@ mod test {
|
|||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_parse_query_term_group() {
|
fn test_parse_query_term_group() {
|
||||||
test_parse_query_to_ast_helper(r#"field:(abc)"#, r#"(*"field":abc)"#);
|
test_parse_query_to_ast_helper(r#"field:(abc)"#, r#""field":abc"#);
|
||||||
test_parse_query_to_ast_helper(r#"field:(+a -"b c")"#, r#"(+"field":a -"field":"b c")"#);
|
test_parse_query_to_ast_helper(r#"field:(+a -"b c")"#, r#"(+"field":a -"field":"b c")"#);
|
||||||
|
test_parse_query_to_ast_helper(r#"field:(a AND "b c")"#, r#"(+"field":a +"field":"b c")"#);
|
||||||
|
test_parse_query_to_ast_helper(r#"field:(a OR "b c")"#, r#"(?"field":a ?"field":"b c")"#);
|
||||||
|
test_parse_query_to_ast_helper(
|
||||||
|
r#"field:(a OR (b AND c))"#,
|
||||||
|
r#"(?"field":a ?(+"field":b +"field":c))"#,
|
||||||
|
);
|
||||||
|
test_parse_query_to_ast_helper(
|
||||||
|
r#"field:(a [b TO c])"#,
|
||||||
|
r#"(*"field":a *"field":["b" TO "c"])"#,
|
||||||
|
);
|
||||||
|
|
||||||
test_is_parse_err(r#"field:(+a -"b c""#, r#"(+"field":a -"field":"b c")"#);
|
test_is_parse_err(r#"field:(+a -"b c""#, r#"(+"field":a -"field":"b c")"#);
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -44,6 +44,26 @@ impl UserInputLeaf {
|
|||||||
},
|
},
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub(crate) fn set_default_field(&mut self, default_field: String) {
|
||||||
|
match self {
|
||||||
|
UserInputLeaf::Literal(ref mut literal) if literal.field_name.is_none() => {
|
||||||
|
literal.field_name = Some(default_field)
|
||||||
|
}
|
||||||
|
UserInputLeaf::All => {
|
||||||
|
*self = UserInputLeaf::Exists {
|
||||||
|
field: default_field,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
UserInputLeaf::Range { ref mut field, .. } if field.is_none() => {
|
||||||
|
*field = Some(default_field)
|
||||||
|
}
|
||||||
|
UserInputLeaf::Set { ref mut field, .. } if field.is_none() => {
|
||||||
|
*field = Some(default_field)
|
||||||
|
}
|
||||||
|
_ => (), // field was already set, do nothing
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Debug for UserInputLeaf {
|
impl Debug for UserInputLeaf {
|
||||||
@@ -205,6 +225,16 @@ impl UserInputAst {
|
|||||||
pub fn or(asts: Vec<UserInputAst>) -> UserInputAst {
|
pub fn or(asts: Vec<UserInputAst>) -> UserInputAst {
|
||||||
UserInputAst::compose(Occur::Should, asts)
|
UserInputAst::compose(Occur::Should, asts)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub(crate) fn set_default_field(&mut self, field: String) {
|
||||||
|
match self {
|
||||||
|
UserInputAst::Clause(clauses) => clauses
|
||||||
|
.iter_mut()
|
||||||
|
.for_each(|(_, ast)| ast.set_default_field(field.clone())),
|
||||||
|
UserInputAst::Leaf(leaf) => leaf.set_default_field(field),
|
||||||
|
UserInputAst::Boost(ref mut ast, _) => ast.set_default_field(field),
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl From<UserInputLiteral> for UserInputLeaf {
|
impl From<UserInputLiteral> for UserInputLeaf {
|
||||||
|
|||||||
@@ -28,6 +28,7 @@ mod term_agg;
|
|||||||
mod term_missing_agg;
|
mod term_missing_agg;
|
||||||
|
|
||||||
use std::collections::HashMap;
|
use std::collections::HashMap;
|
||||||
|
use std::fmt;
|
||||||
|
|
||||||
pub use histogram::*;
|
pub use histogram::*;
|
||||||
pub use range::*;
|
pub use range::*;
|
||||||
@@ -72,12 +73,12 @@ impl From<&str> for OrderTarget {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl ToString for OrderTarget {
|
impl fmt::Display for OrderTarget {
|
||||||
fn to_string(&self) -> String {
|
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
|
||||||
match self {
|
match self {
|
||||||
OrderTarget::Key => "_key".to_string(),
|
OrderTarget::Key => f.write_str("_key"),
|
||||||
OrderTarget::Count => "_count".to_string(),
|
OrderTarget::Count => f.write_str("_count"),
|
||||||
OrderTarget::SubAggregation(agg) => agg.to_string(),
|
OrderTarget::SubAggregation(agg) => agg.fmt(f),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -2,6 +2,7 @@ use std::collections::HashMap;
|
|||||||
use std::net::Ipv6Addr;
|
use std::net::Ipv6Addr;
|
||||||
|
|
||||||
use columnar::{ColumnarReader, DynamicColumn};
|
use columnar::{ColumnarReader, DynamicColumn};
|
||||||
|
use common::json_path_writer::JSON_PATH_SEGMENT_SEP_STR;
|
||||||
use common::DateTime;
|
use common::DateTime;
|
||||||
use regex::Regex;
|
use regex::Regex;
|
||||||
use serde::ser::SerializeMap;
|
use serde::ser::SerializeMap;
|
||||||
@@ -15,7 +16,6 @@ use crate::aggregation::intermediate_agg_result::{
|
|||||||
use crate::aggregation::segment_agg_result::SegmentAggregationCollector;
|
use crate::aggregation::segment_agg_result::SegmentAggregationCollector;
|
||||||
use crate::aggregation::AggregationError;
|
use crate::aggregation::AggregationError;
|
||||||
use crate::collector::TopNComputer;
|
use crate::collector::TopNComputer;
|
||||||
use crate::schema::term::JSON_PATH_SEGMENT_SEP_STR;
|
|
||||||
use crate::schema::OwnedValue;
|
use crate::schema::OwnedValue;
|
||||||
use crate::{DocAddress, DocId, SegmentOrdinal};
|
use crate::{DocAddress, DocId, SegmentOrdinal};
|
||||||
|
|
||||||
|
|||||||
@@ -159,10 +159,6 @@ use itertools::Itertools;
|
|||||||
use serde::de::{self, Visitor};
|
use serde::de::{self, Visitor};
|
||||||
use serde::{Deserialize, Deserializer, Serialize};
|
use serde::{Deserialize, Deserializer, Serialize};
|
||||||
|
|
||||||
pub(crate) fn invalid_agg_request(message: String) -> crate::TantivyError {
|
|
||||||
crate::TantivyError::AggregationError(AggregationError::InvalidRequest(message))
|
|
||||||
}
|
|
||||||
|
|
||||||
fn parse_str_into_f64<E: de::Error>(value: &str) -> Result<f64, E> {
|
fn parse_str_into_f64<E: de::Error>(value: &str) -> Result<f64, E> {
|
||||||
let parsed = value.parse::<f64>().map_err(|_err| {
|
let parsed = value.parse::<f64>().map_err(|_err| {
|
||||||
de::Error::custom(format!("Failed to parse f64 from string: {:?}", value))
|
de::Error::custom(format!("Failed to parse f64 from string: {:?}", value))
|
||||||
|
|||||||
@@ -1,7 +1,7 @@
|
|||||||
use std::cmp::Ordering;
|
use std::cmp::Ordering;
|
||||||
use std::collections::{btree_map, BTreeMap, BTreeSet, BinaryHeap};
|
use std::collections::{btree_map, BTreeMap, BTreeSet, BinaryHeap};
|
||||||
|
use std::io;
|
||||||
use std::ops::Bound;
|
use std::ops::Bound;
|
||||||
use std::{io, u64, usize};
|
|
||||||
|
|
||||||
use crate::collector::{Collector, SegmentCollector};
|
use crate::collector::{Collector, SegmentCollector};
|
||||||
use crate::fastfield::FacetReader;
|
use crate::fastfield::FacetReader;
|
||||||
|
|||||||
@@ -1,12 +1,10 @@
|
|||||||
use columnar::MonotonicallyMappableToU64;
|
use common::json_path_writer::JSON_PATH_SEGMENT_SEP;
|
||||||
use common::{replace_in_place, JsonPathWriter};
|
use common::{replace_in_place, JsonPathWriter};
|
||||||
use rustc_hash::FxHashMap;
|
use rustc_hash::FxHashMap;
|
||||||
|
|
||||||
use crate::fastfield::FastValue;
|
|
||||||
use crate::postings::{IndexingContext, IndexingPosition, PostingsWriter};
|
use crate::postings::{IndexingContext, IndexingPosition, PostingsWriter};
|
||||||
use crate::schema::document::{ReferenceValue, ReferenceValueLeaf, Value};
|
use crate::schema::document::{ReferenceValue, ReferenceValueLeaf, Value};
|
||||||
use crate::schema::term::JSON_PATH_SEGMENT_SEP;
|
use crate::schema::Type;
|
||||||
use crate::schema::{Field, Type, DATE_TIME_PRECISION_INDEXED};
|
|
||||||
use crate::time::format_description::well_known::Rfc3339;
|
use crate::time::format_description::well_known::Rfc3339;
|
||||||
use crate::time::{OffsetDateTime, UtcOffset};
|
use crate::time::{OffsetDateTime, UtcOffset};
|
||||||
use crate::tokenizer::TextAnalyzer;
|
use crate::tokenizer::TextAnalyzer;
|
||||||
@@ -256,71 +254,45 @@ fn index_json_value<'a, V: Value<'a>>(
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// Tries to infer a JSON type from a string.
|
/// Tries to infer a JSON type from a string and append it to the term.
|
||||||
pub fn convert_to_fast_value_and_get_term(
|
///
|
||||||
json_term_writer: &mut JsonTermWriter,
|
/// The term must be json + JSON path.
|
||||||
|
pub(crate) fn convert_to_fast_value_and_append_to_json_term(
|
||||||
|
mut term: Term,
|
||||||
phrase: &str,
|
phrase: &str,
|
||||||
) -> Option<Term> {
|
) -> Option<Term> {
|
||||||
|
assert_eq!(
|
||||||
|
term.value()
|
||||||
|
.as_json_value_bytes()
|
||||||
|
.expect("expecting a Term with a json type and json path")
|
||||||
|
.as_serialized()
|
||||||
|
.len(),
|
||||||
|
0,
|
||||||
|
"JSON value bytes should be empty"
|
||||||
|
);
|
||||||
if let Ok(dt) = OffsetDateTime::parse(phrase, &Rfc3339) {
|
if let Ok(dt) = OffsetDateTime::parse(phrase, &Rfc3339) {
|
||||||
let dt_utc = dt.to_offset(UtcOffset::UTC);
|
let dt_utc = dt.to_offset(UtcOffset::UTC);
|
||||||
return Some(set_fastvalue_and_get_term(
|
term.append_type_and_fast_value(DateTime::from_utc(dt_utc));
|
||||||
json_term_writer,
|
return Some(term);
|
||||||
DateTime::from_utc(dt_utc),
|
|
||||||
));
|
|
||||||
}
|
}
|
||||||
if let Ok(i64_val) = str::parse::<i64>(phrase) {
|
if let Ok(i64_val) = str::parse::<i64>(phrase) {
|
||||||
return Some(set_fastvalue_and_get_term(json_term_writer, i64_val));
|
term.append_type_and_fast_value(i64_val);
|
||||||
|
return Some(term);
|
||||||
}
|
}
|
||||||
if let Ok(u64_val) = str::parse::<u64>(phrase) {
|
if let Ok(u64_val) = str::parse::<u64>(phrase) {
|
||||||
return Some(set_fastvalue_and_get_term(json_term_writer, u64_val));
|
term.append_type_and_fast_value(u64_val);
|
||||||
|
return Some(term);
|
||||||
}
|
}
|
||||||
if let Ok(f64_val) = str::parse::<f64>(phrase) {
|
if let Ok(f64_val) = str::parse::<f64>(phrase) {
|
||||||
return Some(set_fastvalue_and_get_term(json_term_writer, f64_val));
|
term.append_type_and_fast_value(f64_val);
|
||||||
|
return Some(term);
|
||||||
}
|
}
|
||||||
if let Ok(bool_val) = str::parse::<bool>(phrase) {
|
if let Ok(bool_val) = str::parse::<bool>(phrase) {
|
||||||
return Some(set_fastvalue_and_get_term(json_term_writer, bool_val));
|
term.append_type_and_fast_value(bool_val);
|
||||||
|
return Some(term);
|
||||||
}
|
}
|
||||||
None
|
None
|
||||||
}
|
}
|
||||||
// helper function to generate a Term from a json fastvalue
|
|
||||||
pub(crate) fn set_fastvalue_and_get_term<T: FastValue>(
|
|
||||||
json_term_writer: &mut JsonTermWriter,
|
|
||||||
value: T,
|
|
||||||
) -> Term {
|
|
||||||
json_term_writer.set_fast_value(value);
|
|
||||||
json_term_writer.term().clone()
|
|
||||||
}
|
|
||||||
|
|
||||||
// helper function to generate a list of terms with their positions from a textual json value
|
|
||||||
pub(crate) fn set_string_and_get_terms(
|
|
||||||
json_term_writer: &mut JsonTermWriter,
|
|
||||||
value: &str,
|
|
||||||
text_analyzer: &mut TextAnalyzer,
|
|
||||||
) -> Vec<(usize, Term)> {
|
|
||||||
let mut positions_and_terms = Vec::<(usize, Term)>::new();
|
|
||||||
json_term_writer.close_path_and_set_type(Type::Str);
|
|
||||||
let term_num_bytes = json_term_writer.term_buffer.len_bytes();
|
|
||||||
let mut token_stream = text_analyzer.token_stream(value);
|
|
||||||
token_stream.process(&mut |token| {
|
|
||||||
json_term_writer
|
|
||||||
.term_buffer
|
|
||||||
.truncate_value_bytes(term_num_bytes);
|
|
||||||
json_term_writer
|
|
||||||
.term_buffer
|
|
||||||
.append_bytes(token.text.as_bytes());
|
|
||||||
positions_and_terms.push((token.position, json_term_writer.term().clone()));
|
|
||||||
});
|
|
||||||
positions_and_terms
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Writes a value of a JSON field to a `Term`.
|
|
||||||
/// The Term format is as follows:
|
|
||||||
/// `[JSON_TYPE][JSON_PATH][JSON_END_OF_PATH][VALUE_BYTES]`
|
|
||||||
pub struct JsonTermWriter<'a> {
|
|
||||||
term_buffer: &'a mut Term,
|
|
||||||
path_stack: Vec<usize>,
|
|
||||||
expand_dots_enabled: bool,
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Splits a json path supplied to the query parser in such a way that
|
/// Splits a json path supplied to the query parser in such a way that
|
||||||
/// `.` can be escaped.
|
/// `.` can be escaped.
|
||||||
@@ -377,158 +349,48 @@ pub(crate) fn encode_column_name(
|
|||||||
path.into()
|
path.into()
|
||||||
}
|
}
|
||||||
|
|
||||||
impl<'a> JsonTermWriter<'a> {
|
|
||||||
pub fn from_field_and_json_path(
|
|
||||||
field: Field,
|
|
||||||
json_path: &str,
|
|
||||||
expand_dots_enabled: bool,
|
|
||||||
term_buffer: &'a mut Term,
|
|
||||||
) -> Self {
|
|
||||||
term_buffer.set_field_and_type(field, Type::Json);
|
|
||||||
let mut json_term_writer = Self::wrap(term_buffer, expand_dots_enabled);
|
|
||||||
for segment in split_json_path(json_path) {
|
|
||||||
json_term_writer.push_path_segment(&segment);
|
|
||||||
}
|
|
||||||
json_term_writer
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn wrap(term_buffer: &'a mut Term, expand_dots_enabled: bool) -> Self {
|
|
||||||
term_buffer.clear_with_type(Type::Json);
|
|
||||||
let mut path_stack = Vec::with_capacity(10);
|
|
||||||
path_stack.push(0);
|
|
||||||
Self {
|
|
||||||
term_buffer,
|
|
||||||
path_stack,
|
|
||||||
expand_dots_enabled,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn trim_to_end_of_path(&mut self) {
|
|
||||||
let end_of_path = *self.path_stack.last().unwrap();
|
|
||||||
self.term_buffer.truncate_value_bytes(end_of_path);
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn close_path_and_set_type(&mut self, typ: Type) {
|
|
||||||
self.trim_to_end_of_path();
|
|
||||||
self.term_buffer.set_json_path_end();
|
|
||||||
self.term_buffer.append_bytes(&[typ.to_code()]);
|
|
||||||
}
|
|
||||||
|
|
||||||
// TODO: Remove this function and use JsonPathWriter instead.
|
|
||||||
pub fn push_path_segment(&mut self, segment: &str) {
|
|
||||||
// the path stack should never be empty.
|
|
||||||
self.trim_to_end_of_path();
|
|
||||||
|
|
||||||
if self.path_stack.len() > 1 {
|
|
||||||
self.term_buffer.set_json_path_separator();
|
|
||||||
}
|
|
||||||
let appended_segment = self.term_buffer.append_bytes(segment.as_bytes());
|
|
||||||
if self.expand_dots_enabled {
|
|
||||||
// We need to replace `.` by JSON_PATH_SEGMENT_SEP.
|
|
||||||
replace_in_place(b'.', JSON_PATH_SEGMENT_SEP, appended_segment);
|
|
||||||
}
|
|
||||||
self.term_buffer.add_json_path_separator();
|
|
||||||
self.path_stack.push(self.term_buffer.len_bytes());
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn pop_path_segment(&mut self) {
|
|
||||||
self.path_stack.pop();
|
|
||||||
assert!(!self.path_stack.is_empty());
|
|
||||||
self.trim_to_end_of_path();
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Returns the json path of the term being currently built.
|
|
||||||
#[cfg(test)]
|
|
||||||
pub(crate) fn path(&self) -> &[u8] {
|
|
||||||
let end_of_path = self.path_stack.last().cloned().unwrap_or(1);
|
|
||||||
&self.term().serialized_value_bytes()[..end_of_path - 1]
|
|
||||||
}
|
|
||||||
|
|
||||||
pub(crate) fn set_fast_value<T: FastValue>(&mut self, val: T) {
|
|
||||||
self.close_path_and_set_type(T::to_type());
|
|
||||||
let value = if T::to_type() == Type::Date {
|
|
||||||
DateTime::from_u64(val.to_u64())
|
|
||||||
.truncate(DATE_TIME_PRECISION_INDEXED)
|
|
||||||
.to_u64()
|
|
||||||
} else {
|
|
||||||
val.to_u64()
|
|
||||||
};
|
|
||||||
self.term_buffer
|
|
||||||
.append_bytes(value.to_be_bytes().as_slice());
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn set_str(&mut self, text: &str) {
|
|
||||||
self.close_path_and_set_type(Type::Str);
|
|
||||||
self.term_buffer.append_bytes(text.as_bytes());
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn term(&self) -> &Term {
|
|
||||||
self.term_buffer
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod tests {
|
mod tests {
|
||||||
use super::{split_json_path, JsonTermWriter};
|
use super::split_json_path;
|
||||||
use crate::schema::{Field, Type};
|
use crate::schema::Field;
|
||||||
use crate::Term;
|
use crate::Term;
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_json_writer() {
|
fn test_json_writer() {
|
||||||
let field = Field::from_field_id(1);
|
let field = Field::from_field_id(1);
|
||||||
let mut term = Term::with_type_and_field(Type::Json, field);
|
|
||||||
let mut json_writer = JsonTermWriter::wrap(&mut term, false);
|
let mut term = Term::from_field_json_path(field, "attributes.color", false);
|
||||||
json_writer.push_path_segment("attributes");
|
term.append_type_and_str("red");
|
||||||
json_writer.push_path_segment("color");
|
|
||||||
json_writer.set_str("red");
|
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
format!("{:?}", json_writer.term()),
|
format!("{:?}", term),
|
||||||
"Term(field=1, type=Json, path=attributes.color, type=Str, \"red\")"
|
"Term(field=1, type=Json, path=attributes.color, type=Str, \"red\")"
|
||||||
);
|
);
|
||||||
json_writer.set_str("blue");
|
|
||||||
|
let mut term = Term::from_field_json_path(field, "attributes.dimensions.width", false);
|
||||||
|
term.append_type_and_fast_value(400i64);
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
format!("{:?}", json_writer.term()),
|
format!("{:?}", term),
|
||||||
"Term(field=1, type=Json, path=attributes.color, type=Str, \"blue\")"
|
|
||||||
);
|
|
||||||
json_writer.pop_path_segment();
|
|
||||||
json_writer.push_path_segment("dimensions");
|
|
||||||
json_writer.push_path_segment("width");
|
|
||||||
json_writer.set_fast_value(400i64);
|
|
||||||
assert_eq!(
|
|
||||||
format!("{:?}", json_writer.term()),
|
|
||||||
"Term(field=1, type=Json, path=attributes.dimensions.width, type=I64, 400)"
|
"Term(field=1, type=Json, path=attributes.dimensions.width, type=I64, 400)"
|
||||||
);
|
);
|
||||||
json_writer.pop_path_segment();
|
|
||||||
json_writer.push_path_segment("height");
|
|
||||||
json_writer.set_fast_value(300i64);
|
|
||||||
assert_eq!(
|
|
||||||
format!("{:?}", json_writer.term()),
|
|
||||||
"Term(field=1, type=Json, path=attributes.dimensions.height, type=I64, 300)"
|
|
||||||
);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_string_term() {
|
fn test_string_term() {
|
||||||
let field = Field::from_field_id(1);
|
let field = Field::from_field_id(1);
|
||||||
let mut term = Term::with_type_and_field(Type::Json, field);
|
let mut term = Term::from_field_json_path(field, "color", false);
|
||||||
let mut json_writer = JsonTermWriter::wrap(&mut term, false);
|
term.append_type_and_str("red");
|
||||||
json_writer.push_path_segment("color");
|
|
||||||
json_writer.set_str("red");
|
assert_eq!(term.serialized_term(), b"\x00\x00\x00\x01jcolor\x00sred")
|
||||||
assert_eq!(
|
|
||||||
json_writer.term().serialized_term(),
|
|
||||||
b"\x00\x00\x00\x01jcolor\x00sred"
|
|
||||||
)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_i64_term() {
|
fn test_i64_term() {
|
||||||
let field = Field::from_field_id(1);
|
let field = Field::from_field_id(1);
|
||||||
let mut term = Term::with_type_and_field(Type::Json, field);
|
let mut term = Term::from_field_json_path(field, "color", false);
|
||||||
let mut json_writer = JsonTermWriter::wrap(&mut term, false);
|
term.append_type_and_fast_value(-4i64);
|
||||||
json_writer.push_path_segment("color");
|
|
||||||
json_writer.set_fast_value(-4i64);
|
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
json_writer.term().serialized_term(),
|
term.serialized_term(),
|
||||||
b"\x00\x00\x00\x01jcolor\x00i\x7f\xff\xff\xff\xff\xff\xff\xfc"
|
b"\x00\x00\x00\x01jcolor\x00i\x7f\xff\xff\xff\xff\xff\xff\xfc"
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
@@ -536,12 +398,11 @@ mod tests {
|
|||||||
#[test]
|
#[test]
|
||||||
fn test_u64_term() {
|
fn test_u64_term() {
|
||||||
let field = Field::from_field_id(1);
|
let field = Field::from_field_id(1);
|
||||||
let mut term = Term::with_type_and_field(Type::Json, field);
|
let mut term = Term::from_field_json_path(field, "color", false);
|
||||||
let mut json_writer = JsonTermWriter::wrap(&mut term, false);
|
term.append_type_and_fast_value(4u64);
|
||||||
json_writer.push_path_segment("color");
|
|
||||||
json_writer.set_fast_value(4u64);
|
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
json_writer.term().serialized_term(),
|
term.serialized_term(),
|
||||||
b"\x00\x00\x00\x01jcolor\x00u\x00\x00\x00\x00\x00\x00\x00\x04"
|
b"\x00\x00\x00\x01jcolor\x00u\x00\x00\x00\x00\x00\x00\x00\x04"
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
@@ -549,12 +410,10 @@ mod tests {
|
|||||||
#[test]
|
#[test]
|
||||||
fn test_f64_term() {
|
fn test_f64_term() {
|
||||||
let field = Field::from_field_id(1);
|
let field = Field::from_field_id(1);
|
||||||
let mut term = Term::with_type_and_field(Type::Json, field);
|
let mut term = Term::from_field_json_path(field, "color", false);
|
||||||
let mut json_writer = JsonTermWriter::wrap(&mut term, false);
|
term.append_type_and_fast_value(4.0f64);
|
||||||
json_writer.push_path_segment("color");
|
|
||||||
json_writer.set_fast_value(4.0f64);
|
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
json_writer.term().serialized_term(),
|
term.serialized_term(),
|
||||||
b"\x00\x00\x00\x01jcolor\x00f\xc0\x10\x00\x00\x00\x00\x00\x00"
|
b"\x00\x00\x00\x01jcolor\x00f\xc0\x10\x00\x00\x00\x00\x00\x00"
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
@@ -562,90 +421,14 @@ mod tests {
|
|||||||
#[test]
|
#[test]
|
||||||
fn test_bool_term() {
|
fn test_bool_term() {
|
||||||
let field = Field::from_field_id(1);
|
let field = Field::from_field_id(1);
|
||||||
let mut term = Term::with_type_and_field(Type::Json, field);
|
let mut term = Term::from_field_json_path(field, "color", false);
|
||||||
let mut json_writer = JsonTermWriter::wrap(&mut term, false);
|
term.append_type_and_fast_value(true);
|
||||||
json_writer.push_path_segment("color");
|
|
||||||
json_writer.set_fast_value(true);
|
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
json_writer.term().serialized_term(),
|
term.serialized_term(),
|
||||||
b"\x00\x00\x00\x01jcolor\x00o\x00\x00\x00\x00\x00\x00\x00\x01"
|
b"\x00\x00\x00\x01jcolor\x00o\x00\x00\x00\x00\x00\x00\x00\x01"
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_push_after_set_path_segment() {
|
|
||||||
let field = Field::from_field_id(1);
|
|
||||||
let mut term = Term::with_type_and_field(Type::Json, field);
|
|
||||||
let mut json_writer = JsonTermWriter::wrap(&mut term, false);
|
|
||||||
json_writer.push_path_segment("attribute");
|
|
||||||
json_writer.set_str("something");
|
|
||||||
json_writer.push_path_segment("color");
|
|
||||||
json_writer.set_str("red");
|
|
||||||
assert_eq!(
|
|
||||||
json_writer.term().serialized_term(),
|
|
||||||
b"\x00\x00\x00\x01jattribute\x01color\x00sred"
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_pop_segment() {
|
|
||||||
let field = Field::from_field_id(1);
|
|
||||||
let mut term = Term::with_type_and_field(Type::Json, field);
|
|
||||||
let mut json_writer = JsonTermWriter::wrap(&mut term, false);
|
|
||||||
json_writer.push_path_segment("color");
|
|
||||||
json_writer.push_path_segment("hue");
|
|
||||||
json_writer.pop_path_segment();
|
|
||||||
json_writer.set_str("red");
|
|
||||||
assert_eq!(
|
|
||||||
json_writer.term().serialized_term(),
|
|
||||||
b"\x00\x00\x00\x01jcolor\x00sred"
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_json_writer_path() {
|
|
||||||
let field = Field::from_field_id(1);
|
|
||||||
let mut term = Term::with_type_and_field(Type::Json, field);
|
|
||||||
let mut json_writer = JsonTermWriter::wrap(&mut term, false);
|
|
||||||
json_writer.push_path_segment("color");
|
|
||||||
assert_eq!(json_writer.path(), b"color");
|
|
||||||
json_writer.push_path_segment("hue");
|
|
||||||
assert_eq!(json_writer.path(), b"color\x01hue");
|
|
||||||
json_writer.set_str("pink");
|
|
||||||
assert_eq!(json_writer.path(), b"color\x01hue");
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_json_path_expand_dots_disabled() {
|
|
||||||
let field = Field::from_field_id(1);
|
|
||||||
let mut term = Term::with_type_and_field(Type::Json, field);
|
|
||||||
let mut json_writer = JsonTermWriter::wrap(&mut term, false);
|
|
||||||
json_writer.push_path_segment("color.hue");
|
|
||||||
assert_eq!(json_writer.path(), b"color.hue");
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_json_path_expand_dots_enabled() {
|
|
||||||
let field = Field::from_field_id(1);
|
|
||||||
let mut term = Term::with_type_and_field(Type::Json, field);
|
|
||||||
let mut json_writer = JsonTermWriter::wrap(&mut term, true);
|
|
||||||
json_writer.push_path_segment("color.hue");
|
|
||||||
assert_eq!(json_writer.path(), b"color\x01hue");
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_json_path_expand_dots_enabled_pop_segment() {
|
|
||||||
let field = Field::from_field_id(1);
|
|
||||||
let mut term = Term::with_type_and_field(Type::Json, field);
|
|
||||||
let mut json_writer = JsonTermWriter::wrap(&mut term, true);
|
|
||||||
json_writer.push_path_segment("hello");
|
|
||||||
assert_eq!(json_writer.path(), b"hello");
|
|
||||||
json_writer.push_path_segment("color.hue");
|
|
||||||
assert_eq!(json_writer.path(), b"hello\x01color\x01hue");
|
|
||||||
json_writer.pop_path_segment();
|
|
||||||
assert_eq!(json_writer.path(), b"hello");
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_split_json_path_simple() {
|
fn test_split_json_path_simple() {
|
||||||
let json_path = split_json_path("titi.toto");
|
let json_path = split_json_path("titi.toto");
|
||||||
|
|||||||
@@ -1,9 +1,8 @@
|
|||||||
use crate::collector::Count;
|
use crate::collector::Count;
|
||||||
use crate::directory::{RamDirectory, WatchCallback};
|
use crate::directory::{RamDirectory, WatchCallback};
|
||||||
use crate::indexer::{LogMergePolicy, NoMergePolicy};
|
use crate::indexer::{LogMergePolicy, NoMergePolicy};
|
||||||
use crate::json_utils::JsonTermWriter;
|
|
||||||
use crate::query::TermQuery;
|
use crate::query::TermQuery;
|
||||||
use crate::schema::{Field, IndexRecordOption, Schema, Type, INDEXED, STRING, TEXT};
|
use crate::schema::{Field, IndexRecordOption, Schema, INDEXED, STRING, TEXT};
|
||||||
use crate::tokenizer::TokenizerManager;
|
use crate::tokenizer::TokenizerManager;
|
||||||
use crate::{
|
use crate::{
|
||||||
Directory, DocSet, Index, IndexBuilder, IndexReader, IndexSettings, IndexWriter, Postings,
|
Directory, DocSet, Index, IndexBuilder, IndexReader, IndexSettings, IndexWriter, Postings,
|
||||||
@@ -416,16 +415,12 @@ fn test_non_text_json_term_freq() {
|
|||||||
let searcher = reader.searcher();
|
let searcher = reader.searcher();
|
||||||
let segment_reader = searcher.segment_reader(0u32);
|
let segment_reader = searcher.segment_reader(0u32);
|
||||||
let inv_idx = segment_reader.inverted_index(field).unwrap();
|
let inv_idx = segment_reader.inverted_index(field).unwrap();
|
||||||
let mut term = Term::with_type_and_field(Type::Json, field);
|
|
||||||
let mut json_term_writer = JsonTermWriter::wrap(&mut term, false);
|
let mut term = Term::from_field_json_path(field, "tenant_id", false);
|
||||||
json_term_writer.push_path_segment("tenant_id");
|
term.append_type_and_fast_value(75u64);
|
||||||
json_term_writer.close_path_and_set_type(Type::U64);
|
|
||||||
json_term_writer.set_fast_value(75u64);
|
|
||||||
let postings = inv_idx
|
let postings = inv_idx
|
||||||
.read_postings(
|
.read_postings(&term, IndexRecordOption::WithFreqsAndPositions)
|
||||||
json_term_writer.term(),
|
|
||||||
IndexRecordOption::WithFreqsAndPositions,
|
|
||||||
)
|
|
||||||
.unwrap()
|
.unwrap()
|
||||||
.unwrap();
|
.unwrap();
|
||||||
assert_eq!(postings.doc(), 0);
|
assert_eq!(postings.doc(), 0);
|
||||||
@@ -454,16 +449,12 @@ fn test_non_text_json_term_freq_bitpacked() {
|
|||||||
let searcher = reader.searcher();
|
let searcher = reader.searcher();
|
||||||
let segment_reader = searcher.segment_reader(0u32);
|
let segment_reader = searcher.segment_reader(0u32);
|
||||||
let inv_idx = segment_reader.inverted_index(field).unwrap();
|
let inv_idx = segment_reader.inverted_index(field).unwrap();
|
||||||
let mut term = Term::with_type_and_field(Type::Json, field);
|
|
||||||
let mut json_term_writer = JsonTermWriter::wrap(&mut term, false);
|
let mut term = Term::from_field_json_path(field, "tenant_id", false);
|
||||||
json_term_writer.push_path_segment("tenant_id");
|
term.append_type_and_fast_value(75u64);
|
||||||
json_term_writer.close_path_and_set_type(Type::U64);
|
|
||||||
json_term_writer.set_fast_value(75u64);
|
|
||||||
let mut postings = inv_idx
|
let mut postings = inv_idx
|
||||||
.read_postings(
|
.read_postings(&term, IndexRecordOption::WithFreqsAndPositions)
|
||||||
json_term_writer.term(),
|
|
||||||
IndexRecordOption::WithFreqsAndPositions,
|
|
||||||
)
|
|
||||||
.unwrap()
|
.unwrap()
|
||||||
.unwrap();
|
.unwrap();
|
||||||
assert_eq!(postings.doc(), 0);
|
assert_eq!(postings.doc(), 0);
|
||||||
|
|||||||
@@ -1,9 +1,12 @@
|
|||||||
|
#![allow(deprecated)] // Remove with index sorting
|
||||||
|
|
||||||
use std::collections::HashSet;
|
use std::collections::HashSet;
|
||||||
|
|
||||||
use rand::{thread_rng, Rng};
|
use rand::{thread_rng, Rng};
|
||||||
|
|
||||||
use crate::indexer::index_writer::MEMORY_BUDGET_NUM_BYTES_MIN;
|
use crate::indexer::index_writer::MEMORY_BUDGET_NUM_BYTES_MIN;
|
||||||
use crate::schema::*;
|
use crate::schema::*;
|
||||||
|
#[allow(deprecated)]
|
||||||
use crate::{doc, schema, Index, IndexSettings, IndexSortByField, IndexWriter, Order, Searcher};
|
use crate::{doc, schema, Index, IndexSettings, IndexSortByField, IndexWriter, Order, Searcher};
|
||||||
|
|
||||||
fn check_index_content(searcher: &Searcher, vals: &[u64]) -> crate::Result<()> {
|
fn check_index_content(searcher: &Searcher, vals: &[u64]) -> crate::Result<()> {
|
||||||
|
|||||||
@@ -20,7 +20,7 @@ use crate::indexer::segment_updater::save_metas;
|
|||||||
use crate::indexer::{IndexWriter, SingleSegmentIndexWriter};
|
use crate::indexer::{IndexWriter, SingleSegmentIndexWriter};
|
||||||
use crate::reader::{IndexReader, IndexReaderBuilder};
|
use crate::reader::{IndexReader, IndexReaderBuilder};
|
||||||
use crate::schema::document::Document;
|
use crate::schema::document::Document;
|
||||||
use crate::schema::{Field, FieldType, Schema};
|
use crate::schema::{Field, FieldType, Schema, Type};
|
||||||
use crate::tokenizer::{TextAnalyzer, TokenizerManager};
|
use crate::tokenizer::{TextAnalyzer, TokenizerManager};
|
||||||
use crate::SegmentReader;
|
use crate::SegmentReader;
|
||||||
|
|
||||||
@@ -248,6 +248,15 @@ impl IndexBuilder {
|
|||||||
sort_by_field.field
|
sort_by_field.field
|
||||||
)));
|
)));
|
||||||
}
|
}
|
||||||
|
let supported_field_types = [Type::I64, Type::U64, Type::F64, Type::Date];
|
||||||
|
let field_type = entry.field_type().value_type();
|
||||||
|
if !supported_field_types.contains(&field_type) {
|
||||||
|
return Err(TantivyError::InvalidArgument(format!(
|
||||||
|
"Unsupported field type in sort_by_field: {:?}. Supported field types: \
|
||||||
|
{:?} ",
|
||||||
|
field_type, supported_field_types,
|
||||||
|
)));
|
||||||
|
}
|
||||||
}
|
}
|
||||||
Ok(())
|
Ok(())
|
||||||
} else {
|
} else {
|
||||||
|
|||||||
@@ -1,12 +1,13 @@
|
|||||||
use std::io;
|
use std::io;
|
||||||
|
|
||||||
|
use common::json_path_writer::JSON_END_OF_PATH;
|
||||||
use common::BinarySerializable;
|
use common::BinarySerializable;
|
||||||
use fnv::FnvHashSet;
|
use fnv::FnvHashSet;
|
||||||
|
|
||||||
use crate::directory::FileSlice;
|
use crate::directory::FileSlice;
|
||||||
use crate::positions::PositionReader;
|
use crate::positions::PositionReader;
|
||||||
use crate::postings::{BlockSegmentPostings, SegmentPostings, TermInfo};
|
use crate::postings::{BlockSegmentPostings, SegmentPostings, TermInfo};
|
||||||
use crate::schema::{IndexRecordOption, Term, Type, JSON_END_OF_PATH};
|
use crate::schema::{IndexRecordOption, Term, Type};
|
||||||
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
|
||||||
|
|||||||
@@ -406,7 +406,7 @@ impl SegmentReader {
|
|||||||
}
|
}
|
||||||
|
|
||||||
/// Returns an iterator that will iterate over the alive document ids
|
/// Returns an iterator that will iterate over the alive document ids
|
||||||
pub fn doc_ids_alive(&self) -> Box<dyn Iterator<Item = DocId> + '_> {
|
pub fn doc_ids_alive(&self) -> Box<dyn Iterator<Item = DocId> + Send + '_> {
|
||||||
if let Some(alive_bitset) = &self.alive_bitset_opt {
|
if let Some(alive_bitset) = &self.alive_bitset_opt {
|
||||||
Box::new(alive_bitset.iter_alive())
|
Box::new(alive_bitset.iter_alive())
|
||||||
} else {
|
} else {
|
||||||
|
|||||||
@@ -159,7 +159,7 @@ mod tests_indexsorting {
|
|||||||
use crate::indexer::NoMergePolicy;
|
use crate::indexer::NoMergePolicy;
|
||||||
use crate::query::QueryParser;
|
use crate::query::QueryParser;
|
||||||
use crate::schema::*;
|
use crate::schema::*;
|
||||||
use crate::{DocAddress, Index, IndexSettings, IndexSortByField, Order};
|
use crate::{DocAddress, Index, IndexBuilder, IndexSettings, IndexSortByField, Order};
|
||||||
|
|
||||||
fn create_test_index(
|
fn create_test_index(
|
||||||
index_settings: Option<IndexSettings>,
|
index_settings: Option<IndexSettings>,
|
||||||
@@ -557,4 +557,28 @@ mod tests_indexsorting {
|
|||||||
&[2000, 8000, 3000]
|
&[2000, 8000, 3000]
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_text_sort() -> crate::Result<()> {
|
||||||
|
let mut schema_builder = SchemaBuilder::new();
|
||||||
|
schema_builder.add_text_field("id", STRING | FAST | STORED);
|
||||||
|
schema_builder.add_text_field("name", TEXT | STORED);
|
||||||
|
|
||||||
|
let resp = IndexBuilder::new()
|
||||||
|
.schema(schema_builder.build())
|
||||||
|
.settings(IndexSettings {
|
||||||
|
sort_by_field: Some(IndexSortByField {
|
||||||
|
field: "id".to_string(),
|
||||||
|
order: Order::Asc,
|
||||||
|
}),
|
||||||
|
..Default::default()
|
||||||
|
})
|
||||||
|
.create_in_ram();
|
||||||
|
assert!(resp
|
||||||
|
.unwrap_err()
|
||||||
|
.to_string()
|
||||||
|
.contains("Unsupported field type"));
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -156,16 +156,24 @@ mod tests_mmap {
|
|||||||
}
|
}
|
||||||
#[test]
|
#[test]
|
||||||
fn test_json_field_1byte() {
|
fn test_json_field_1byte() {
|
||||||
// Test when field name contains a 1 byte, which has special meaning in tantivy.
|
// Test when field name contains a '1' byte, which has special meaning in tantivy.
|
||||||
|
// The 1 byte can be addressed as '1' byte or '.'.
|
||||||
let field_name_in = "\u{0001}";
|
let field_name_in = "\u{0001}";
|
||||||
let field_name_out = "\u{0001}";
|
let field_name_out = "\u{0001}";
|
||||||
test_json_field_name(field_name_in, field_name_out);
|
test_json_field_name(field_name_in, field_name_out);
|
||||||
|
|
||||||
// Test when field name contains a 1 byte, which has special meaning in tantivy.
|
// Test when field name contains a '1' byte, which has special meaning in tantivy.
|
||||||
let field_name_in = "\u{0001}";
|
let field_name_in = "\u{0001}";
|
||||||
let field_name_out = ".";
|
let field_name_out = ".";
|
||||||
test_json_field_name(field_name_in, field_name_out);
|
test_json_field_name(field_name_in, field_name_out);
|
||||||
}
|
}
|
||||||
|
#[test]
|
||||||
|
fn test_json_field_dot() {
|
||||||
|
// Test when field name contains a '.'
|
||||||
|
let field_name_in = ".";
|
||||||
|
let field_name_out = ".";
|
||||||
|
test_json_field_name(field_name_in, field_name_out);
|
||||||
|
}
|
||||||
fn test_json_field_name(field_name_in: &str, field_name_out: &str) {
|
fn test_json_field_name(field_name_in: &str, field_name_out: &str) {
|
||||||
let mut schema_builder = Schema::builder();
|
let mut schema_builder = Schema::builder();
|
||||||
|
|
||||||
@@ -205,10 +213,10 @@ mod tests_mmap {
|
|||||||
let reader = index.reader().unwrap();
|
let reader = index.reader().unwrap();
|
||||||
let searcher = reader.searcher();
|
let searcher = reader.searcher();
|
||||||
let parse_query = QueryParser::for_index(&index, Vec::new());
|
let parse_query = QueryParser::for_index(&index, Vec::new());
|
||||||
let test_query = |field_name: &str| {
|
let test_query = |query_str: &str| {
|
||||||
let query = parse_query.parse_query(field_name).unwrap();
|
let query = parse_query.parse_query(query_str).unwrap();
|
||||||
let num_docs = searcher.search(&query, &Count).unwrap();
|
let num_docs = searcher.search(&query, &Count).unwrap();
|
||||||
assert_eq!(num_docs, 1);
|
assert_eq!(num_docs, 1, "{}", query_str);
|
||||||
};
|
};
|
||||||
test_query(format!("json.{field_name_out}:test1").as_str());
|
test_query(format!("json.{field_name_out}:test1").as_str());
|
||||||
test_query(format!("json.a{field_name_out}:test2").as_str());
|
test_query(format!("json.a{field_name_out}:test2").as_str());
|
||||||
|
|||||||
@@ -496,14 +496,13 @@ mod tests {
|
|||||||
use tempfile::TempDir;
|
use tempfile::TempDir;
|
||||||
|
|
||||||
use crate::collector::{Count, TopDocs};
|
use crate::collector::{Count, TopDocs};
|
||||||
use crate::core::json_utils::JsonTermWriter;
|
|
||||||
use crate::directory::RamDirectory;
|
use crate::directory::RamDirectory;
|
||||||
|
use crate::fastfield::FastValue;
|
||||||
use crate::postings::TermInfo;
|
use crate::postings::TermInfo;
|
||||||
use crate::query::{PhraseQuery, QueryParser};
|
use crate::query::{PhraseQuery, QueryParser};
|
||||||
use crate::schema::document::Value;
|
use crate::schema::document::Value;
|
||||||
use crate::schema::{
|
use crate::schema::{
|
||||||
Document, IndexRecordOption, Schema, TextFieldIndexing, TextOptions, Type, STORED, STRING,
|
Document, IndexRecordOption, Schema, TextFieldIndexing, TextOptions, STORED, STRING, TEXT,
|
||||||
TEXT,
|
|
||||||
};
|
};
|
||||||
use crate::store::{Compressor, StoreReader, StoreWriter};
|
use crate::store::{Compressor, StoreReader, StoreWriter};
|
||||||
use crate::time::format_description::well_known::Rfc3339;
|
use crate::time::format_description::well_known::Rfc3339;
|
||||||
@@ -645,115 +644,115 @@ mod tests {
|
|||||||
let inv_idx = segment_reader.inverted_index(json_field).unwrap();
|
let inv_idx = segment_reader.inverted_index(json_field).unwrap();
|
||||||
let term_dict = inv_idx.terms();
|
let term_dict = inv_idx.terms();
|
||||||
|
|
||||||
let mut term = Term::with_type_and_field(Type::Json, json_field);
|
|
||||||
let mut term_stream = term_dict.stream().unwrap();
|
let mut term_stream = term_dict.stream().unwrap();
|
||||||
|
|
||||||
let mut json_term_writer = JsonTermWriter::wrap(&mut term, false);
|
let term_from_path =
|
||||||
|
|path: &str| -> Term { Term::from_field_json_path(json_field, path, false) };
|
||||||
|
|
||||||
json_term_writer.push_path_segment("bool");
|
fn set_fast_val<T: FastValue>(val: T, mut term: Term) -> Term {
|
||||||
json_term_writer.set_fast_value(true);
|
term.append_type_and_fast_value(val);
|
||||||
|
term
|
||||||
|
}
|
||||||
|
fn set_str(val: &str, mut term: Term) -> Term {
|
||||||
|
term.append_type_and_str(val);
|
||||||
|
term
|
||||||
|
}
|
||||||
|
|
||||||
|
let term = term_from_path("bool");
|
||||||
assert!(term_stream.advance());
|
assert!(term_stream.advance());
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
term_stream.key(),
|
term_stream.key(),
|
||||||
json_term_writer.term().serialized_value_bytes()
|
set_fast_val(true, term).serialized_value_bytes()
|
||||||
);
|
);
|
||||||
|
|
||||||
json_term_writer.pop_path_segment();
|
let term = term_from_path("complexobject.field\\.with\\.dot");
|
||||||
json_term_writer.push_path_segment("complexobject");
|
|
||||||
json_term_writer.push_path_segment("field.with.dot");
|
|
||||||
json_term_writer.set_fast_value(1i64);
|
|
||||||
assert!(term_stream.advance());
|
assert!(term_stream.advance());
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
term_stream.key(),
|
term_stream.key(),
|
||||||
json_term_writer.term().serialized_value_bytes()
|
set_fast_val(1i64, term).serialized_value_bytes()
|
||||||
);
|
);
|
||||||
|
|
||||||
json_term_writer.pop_path_segment();
|
// Date
|
||||||
json_term_writer.pop_path_segment();
|
let term = term_from_path("date");
|
||||||
json_term_writer.push_path_segment("date");
|
|
||||||
json_term_writer.set_fast_value(DateTime::from_utc(
|
|
||||||
OffsetDateTime::parse("1985-04-12T23:20:50.52Z", &Rfc3339).unwrap(),
|
|
||||||
));
|
|
||||||
assert!(term_stream.advance());
|
assert!(term_stream.advance());
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
term_stream.key(),
|
term_stream.key(),
|
||||||
json_term_writer.term().serialized_value_bytes()
|
set_fast_val(
|
||||||
|
DateTime::from_utc(
|
||||||
|
OffsetDateTime::parse("1985-04-12T23:20:50.52Z", &Rfc3339).unwrap(),
|
||||||
|
),
|
||||||
|
term
|
||||||
|
)
|
||||||
|
.serialized_value_bytes()
|
||||||
);
|
);
|
||||||
|
|
||||||
json_term_writer.pop_path_segment();
|
// Float
|
||||||
json_term_writer.push_path_segment("float");
|
let term = term_from_path("float");
|
||||||
json_term_writer.set_fast_value(-0.2f64);
|
|
||||||
assert!(term_stream.advance());
|
assert!(term_stream.advance());
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
term_stream.key(),
|
term_stream.key(),
|
||||||
json_term_writer.term().serialized_value_bytes()
|
set_fast_val(-0.2f64, term).serialized_value_bytes()
|
||||||
);
|
);
|
||||||
|
|
||||||
json_term_writer.pop_path_segment();
|
// Number In Array
|
||||||
json_term_writer.push_path_segment("my_arr");
|
let term = term_from_path("my_arr");
|
||||||
json_term_writer.set_fast_value(2i64);
|
|
||||||
assert!(term_stream.advance());
|
assert!(term_stream.advance());
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
term_stream.key(),
|
term_stream.key(),
|
||||||
json_term_writer.term().serialized_value_bytes()
|
set_fast_val(2i64, term).serialized_value_bytes()
|
||||||
);
|
);
|
||||||
|
|
||||||
json_term_writer.set_fast_value(3i64);
|
let term = term_from_path("my_arr");
|
||||||
assert!(term_stream.advance());
|
assert!(term_stream.advance());
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
term_stream.key(),
|
term_stream.key(),
|
||||||
json_term_writer.term().serialized_value_bytes()
|
set_fast_val(3i64, term).serialized_value_bytes()
|
||||||
);
|
);
|
||||||
|
|
||||||
json_term_writer.set_fast_value(4i64);
|
let term = term_from_path("my_arr");
|
||||||
assert!(term_stream.advance());
|
assert!(term_stream.advance());
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
term_stream.key(),
|
term_stream.key(),
|
||||||
json_term_writer.term().serialized_value_bytes()
|
set_fast_val(4i64, term).serialized_value_bytes()
|
||||||
);
|
);
|
||||||
|
|
||||||
json_term_writer.push_path_segment("my_key");
|
// El in Array
|
||||||
json_term_writer.set_str("tokens");
|
let term = term_from_path("my_arr.my_key");
|
||||||
assert!(term_stream.advance());
|
assert!(term_stream.advance());
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
term_stream.key(),
|
term_stream.key(),
|
||||||
json_term_writer.term().serialized_value_bytes()
|
set_str("tokens", term).serialized_value_bytes()
|
||||||
);
|
);
|
||||||
|
let term = term_from_path("my_arr.my_key");
|
||||||
json_term_writer.set_str("two");
|
|
||||||
assert!(term_stream.advance());
|
assert!(term_stream.advance());
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
term_stream.key(),
|
term_stream.key(),
|
||||||
json_term_writer.term().serialized_value_bytes()
|
set_str("two", term).serialized_value_bytes()
|
||||||
);
|
);
|
||||||
|
|
||||||
json_term_writer.pop_path_segment();
|
// Signed
|
||||||
json_term_writer.pop_path_segment();
|
let term = term_from_path("signed");
|
||||||
json_term_writer.push_path_segment("signed");
|
|
||||||
json_term_writer.set_fast_value(-2i64);
|
|
||||||
assert!(term_stream.advance());
|
assert!(term_stream.advance());
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
term_stream.key(),
|
term_stream.key(),
|
||||||
json_term_writer.term().serialized_value_bytes()
|
set_fast_val(-2i64, term).serialized_value_bytes()
|
||||||
);
|
);
|
||||||
|
|
||||||
json_term_writer.pop_path_segment();
|
let term = term_from_path("toto");
|
||||||
json_term_writer.push_path_segment("toto");
|
|
||||||
json_term_writer.set_str("titi");
|
|
||||||
assert!(term_stream.advance());
|
assert!(term_stream.advance());
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
term_stream.key(),
|
term_stream.key(),
|
||||||
json_term_writer.term().serialized_value_bytes()
|
set_str("titi", term).serialized_value_bytes()
|
||||||
);
|
);
|
||||||
|
// Unsigned
|
||||||
json_term_writer.pop_path_segment();
|
let term = term_from_path("unsigned");
|
||||||
json_term_writer.push_path_segment("unsigned");
|
|
||||||
json_term_writer.set_fast_value(1i64);
|
|
||||||
assert!(term_stream.advance());
|
assert!(term_stream.advance());
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
term_stream.key(),
|
term_stream.key(),
|
||||||
json_term_writer.term().serialized_value_bytes()
|
set_fast_val(1i64, term).serialized_value_bytes()
|
||||||
);
|
);
|
||||||
|
|
||||||
assert!(!term_stream.advance());
|
assert!(!term_stream.advance());
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -774,14 +773,9 @@ mod tests {
|
|||||||
let searcher = reader.searcher();
|
let searcher = reader.searcher();
|
||||||
let segment_reader = searcher.segment_reader(0u32);
|
let segment_reader = searcher.segment_reader(0u32);
|
||||||
let inv_index = segment_reader.inverted_index(json_field).unwrap();
|
let inv_index = segment_reader.inverted_index(json_field).unwrap();
|
||||||
let mut term = Term::with_type_and_field(Type::Json, json_field);
|
let mut term = Term::from_field_json_path(json_field, "mykey", false);
|
||||||
let mut json_term_writer = JsonTermWriter::wrap(&mut term, false);
|
term.append_type_and_str("token");
|
||||||
json_term_writer.push_path_segment("mykey");
|
let term_info = inv_index.get_term_info(&term).unwrap().unwrap();
|
||||||
json_term_writer.set_str("token");
|
|
||||||
let term_info = inv_index
|
|
||||||
.get_term_info(json_term_writer.term())
|
|
||||||
.unwrap()
|
|
||||||
.unwrap();
|
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
term_info,
|
term_info,
|
||||||
TermInfo {
|
TermInfo {
|
||||||
@@ -818,14 +812,9 @@ mod tests {
|
|||||||
let searcher = reader.searcher();
|
let searcher = reader.searcher();
|
||||||
let segment_reader = searcher.segment_reader(0u32);
|
let segment_reader = searcher.segment_reader(0u32);
|
||||||
let inv_index = segment_reader.inverted_index(json_field).unwrap();
|
let inv_index = segment_reader.inverted_index(json_field).unwrap();
|
||||||
let mut term = Term::with_type_and_field(Type::Json, json_field);
|
let mut term = Term::from_field_json_path(json_field, "mykey", false);
|
||||||
let mut json_term_writer = JsonTermWriter::wrap(&mut term, false);
|
term.append_type_and_str("two tokens");
|
||||||
json_term_writer.push_path_segment("mykey");
|
let term_info = inv_index.get_term_info(&term).unwrap().unwrap();
|
||||||
json_term_writer.set_str("two tokens");
|
|
||||||
let term_info = inv_index
|
|
||||||
.get_term_info(json_term_writer.term())
|
|
||||||
.unwrap()
|
|
||||||
.unwrap();
|
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
term_info,
|
term_info,
|
||||||
TermInfo {
|
TermInfo {
|
||||||
@@ -863,16 +852,18 @@ mod tests {
|
|||||||
writer.commit().unwrap();
|
writer.commit().unwrap();
|
||||||
let reader = index.reader().unwrap();
|
let reader = index.reader().unwrap();
|
||||||
let searcher = reader.searcher();
|
let searcher = reader.searcher();
|
||||||
let mut term = Term::with_type_and_field(Type::Json, json_field);
|
|
||||||
let mut json_term_writer = JsonTermWriter::wrap(&mut term, false);
|
let term = Term::from_field_json_path(json_field, "mykey.field", false);
|
||||||
json_term_writer.push_path_segment("mykey");
|
|
||||||
json_term_writer.push_path_segment("field");
|
let mut hello_term = term.clone();
|
||||||
json_term_writer.set_str("hello");
|
hello_term.append_type_and_str("hello");
|
||||||
let hello_term = json_term_writer.term().clone();
|
|
||||||
json_term_writer.set_str("nothello");
|
let mut nothello_term = term.clone();
|
||||||
let nothello_term = json_term_writer.term().clone();
|
nothello_term.append_type_and_str("nothello");
|
||||||
json_term_writer.set_str("happy");
|
|
||||||
let happy_term = json_term_writer.term().clone();
|
let mut happy_term = term.clone();
|
||||||
|
happy_term.append_type_and_str("happy");
|
||||||
|
|
||||||
let phrase_query = PhraseQuery::new(vec![hello_term, happy_term.clone()]);
|
let phrase_query = PhraseQuery::new(vec![hello_term, happy_term.clone()]);
|
||||||
assert_eq!(searcher.search(&phrase_query, &Count).unwrap(), 1);
|
assert_eq!(searcher.search(&phrase_query, &Count).unwrap(), 1);
|
||||||
let phrase_query = PhraseQuery::new(vec![nothello_term, happy_term]);
|
let phrase_query = PhraseQuery::new(vec![nothello_term, happy_term]);
|
||||||
|
|||||||
@@ -255,7 +255,7 @@ pub struct Version {
|
|||||||
|
|
||||||
impl fmt::Debug for Version {
|
impl fmt::Debug for Version {
|
||||||
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
|
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
|
||||||
write!(f, "{}", self.to_string())
|
fmt::Display::fmt(self, f)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -266,9 +266,10 @@ static VERSION: Lazy<Version> = Lazy::new(|| Version {
|
|||||||
index_format_version: INDEX_FORMAT_VERSION,
|
index_format_version: INDEX_FORMAT_VERSION,
|
||||||
});
|
});
|
||||||
|
|
||||||
impl ToString for Version {
|
impl fmt::Display for Version {
|
||||||
fn to_string(&self) -> String {
|
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
|
||||||
format!(
|
write!(
|
||||||
|
f,
|
||||||
"tantivy v{}.{}.{}, index_format v{}",
|
"tantivy v{}.{}.{}, index_format v{}",
|
||||||
self.major, self.minor, self.patch, self.index_format_version
|
self.major, self.minor, self.patch, self.index_format_version
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -1,5 +1,6 @@
|
|||||||
use std::io;
|
use std::io;
|
||||||
|
|
||||||
|
use common::json_path_writer::JSON_END_OF_PATH;
|
||||||
use stacker::Addr;
|
use stacker::Addr;
|
||||||
|
|
||||||
use crate::indexer::doc_id_mapping::DocIdMapping;
|
use crate::indexer::doc_id_mapping::DocIdMapping;
|
||||||
@@ -7,7 +8,7 @@ use crate::indexer::path_to_unordered_id::OrderedPathId;
|
|||||||
use crate::postings::postings_writer::SpecializedPostingsWriter;
|
use crate::postings::postings_writer::SpecializedPostingsWriter;
|
||||||
use crate::postings::recorder::{BufferLender, DocIdRecorder, Recorder};
|
use crate::postings::recorder::{BufferLender, DocIdRecorder, Recorder};
|
||||||
use crate::postings::{FieldSerializer, IndexingContext, IndexingPosition, PostingsWriter};
|
use crate::postings::{FieldSerializer, IndexingContext, IndexingPosition, PostingsWriter};
|
||||||
use crate::schema::{Field, Type, JSON_END_OF_PATH};
|
use crate::schema::{Field, Type};
|
||||||
use crate::tokenizer::TokenStream;
|
use crate::tokenizer::TokenStream;
|
||||||
use crate::{DocId, Term};
|
use crate::{DocId, Term};
|
||||||
|
|
||||||
|
|||||||
@@ -10,10 +10,8 @@ use query_grammar::{UserInputAst, UserInputBound, UserInputLeaf, UserInputLitera
|
|||||||
use rustc_hash::FxHashMap;
|
use rustc_hash::FxHashMap;
|
||||||
|
|
||||||
use super::logical_ast::*;
|
use super::logical_ast::*;
|
||||||
use crate::core::json_utils::{
|
|
||||||
convert_to_fast_value_and_get_term, set_string_and_get_terms, JsonTermWriter,
|
|
||||||
};
|
|
||||||
use crate::index::Index;
|
use crate::index::Index;
|
||||||
|
use crate::json_utils::convert_to_fast_value_and_append_to_json_term;
|
||||||
use crate::query::range_query::{is_type_valid_for_fastfield_range_query, RangeQuery};
|
use crate::query::range_query::{is_type_valid_for_fastfield_range_query, RangeQuery};
|
||||||
use crate::query::{
|
use crate::query::{
|
||||||
AllQuery, BooleanQuery, BoostQuery, EmptyQuery, FuzzyTermQuery, Occur, PhrasePrefixQuery,
|
AllQuery, BooleanQuery, BoostQuery, EmptyQuery, FuzzyTermQuery, Occur, PhrasePrefixQuery,
|
||||||
@@ -965,20 +963,27 @@ fn generate_literals_for_json_object(
|
|||||||
})?;
|
})?;
|
||||||
let index_record_option = text_options.index_option();
|
let index_record_option = text_options.index_option();
|
||||||
let mut logical_literals = Vec::new();
|
let mut logical_literals = Vec::new();
|
||||||
let mut term = Term::with_capacity(100);
|
|
||||||
let mut json_term_writer = JsonTermWriter::from_field_and_json_path(
|
let get_term_with_path =
|
||||||
field,
|
|| Term::from_field_json_path(field, json_path, json_options.is_expand_dots_enabled());
|
||||||
json_path,
|
|
||||||
json_options.is_expand_dots_enabled(),
|
// Try to convert the phrase to a fast value
|
||||||
&mut term,
|
if let Some(term) = convert_to_fast_value_and_append_to_json_term(get_term_with_path(), phrase)
|
||||||
);
|
{
|
||||||
if let Some(term) = convert_to_fast_value_and_get_term(&mut json_term_writer, phrase) {
|
|
||||||
logical_literals.push(LogicalLiteral::Term(term));
|
logical_literals.push(LogicalLiteral::Term(term));
|
||||||
}
|
}
|
||||||
let terms = set_string_and_get_terms(&mut json_term_writer, phrase, &mut text_analyzer);
|
|
||||||
drop(json_term_writer);
|
// Try to tokenize the phrase and create Terms.
|
||||||
if terms.len() <= 1 {
|
let mut positions_and_terms = Vec::<(usize, Term)>::new();
|
||||||
for (_, term) in terms {
|
let mut token_stream = text_analyzer.token_stream(phrase);
|
||||||
|
token_stream.process(&mut |token| {
|
||||||
|
let mut term = get_term_with_path();
|
||||||
|
term.append_type_and_str(&token.text);
|
||||||
|
positions_and_terms.push((token.position, term.clone()));
|
||||||
|
});
|
||||||
|
|
||||||
|
if positions_and_terms.len() <= 1 {
|
||||||
|
for (_, term) in positions_and_terms {
|
||||||
logical_literals.push(LogicalLiteral::Term(term));
|
logical_literals.push(LogicalLiteral::Term(term));
|
||||||
}
|
}
|
||||||
return Ok(logical_literals);
|
return Ok(logical_literals);
|
||||||
@@ -989,7 +994,7 @@ fn generate_literals_for_json_object(
|
|||||||
));
|
));
|
||||||
}
|
}
|
||||||
logical_literals.push(LogicalLiteral::Phrase {
|
logical_literals.push(LogicalLiteral::Phrase {
|
||||||
terms,
|
terms: positions_and_terms,
|
||||||
slop: 0,
|
slop: 0,
|
||||||
prefix: false,
|
prefix: false,
|
||||||
});
|
});
|
||||||
|
|||||||
@@ -960,13 +960,19 @@ mod tests {
|
|||||||
"my-third-key".to_string(),
|
"my-third-key".to_string(),
|
||||||
crate::schema::OwnedValue::F64(123.0),
|
crate::schema::OwnedValue::F64(123.0),
|
||||||
);
|
);
|
||||||
assert_eq!(value, crate::schema::OwnedValue::Object(expected_object));
|
assert_eq!(
|
||||||
|
value,
|
||||||
|
crate::schema::OwnedValue::Object(expected_object.into_iter().collect())
|
||||||
|
);
|
||||||
|
|
||||||
let object = serde_json::Map::new();
|
let object = serde_json::Map::new();
|
||||||
let result = serialize_value(ReferenceValue::Object(JsonObjectIter(object.iter())));
|
let result = serialize_value(ReferenceValue::Object(JsonObjectIter(object.iter())));
|
||||||
let value = deserialize_value(result);
|
let value = deserialize_value(result);
|
||||||
let expected_object = BTreeMap::new();
|
let expected_object = BTreeMap::new();
|
||||||
assert_eq!(value, crate::schema::OwnedValue::Object(expected_object));
|
assert_eq!(
|
||||||
|
value,
|
||||||
|
crate::schema::OwnedValue::Object(expected_object.into_iter().collect())
|
||||||
|
);
|
||||||
|
|
||||||
let mut object = serde_json::Map::new();
|
let mut object = serde_json::Map::new();
|
||||||
object.insert("my-first-key".into(), serde_json::Value::Null);
|
object.insert("my-first-key".into(), serde_json::Value::Null);
|
||||||
@@ -978,7 +984,10 @@ mod tests {
|
|||||||
expected_object.insert("my-first-key".to_string(), crate::schema::OwnedValue::Null);
|
expected_object.insert("my-first-key".to_string(), crate::schema::OwnedValue::Null);
|
||||||
expected_object.insert("my-second-key".to_string(), crate::schema::OwnedValue::Null);
|
expected_object.insert("my-second-key".to_string(), crate::schema::OwnedValue::Null);
|
||||||
expected_object.insert("my-third-key".to_string(), crate::schema::OwnedValue::Null);
|
expected_object.insert("my-third-key".to_string(), crate::schema::OwnedValue::Null);
|
||||||
assert_eq!(value, crate::schema::OwnedValue::Object(expected_object));
|
assert_eq!(
|
||||||
|
value,
|
||||||
|
crate::schema::OwnedValue::Object(expected_object.into_iter().collect())
|
||||||
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
@@ -1055,7 +1064,10 @@ mod tests {
|
|||||||
.collect(),
|
.collect(),
|
||||||
),
|
),
|
||||||
);
|
);
|
||||||
assert_eq!(value, crate::schema::OwnedValue::Object(expected_object));
|
assert_eq!(
|
||||||
|
value,
|
||||||
|
crate::schema::OwnedValue::Object(expected_object.into_iter().collect())
|
||||||
|
);
|
||||||
|
|
||||||
// Some more extreme nesting that might behave weirdly
|
// Some more extreme nesting that might behave weirdly
|
||||||
let mut object = serde_json::Map::new();
|
let mut object = serde_json::Map::new();
|
||||||
@@ -1077,6 +1089,9 @@ mod tests {
|
|||||||
OwnedValue::Array(vec![OwnedValue::Null]),
|
OwnedValue::Array(vec![OwnedValue::Null]),
|
||||||
])]),
|
])]),
|
||||||
);
|
);
|
||||||
assert_eq!(value, OwnedValue::Object(expected_object));
|
assert_eq!(
|
||||||
|
value,
|
||||||
|
OwnedValue::Object(expected_object.into_iter().collect())
|
||||||
|
);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -5,22 +5,24 @@
|
|||||||
//! - [Value] which provides tantivy with a way to access the document's values in a common way
|
//! - [Value] which provides tantivy with a way to access the document's values in a common way
|
||||||
//! without performing any additional allocations.
|
//! without performing any additional allocations.
|
||||||
//! - [DocumentDeserialize] which implements the necessary code to deserialize the document from the
|
//! - [DocumentDeserialize] which implements the necessary code to deserialize the document from the
|
||||||
//! doc store.
|
//! doc store. If you are fine with fetching [TantivyDocument] from the doc store, you can skip
|
||||||
|
//! implementing this trait for your type.
|
||||||
//!
|
//!
|
||||||
//! Tantivy provides a few out-of-box implementations of these core traits to provide
|
//! Tantivy provides a few out-of-box implementations of these core traits to provide
|
||||||
//! some simple usage if you don't want to implement these traits on a custom type yourself.
|
//! some simple usage if you don't want to implement these traits on a custom type yourself.
|
||||||
//!
|
//!
|
||||||
//! # Out-of-box document implementations
|
//! # Out-of-box document implementations
|
||||||
//! - [Document] the old document type used by Tantivy before the trait based approach was
|
//! - [TantivyDocument] the old document type used by Tantivy before the trait based approach was
|
||||||
//! implemented. This type is still valid and provides all of the original behaviour you might
|
//! implemented. This type is still valid and provides all of the original behaviour you might
|
||||||
//! expect.
|
//! expect.
|
||||||
//! - `BTreeMap<Field, Value>` a mapping of field_ids to their relevant schema value using a
|
//! - `BTreeMap<Field, OwnedValue>` a mapping of field_ids to their relevant schema value using a
|
||||||
//! BTreeMap.
|
//! BTreeMap.
|
||||||
//! - `HashMap<Field, Value>` a mapping of field_ids to their relevant schema value using a HashMap.
|
//! - `HashMap<Field, OwnedValue>` a mapping of field_ids to their relevant schema value using a
|
||||||
|
//! HashMap.
|
||||||
//!
|
//!
|
||||||
//! # Implementing your custom documents
|
//! # Implementing your custom documents
|
||||||
//! Often in larger projects or higher performance applications you want to avoid the extra overhead
|
//! Often in larger projects or higher performance applications you want to avoid the extra overhead
|
||||||
//! of converting your own types to the Tantivy [Document] type, this can often save you a
|
//! of converting your own types to the [TantivyDocument] type, this can often save you a
|
||||||
//! significant amount of time when indexing by avoiding the additional allocations.
|
//! significant amount of time when indexing by avoiding the additional allocations.
|
||||||
//!
|
//!
|
||||||
//! ### Important Note
|
//! ### Important Note
|
||||||
@@ -46,6 +48,7 @@
|
|||||||
//!
|
//!
|
||||||
//! impl Document for MyCustomDocument {
|
//! impl Document for MyCustomDocument {
|
||||||
//! // The value type produced by the `iter_fields_and_values` iterator.
|
//! // The value type produced by the `iter_fields_and_values` iterator.
|
||||||
|
//! // tantivy already implements the Value trait for serde_json::Value.
|
||||||
//! type Value<'a> = &'a serde_json::Value;
|
//! type Value<'a> = &'a serde_json::Value;
|
||||||
//! // The iterator which is produced by `iter_fields_and_values`.
|
//! // The iterator which is produced by `iter_fields_and_values`.
|
||||||
//! // Often this is a simple new-type wrapper unless you like super long generics.
|
//! // Often this is a simple new-type wrapper unless you like super long generics.
|
||||||
@@ -94,10 +97,11 @@
|
|||||||
//! implementation for.
|
//! implementation for.
|
||||||
//!
|
//!
|
||||||
//! ## Implementing custom values
|
//! ## Implementing custom values
|
||||||
//! Internally, Tantivy only works with `ReferenceValue` which is an enum that tries to borrow
|
//! In order to allow documents to return custom types, they must implement
|
||||||
//! as much data as it can, in order to allow documents to return custom types, they must implement
|
//! the [Value] trait which provides a way for Tantivy to get a `ReferenceValue` that it can then
|
||||||
//! the `Value` trait which provides a way for Tantivy to get a `ReferenceValue` that it can then
|
|
||||||
//! index and store.
|
//! index and store.
|
||||||
|
//! Internally, Tantivy only works with `ReferenceValue` which is an enum that tries to borrow
|
||||||
|
//! as much data as it can
|
||||||
//!
|
//!
|
||||||
//! Values can just as easily be customised as documents by implementing the `Value` trait.
|
//! Values can just as easily be customised as documents by implementing the `Value` trait.
|
||||||
//!
|
//!
|
||||||
@@ -105,9 +109,9 @@
|
|||||||
//! hold references of the data held by the parent [Document] which can then be passed
|
//! hold references of the data held by the parent [Document] which can then be passed
|
||||||
//! on to the [ReferenceValue].
|
//! on to the [ReferenceValue].
|
||||||
//!
|
//!
|
||||||
//! This is why `Value` is implemented for `&'a serde_json::Value` and `&'a
|
//! This is why [Value] is implemented for `&'a serde_json::Value` and
|
||||||
//! tantivy::schema::Value` but not for their owned counterparts, as we cannot satisfy the lifetime
|
//! [&'a tantivy::schema::document::OwnedValue](OwnedValue) but not for their owned counterparts, as
|
||||||
//! bounds necessary when indexing the documents.
|
//! we cannot satisfy the lifetime bounds necessary when indexing the documents.
|
||||||
//!
|
//!
|
||||||
//! ### A note about returning values
|
//! ### A note about returning values
|
||||||
//! The custom value type does not have to be the type stored by the document, instead the
|
//! The custom value type does not have to be the type stored by the document, instead the
|
||||||
|
|||||||
@@ -1,4 +1,4 @@
|
|||||||
use std::collections::{btree_map, BTreeMap};
|
use std::collections::BTreeMap;
|
||||||
use std::fmt;
|
use std::fmt;
|
||||||
use std::net::Ipv6Addr;
|
use std::net::Ipv6Addr;
|
||||||
|
|
||||||
@@ -45,7 +45,7 @@ pub enum OwnedValue {
|
|||||||
/// A set of values.
|
/// A set of values.
|
||||||
Array(Vec<Self>),
|
Array(Vec<Self>),
|
||||||
/// Dynamic object value.
|
/// Dynamic object value.
|
||||||
Object(BTreeMap<String, Self>),
|
Object(Vec<(String, Self)>),
|
||||||
/// IpV6 Address. Internally there is no IpV4, it needs to be converted to `Ipv6Addr`.
|
/// IpV6 Address. Internally there is no IpV4, it needs to be converted to `Ipv6Addr`.
|
||||||
IpAddr(Ipv6Addr),
|
IpAddr(Ipv6Addr),
|
||||||
}
|
}
|
||||||
@@ -148,10 +148,10 @@ impl ValueDeserialize for OwnedValue {
|
|||||||
|
|
||||||
fn visit_object<'de, A>(&self, mut access: A) -> Result<Self::Value, DeserializeError>
|
fn visit_object<'de, A>(&self, mut access: A) -> Result<Self::Value, DeserializeError>
|
||||||
where A: ObjectAccess<'de> {
|
where A: ObjectAccess<'de> {
|
||||||
let mut elements = BTreeMap::new();
|
let mut elements = Vec::with_capacity(access.size_hint());
|
||||||
|
|
||||||
while let Some((key, value)) = access.next_entry()? {
|
while let Some((key, value)) = access.next_entry()? {
|
||||||
elements.insert(key, value);
|
elements.push((key, value));
|
||||||
}
|
}
|
||||||
|
|
||||||
Ok(OwnedValue::Object(elements))
|
Ok(OwnedValue::Object(elements))
|
||||||
@@ -167,6 +167,7 @@ impl Eq for OwnedValue {}
|
|||||||
impl serde::Serialize for OwnedValue {
|
impl serde::Serialize for OwnedValue {
|
||||||
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
|
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
|
||||||
where S: serde::Serializer {
|
where S: serde::Serializer {
|
||||||
|
use serde::ser::SerializeMap;
|
||||||
match *self {
|
match *self {
|
||||||
OwnedValue::Null => serializer.serialize_unit(),
|
OwnedValue::Null => serializer.serialize_unit(),
|
||||||
OwnedValue::Str(ref v) => serializer.serialize_str(v),
|
OwnedValue::Str(ref v) => serializer.serialize_str(v),
|
||||||
@@ -180,7 +181,13 @@ impl serde::Serialize for OwnedValue {
|
|||||||
}
|
}
|
||||||
OwnedValue::Facet(ref facet) => facet.serialize(serializer),
|
OwnedValue::Facet(ref facet) => facet.serialize(serializer),
|
||||||
OwnedValue::Bytes(ref bytes) => serializer.serialize_str(&BASE64.encode(bytes)),
|
OwnedValue::Bytes(ref bytes) => serializer.serialize_str(&BASE64.encode(bytes)),
|
||||||
OwnedValue::Object(ref obj) => obj.serialize(serializer),
|
OwnedValue::Object(ref obj) => {
|
||||||
|
let mut map = serializer.serialize_map(Some(obj.len()))?;
|
||||||
|
for &(ref k, ref v) in obj {
|
||||||
|
map.serialize_entry(k, v)?;
|
||||||
|
}
|
||||||
|
map.end()
|
||||||
|
}
|
||||||
OwnedValue::IpAddr(ref ip_v6) => {
|
OwnedValue::IpAddr(ref ip_v6) => {
|
||||||
// Ensure IpV4 addresses get serialized as IpV4, but excluding IpV6 loopback.
|
// Ensure IpV4 addresses get serialized as IpV4, but excluding IpV6 loopback.
|
||||||
if let Some(ip_v4) = ip_v6.to_ipv4_mapped() {
|
if let Some(ip_v4) = ip_v6.to_ipv4_mapped() {
|
||||||
@@ -248,12 +255,10 @@ impl<'de> serde::Deserialize<'de> for OwnedValue {
|
|||||||
|
|
||||||
fn visit_map<A>(self, mut map: A) -> Result<Self::Value, A::Error>
|
fn visit_map<A>(self, mut map: A) -> Result<Self::Value, A::Error>
|
||||||
where A: MapAccess<'de> {
|
where A: MapAccess<'de> {
|
||||||
let mut object = BTreeMap::new();
|
let mut object = map.size_hint().map(Vec::with_capacity).unwrap_or_default();
|
||||||
|
|
||||||
while let Some((key, value)) = map.next_entry()? {
|
while let Some((key, value)) = map.next_entry()? {
|
||||||
object.insert(key, value);
|
object.push((key, value));
|
||||||
}
|
}
|
||||||
|
|
||||||
Ok(OwnedValue::Object(object))
|
Ok(OwnedValue::Object(object))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -363,7 +368,8 @@ impl From<PreTokenizedString> for OwnedValue {
|
|||||||
|
|
||||||
impl From<BTreeMap<String, OwnedValue>> for OwnedValue {
|
impl From<BTreeMap<String, OwnedValue>> for OwnedValue {
|
||||||
fn from(object: BTreeMap<String, OwnedValue>) -> OwnedValue {
|
fn from(object: BTreeMap<String, OwnedValue>) -> OwnedValue {
|
||||||
OwnedValue::Object(object)
|
let key_values = object.into_iter().collect();
|
||||||
|
OwnedValue::Object(key_values)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -417,18 +423,16 @@ impl From<serde_json::Value> for OwnedValue {
|
|||||||
|
|
||||||
impl From<serde_json::Map<String, serde_json::Value>> for OwnedValue {
|
impl From<serde_json::Map<String, serde_json::Value>> for OwnedValue {
|
||||||
fn from(map: serde_json::Map<String, serde_json::Value>) -> Self {
|
fn from(map: serde_json::Map<String, serde_json::Value>) -> Self {
|
||||||
let mut object = BTreeMap::new();
|
let object: Vec<(String, OwnedValue)> = map
|
||||||
|
.into_iter()
|
||||||
for (key, value) in map {
|
.map(|(key, value)| (key, OwnedValue::from(value)))
|
||||||
object.insert(key, OwnedValue::from(value));
|
.collect();
|
||||||
}
|
|
||||||
|
|
||||||
OwnedValue::Object(object)
|
OwnedValue::Object(object)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// A wrapper type for iterating over a serde_json object producing reference values.
|
/// A wrapper type for iterating over a serde_json object producing reference values.
|
||||||
pub struct ObjectMapIter<'a>(btree_map::Iter<'a, String, OwnedValue>);
|
pub struct ObjectMapIter<'a>(std::slice::Iter<'a, (String, OwnedValue)>);
|
||||||
|
|
||||||
impl<'a> Iterator for ObjectMapIter<'a> {
|
impl<'a> Iterator for ObjectMapIter<'a> {
|
||||||
type Item = (&'a str, &'a OwnedValue);
|
type Item = (&'a str, &'a OwnedValue);
|
||||||
|
|||||||
@@ -146,7 +146,7 @@ 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;
|
||||||
pub use self::schema::{Schema, SchemaBuilder};
|
pub use self::schema::{Schema, SchemaBuilder};
|
||||||
pub use self::term::{Term, ValueBytes, JSON_END_OF_PATH};
|
pub use self::term::{Term, ValueBytes};
|
||||||
pub use self::text_options::{TextFieldIndexing, TextOptions, STRING, TEXT};
|
pub use self::text_options::{TextFieldIndexing, TextOptions, STRING, TEXT};
|
||||||
|
|
||||||
/// Validator for a potential `field_name`.
|
/// Validator for a potential `field_name`.
|
||||||
|
|||||||
@@ -3,22 +3,16 @@ use std::net::Ipv6Addr;
|
|||||||
use std::{fmt, str};
|
use std::{fmt, str};
|
||||||
|
|
||||||
use columnar::{MonotonicallyMappableToU128, MonotonicallyMappableToU64};
|
use columnar::{MonotonicallyMappableToU128, MonotonicallyMappableToU64};
|
||||||
|
use common::json_path_writer::{JSON_END_OF_PATH, JSON_PATH_SEGMENT_SEP_STR};
|
||||||
|
use common::JsonPathWriter;
|
||||||
|
|
||||||
use super::date_time_options::DATE_TIME_PRECISION_INDEXED;
|
use super::date_time_options::DATE_TIME_PRECISION_INDEXED;
|
||||||
use super::Field;
|
use super::Field;
|
||||||
use crate::fastfield::FastValue;
|
use crate::fastfield::FastValue;
|
||||||
|
use crate::json_utils::split_json_path;
|
||||||
use crate::schema::{Facet, Type};
|
use crate::schema::{Facet, Type};
|
||||||
use crate::DateTime;
|
use crate::DateTime;
|
||||||
|
|
||||||
/// Separates the different segments of a json path.
|
|
||||||
pub const JSON_PATH_SEGMENT_SEP: u8 = 1u8;
|
|
||||||
pub const JSON_PATH_SEGMENT_SEP_STR: &str =
|
|
||||||
unsafe { std::str::from_utf8_unchecked(&[JSON_PATH_SEGMENT_SEP]) };
|
|
||||||
|
|
||||||
/// Separates the json path and the value in
|
|
||||||
/// a JSON term binary representation.
|
|
||||||
pub const JSON_END_OF_PATH: u8 = 0u8;
|
|
||||||
|
|
||||||
/// Term represents the value that the token can take.
|
/// Term represents the value that the token can take.
|
||||||
/// It's a serialized representation over different types.
|
/// It's a serialized representation over different types.
|
||||||
///
|
///
|
||||||
@@ -41,6 +35,28 @@ impl Term {
|
|||||||
Term(data)
|
Term(data)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Creates a term from a json path.
|
||||||
|
///
|
||||||
|
/// The json path can address a nested value in a JSON object.
|
||||||
|
/// e.g. `{"k8s": {"node": {"id": 5}}}` can be addressed via `k8s.node.id`.
|
||||||
|
///
|
||||||
|
/// In case there are dots in the field name, and the `expand_dots_enabled` parameter is not
|
||||||
|
/// set they need to be escaped with a backslash.
|
||||||
|
/// e.g. `{"k8s.node": {"id": 5}}` can be addressed via `k8s\.node.id`.
|
||||||
|
pub fn from_field_json_path(field: Field, json_path: &str, expand_dots_enabled: bool) -> Term {
|
||||||
|
let paths = split_json_path(json_path);
|
||||||
|
let mut json_path = JsonPathWriter::with_expand_dots(expand_dots_enabled);
|
||||||
|
for path in paths {
|
||||||
|
json_path.push(&path);
|
||||||
|
}
|
||||||
|
json_path.set_end();
|
||||||
|
let mut term = Term::with_type_and_field(Type::Json, field);
|
||||||
|
|
||||||
|
term.append_bytes(json_path.as_str().as_bytes());
|
||||||
|
|
||||||
|
term
|
||||||
|
}
|
||||||
|
|
||||||
pub(crate) fn with_type_and_field(typ: Type, field: Field) -> Term {
|
pub(crate) fn with_type_and_field(typ: Type, field: Field) -> Term {
|
||||||
let mut term = Self::with_capacity(8);
|
let mut term = Self::with_capacity(8);
|
||||||
term.set_field_and_type(field, typ);
|
term.set_field_and_type(field, typ);
|
||||||
@@ -169,7 +185,11 @@ impl Term {
|
|||||||
self.set_bytes(val.to_u64().to_be_bytes().as_ref());
|
self.set_bytes(val.to_u64().to_be_bytes().as_ref());
|
||||||
}
|
}
|
||||||
|
|
||||||
pub(crate) fn append_type_and_fast_value<T: FastValue>(&mut self, val: T) {
|
/// Append a type marker + fast value to a term.
|
||||||
|
/// This is used in JSON type to append a fast value after the path.
|
||||||
|
///
|
||||||
|
/// It will not clear existing bytes.
|
||||||
|
pub fn append_type_and_fast_value<T: FastValue>(&mut self, val: T) {
|
||||||
self.0.push(T::to_type().to_code());
|
self.0.push(T::to_type().to_code());
|
||||||
let value = if T::to_type() == Type::Date {
|
let value = if T::to_type() == Type::Date {
|
||||||
DateTime::from_u64(val.to_u64())
|
DateTime::from_u64(val.to_u64())
|
||||||
@@ -181,6 +201,15 @@ impl Term {
|
|||||||
self.0.extend(value.to_be_bytes().as_ref());
|
self.0.extend(value.to_be_bytes().as_ref());
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Append a string type marker + string to a term.
|
||||||
|
/// This is used in JSON type to append a str after the path.
|
||||||
|
///
|
||||||
|
/// It will not clear existing bytes.
|
||||||
|
pub fn append_type_and_str(&mut self, val: &str) {
|
||||||
|
self.0.push(Type::Str.to_code());
|
||||||
|
self.0.extend(val.as_bytes().as_ref());
|
||||||
|
}
|
||||||
|
|
||||||
/// Sets a `Ipv6Addr` value in the term.
|
/// Sets a `Ipv6Addr` value in the term.
|
||||||
pub fn set_ip_addr(&mut self, val: Ipv6Addr) {
|
pub fn set_ip_addr(&mut self, val: Ipv6Addr) {
|
||||||
self.set_bytes(val.to_u128().to_be_bytes().as_ref());
|
self.set_bytes(val.to_u128().to_be_bytes().as_ref());
|
||||||
@@ -192,11 +221,6 @@ impl Term {
|
|||||||
self.0.extend(bytes);
|
self.0.extend(bytes);
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Set the texts only, keeping the field untouched.
|
|
||||||
pub fn set_text(&mut self, text: &str) {
|
|
||||||
self.set_bytes(text.as_bytes());
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Truncates the value bytes of the term. Value and field type stays the same.
|
/// Truncates the value bytes of the term. Value and field type stays the same.
|
||||||
pub fn truncate_value_bytes(&mut self, len: usize) {
|
pub fn truncate_value_bytes(&mut self, len: usize) {
|
||||||
self.0.truncate(len + TERM_METADATA_LENGTH);
|
self.0.truncate(len + TERM_METADATA_LENGTH);
|
||||||
@@ -233,27 +257,6 @@ impl Term {
|
|||||||
}
|
}
|
||||||
&mut self.0[len_before..]
|
&mut self.0[len_before..]
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Appends a JSON_PATH_SEGMENT_SEP to the term.
|
|
||||||
/// Only used for JSON type.
|
|
||||||
#[inline]
|
|
||||||
pub fn add_json_path_separator(&mut self) {
|
|
||||||
self.0.push(JSON_PATH_SEGMENT_SEP);
|
|
||||||
}
|
|
||||||
/// Sets the current end to JSON_END_OF_PATH.
|
|
||||||
/// Only used for JSON type.
|
|
||||||
#[inline]
|
|
||||||
pub fn set_json_path_end(&mut self) {
|
|
||||||
let buffer_len = self.0.len();
|
|
||||||
self.0[buffer_len - 1] = JSON_END_OF_PATH;
|
|
||||||
}
|
|
||||||
/// Sets the current end to JSON_PATH_SEGMENT_SEP.
|
|
||||||
/// Only used for JSON type.
|
|
||||||
#[inline]
|
|
||||||
pub fn set_json_path_separator(&mut self) {
|
|
||||||
let buffer_len = self.0.len();
|
|
||||||
self.0[buffer_len - 1] = JSON_PATH_SEGMENT_SEP;
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
impl<B> Term<B>
|
impl<B> Term<B>
|
||||||
|
|||||||
@@ -14,7 +14,7 @@ use super::Decompressor;
|
|||||||
use crate::directory::FileSlice;
|
use crate::directory::FileSlice;
|
||||||
use crate::error::DataCorruption;
|
use crate::error::DataCorruption;
|
||||||
use crate::fastfield::AliveBitSet;
|
use crate::fastfield::AliveBitSet;
|
||||||
use crate::schema::document::{BinaryDocumentDeserializer, Document, DocumentDeserialize};
|
use crate::schema::document::{BinaryDocumentDeserializer, DocumentDeserialize};
|
||||||
use crate::space_usage::StoreSpaceUsage;
|
use crate::space_usage::StoreSpaceUsage;
|
||||||
use crate::store::index::Checkpoint;
|
use crate::store::index::Checkpoint;
|
||||||
use crate::DocId;
|
use crate::DocId;
|
||||||
@@ -235,7 +235,7 @@ impl StoreReader {
|
|||||||
/// Iterator over all Documents in their order as they are stored in the doc store.
|
/// Iterator over all Documents in their order as they are stored in the doc store.
|
||||||
/// Use this, if you want to extract all Documents from the doc store.
|
/// Use this, if you want to extract all Documents from the doc store.
|
||||||
/// The `alive_bitset` has to be forwarded from the `SegmentReader` or the results may be wrong.
|
/// The `alive_bitset` has to be forwarded from the `SegmentReader` or the results may be wrong.
|
||||||
pub fn iter<'a: 'b, 'b, D: Document + DocumentDeserialize>(
|
pub fn iter<'a: 'b, 'b, D: DocumentDeserialize>(
|
||||||
&'b self,
|
&'b self,
|
||||||
alive_bitset: Option<&'a AliveBitSet>,
|
alive_bitset: Option<&'a AliveBitSet>,
|
||||||
) -> impl Iterator<Item = crate::Result<D>> + 'b {
|
) -> impl Iterator<Item = crate::Result<D>> + 'b {
|
||||||
|
|||||||
@@ -1,6 +1,5 @@
|
|||||||
use std::io::{self, Write};
|
use std::io::{self, Write};
|
||||||
use std::ops::Range;
|
use std::ops::Range;
|
||||||
use std::usize;
|
|
||||||
|
|
||||||
use merge::ValueMerger;
|
use merge::ValueMerger;
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user