elastic_query_builder/mapping/
mod.rs

1pub mod field_type;
2pub mod properties;
3
4use crate::mapping::field_type::keyword::KeywordFieldType;
5use crate::mapping::field_type::text::TextFieldType;
6use crate::mapping::properties::MappingProperties;
7use crate::util::UtilMap;
8use serde_json::{json, Value};
9
10pub trait MappingTrait {
11    fn build(&self) -> Value;
12    fn query_name(&self) -> String;
13}
14
15pub struct MappingBuilder {
16    properties: MappingProperties,
17    settings: Option<Value>,
18    aliases: Option<Value>,
19}
20
21impl MappingBuilder {
22    pub fn new() -> MappingBuilder {
23        MappingBuilder {
24            properties: MappingProperties::new(),
25            settings: None,
26            aliases: None,
27        }
28    }
29    pub fn add_property<T>(&mut self, key: &str, value: T) -> &mut MappingBuilder
30    where
31        T: MappingTrait + 'static,
32    {
33        self.properties.add_property(key, value);
34        self
35    }
36    pub fn set_properties(&mut self, properties: MappingProperties) {
37        self.properties = properties;
38    }
39    pub fn set_settings(&mut self, settings: Value) {
40        self.settings = Some(settings);
41    }
42    pub fn set_aliases(&mut self, aliases: Value) {
43        self.aliases = Some(aliases);
44    }
45    pub fn build(self) -> Value {
46        let mut map = UtilMap::new();
47        map.append_value("mappings", self.properties.build());
48        if let Some(ref settings) = self.settings {
49            map.append_value("settings", settings.clone())
50        }
51      if let Some(ref aliases) = self.aliases {
52            map.append_value("aliases", aliases.clone())
53        }
54        map.build()
55    }
56}
57
58#[test]
59fn test() {
60    let mut mapping = MappingBuilder::new();
61    mapping
62        .add_property("title", KeywordFieldType::new())
63        .add_property("content", TextFieldType::new());
64    mapping.set_settings(json!({
65       "index.lifecycle.name": "logs_policy"
66    }));
67    println!("{}", mapping.build().to_string())
68}