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

use crate::query::QueryTrait;
use serde_json::{json, Value};

#[derive(Default)]
pub struct ScriptQuery {
    script: String,
    source: String,
    lang: String,
    params: Value,
}

impl ScriptQuery {
    pub fn new(script: &str) -> ScriptQuery {
        let mut query = ScriptQuery::default();
        query.script = script.to_string();
        return query;
    }
}

impl QueryTrait for ScriptQuery {
    fn build(&self) -> Value {
        let script = self.script.to_string();
        let name = self.query_name();
        json!({
            name:{
                "script":script
            }
        })
    }

    fn query_name(&self) -> String {
        return "script".to_string();
    }
}


#[test]
fn test(){
    let q = ScriptQuery::new("doc['keywords'].length == 2").build().to_string();
    assert_eq!(q,"{\"script\":{\"script\":\"doc['keywords'].length == 2\"}}")
}