51 lines
1.2 KiB
Rust
51 lines
1.2 KiB
Rust
|
use crate::sql::idiom::Idiom;
|
||
|
|
use crate::sql::part::Part;
|
||
|
|
use crate::sql::value::Value;
|
||
|
|
|
||
|
|
impl Value {
|
||
|
|
pub fn every(&self) -> Vec<Idiom> {
|
||
|
|
self._every(Idiom::default())
|
||
|
|
}
|
||
|
|
fn _every(&self, prev: Idiom) -> Vec<Idiom> {
|
||
|
|
match self {
|
||
|
|
// Current path part is an object
|
||
|
|
Value::Object(v) => v
|
||
|
|
.value
|
||
|
|
.iter()
|
||
|
|
.flat_map(|(k, v)| v._every(prev.add(Part::from(k))))
|
||
|
|
.collect::<Vec<_>>(),
|
||
|
|
// Current path part is an array
|
||
|
|
Value::Array(v) => v
|
||
|
|
.value
|
||
|
|
.iter()
|
||
|
|
.enumerate()
|
||
|
|
.flat_map(|(i, v)| v._every(prev.add(Part::from(i))))
|
||
|
|
.collect::<Vec<_>>(),
|
||
|
|
// Process everything else
|
||
|
|
_ => vec![prev],
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
#[cfg(test)]
|
||
|
|
mod tests {
|
||
|
|
|
||
|
|
use super::*;
|
||
|
|
use crate::sql::idiom::Idiom;
|
||
|
|
use crate::sql::test::Parse;
|
||
|
|
|
||
|
|
#[test]
|
||
|
|
fn every() {
|
||
|
|
let val = Value::parse("{ test: { something: [{ age: 34, tags: ['code', 'databases'] }, { age: 36, tags: ['design', 'operations'] }] } }");
|
||
|
|
let res = vec![
|
||
|
|
Idiom::parse("test.something[0].age"),
|
||
|
|
Idiom::parse("test.something[0].tags[0]"),
|
||
|
|
Idiom::parse("test.something[0].tags[1]"),
|
||
|
|
Idiom::parse("test.something[1].age"),
|
||
|
|
Idiom::parse("test.something[1].tags[0]"),
|
||
|
|
Idiom::parse("test.something[1].tags[1]"),
|
||
|
|
];
|
||
|
|
assert_eq!(res, val.every());
|
||
|
|
}
|
||
|
|
}
|