gremlin_client/structure/
p.rs

1use crate::structure::either::Either2;
2use crate::structure::text_p::TextP;
3use crate::{GValue, ToGValue};
4
5#[derive(Debug, PartialEq, Clone)]
6pub struct P {
7    pub(crate) operator: String,
8    pub(crate) value: Box<GValue>,
9}
10
11impl P {
12    pub fn operator(&self) -> &String {
13        &self.operator
14    }
15
16    pub fn value(&self) -> &GValue {
17        &self.value
18    }
19
20    pub(crate) fn new<T>(operator: T, value: GValue) -> P
21    where
22        T: Into<String>,
23    {
24        P {
25            operator: operator.into(),
26            value: Box::new(value),
27        }
28    }
29    pub fn eq<V>(value: V) -> P
30    where
31        V: ToGValue,
32    {
33        P::new("eq", value.to_gvalue())
34    }
35
36    pub fn neq<V>(value: V) -> P
37    where
38        V: ToGValue,
39    {
40        P::new("neq", value.to_gvalue())
41    }
42
43    pub fn gt<V>(value: V) -> P
44    where
45        V: ToGValue,
46    {
47        P::new("gt", value.to_gvalue())
48    }
49
50    pub fn gte<V>(value: V) -> P
51    where
52        V: ToGValue,
53    {
54        P::new("gte", value.to_gvalue())
55    }
56
57    pub fn lt<V>(value: V) -> P
58    where
59        V: ToGValue,
60    {
61        P::new("lt", value.to_gvalue())
62    }
63
64    pub fn lte<V>(value: V) -> P
65    where
66        V: ToGValue,
67    {
68        P::new("lte", value.to_gvalue())
69    }
70
71    pub fn within<V>(value: V) -> P
72    where
73        V: IntoRange,
74    {
75        P::new("within", value.into_range().values.into())
76    }
77}
78
79pub trait IntoPredicate {
80    fn into_predicate(self) -> Either2<P, TextP>;
81}
82
83impl<T: ToGValue> IntoPredicate for T {
84    fn into_predicate(self) -> Either2<P, TextP> {
85        let val = self.to_gvalue();
86        match val {
87            GValue::P(ref p) => Either2::A(p.clone()),
88            GValue::TextP(ref p) => Either2::B(p.clone()),
89            _ => Either2::A(P::new("eq", val)),
90        }
91    }
92}
93
94pub struct Range {
95    values: Vec<GValue>,
96}
97
98pub trait IntoRange {
99    fn into_range(self) -> Range;
100}
101
102impl<T> IntoRange for (T, T)
103where
104    T: ToGValue,
105{
106    fn into_range(self) -> Range {
107        let v1 = self.0.to_gvalue();
108        let v2 = self.1.to_gvalue();
109
110        Range {
111            values: vec![v1, v2],
112        }
113    }
114}
115
116impl<T> IntoRange for Vec<T>
117where
118    T: ToGValue,
119{
120    fn into_range(self) -> Range {
121        Range {
122            values: self.into_iter().map(|e| e.to_gvalue()).collect(),
123        }
124    }
125}
gremlin_client/structure/p.rs

gremlin_client/structure/
p.rs
