surrealdb_core/sql/value/
pick.rs

1use crate::sql::part::Next;
2use crate::sql::part::Part;
3use crate::sql::value::Value;
4
5impl Value {
6	/// Synchronous method for getting a field from a `Value`
7	pub fn pick(&self, path: &[Part]) -> Self {
8		match path.first() {
9			// Get the current value at path
10			Some(p) => match self {
11				// Current value at path is an object
12				Value::Object(v) => match p {
13					Part::Field(f) => match v.get(f as &str) {
14						Some(v) => v.pick(path.next()),
15						None => Value::None,
16					},
17					Part::Index(i) => match v.get(&i.to_string()) {
18						Some(v) => v.pick(path.next()),
19						None => Value::None,
20					},
21					Part::All => {
22						v.iter().map(|(_, v)| v.pick(path.next())).collect::<Vec<_>>().into()
23					}
24					_ => Value::None,
25				},
26				// Current value at path is an array
27				Value::Array(v) => match p {
28					Part::All => v.iter().map(|v| v.pick(path.next())).collect::<Vec<_>>().into(),
29					Part::First => match v.first() {
30						Some(v) => v.pick(path.next()),
31						None => Value::None,
32					},
33					Part::Last => match v.last() {
34						Some(v) => v.pick(path.next()),
35						None => Value::None,
36					},
37					Part::Index(i) => match v.get(i.to_usize()) {
38						Some(v) => v.pick(path.next()),
39						None => Value::None,
40					},
41					_ => v.iter().map(|v| v.pick(path)).collect::<Vec<_>>().into(),
42				},
43				// Ignore everything else
44				_ => Value::None,
45			},
46			// No more parts so get the value
47			None => self.clone(),
48		}
49	}
50}
51
52#[cfg(test)]
53mod tests {
54
55	use super::*;
56	use crate::sql::id::Id;
57	use crate::sql::idiom::Idiom;
58	use crate::sql::thing::Thing;
59	use crate::syn::Parse;
60
61	#[test]
62	fn pick_none() {
63		let idi = Idiom::default();
64		let val = Value::parse("{ test: { other: null, something: 123 } }");
65		let res = val.pick(&idi);
66		assert_eq!(res, val);
67	}
68
69	#[test]
70	fn pick_basic() {
71		let idi = Idiom::parse("test.something");
72		let val = Value::parse("{ test: { other: null, something: 123 } }");
73		let res = val.pick(&idi);
74		assert_eq!(res, Value::from(123));
75	}
76
77	#[test]
78	fn pick_thing() {
79		let idi = Idiom::parse("test.other");
80		let val = Value::parse("{ test: { other: test:tobie, something: 123 } }");
81		let res = val.pick(&idi);
82		assert_eq!(
83			res,
84			Value::from(Thing {
85				tb: String::from("test"),
86				id: Id::from("tobie")
87			})
88		);
89	}
90
91	#[test]
92	fn pick_array() {
93		let idi = Idiom::parse("test.something[1]");
94		let val = Value::parse("{ test: { something: [123, 456, 789] } }");
95		let res = val.pick(&idi);
96		assert_eq!(res, Value::from(456));
97	}
98
99	#[test]
100	fn pick_array_thing() {
101		let idi = Idiom::parse("test.something[1]");
102		let val = Value::parse("{ test: { something: [test:tobie, test:jaime] } }");
103		let res = val.pick(&idi);
104		assert_eq!(
105			res,
106			Value::from(Thing {
107				tb: String::from("test"),
108				id: Id::from("jaime")
109			})
110		);
111	}
112
113	#[test]
114	fn pick_array_field() {
115		let idi = Idiom::parse("test.something[1].age");
116		let val = Value::parse("{ test: { something: [{ age: 34 }, { age: 36 }] } }");
117		let res = val.pick(&idi);
118		assert_eq!(res, Value::from(36));
119	}
120
121	#[test]
122	fn pick_array_fields() {
123		let idi = Idiom::parse("test.something[*].age");
124		let val = Value::parse("{ test: { something: [{ age: 34 }, { age: 36 }] } }");
125		let res = val.pick(&idi);
126		assert_eq!(res, Value::from(vec![34, 36]));
127	}
128
129	#[test]
130	fn pick_array_fields_flat() {
131		let idi = Idiom::parse("test.something.age");
132		let val = Value::parse("{ test: { something: [{ age: 34 }, { age: 36 }] } }");
133		let res = val.pick(&idi);
134		assert_eq!(res, Value::from(vec![34, 36]));
135	}
136}
surrealdb_core/sql/value/pick.rs

surrealdb_core/sql/value/
pick.rs
