gchdb 0.2.8

Provides a record abstraction for storing chat records extracted by different chat software, and provides full-text search feature
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55

use super::*;
use tantivy::{schema::*, Document};

pub struct Fields {
    pub idx: Field,
    pub content: Field,
    pub timestamp: Field,
    #[allow(dead_code)]
    pub custom: Vec<Field>,
    pub schema: Schema,
}

impl Fields {
    fn new<S: ToString>(custom: Vec<(S, TextOptions)>) -> Self {
        let mut schema_builder = Schema::builder();
        let mut custom_field = vec![];
        for (name, options) in custom {
            custom_field.push(schema_builder.add_text_field(name.to_string().as_str(), options));
        }
        Self {
            idx: schema_builder.add_i64_field("idx", FAST | STORED),
            content: schema_builder.add_text_field(
                "content",
                TextOptions::default().set_indexing_options(
                    TextFieldIndexing::default()
                        .set_tokenizer(LANG_CN)
                        .set_index_option(IndexRecordOption::WithFreqsAndPositions),
                ),
            ),
            timestamp: schema_builder.add_u64_field("timestamp", FAST),
            custom: custom_field,
            schema: schema_builder.build(),
        }
    }
}

impl Default for Fields {
    fn default() -> Self {
        Self::new::<String>(vec![])
    }
}

pub trait GetDocument {
    fn get_document(&self, fields: &Fields) -> ChatRecordResult<Document>;
}

impl GetDocument for Record {
    fn get_document(&self, fields: &Fields) -> ChatRecordResult<Document> {
        Ok(doc! {
            fields.idx => self.get_id() as i64,
            fields.content => self.content.as_str(),
            fields.timestamp => self.timestamp
        })
    }
}