elastic_query_builder/mapping/
mod.rs1pub 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}