traversal_complex/
traversal_complex.rs

1use gremlin_client::{
2    process::traversal::{traversal, GraphTraversalSource, SyncTerminator, __},
3    structure::{List, Vertex, P},
4    GremlinClient,
5};
6
7fn main() -> Result<(), Box<dyn std::error::Error>> {
8    let client = GremlinClient::connect("localhost")?;
9
10    let g = traversal().with_remote(client);
11
12    create_graph(&g)?;
13
14    let result = g
15        .v(())
16        .has_label("complex_vertex")
17        .has(("name", "test1"))
18        .out("complex_label")
19        .out("complex_label")
20        .value_map(())
21        .next()?
22        .expect("no vertices found");
23
24    println!(
25        "Found vertex with name {:?}",
26        result["name"].get::<List>().unwrap()[0]
27    );
28
29    let results = g
30        .v(())
31        .has_label("complex_vertex")
32        .has(("number", P::gt(3)))
33        .to_list()?;
34
35    println!(
36        "Found {} vertices with number greater than 3",
37        results.len()
38    );
39
40    let results = g
41        .v(())
42        .has_label("complex_vertex")
43        .has(("number", P::within((3, 6))))
44        .to_list()?;
45
46    println!("Found {} vertices with number 3 or 6", results.len());
47
48    let results = g
49        .v(())
50        .has_label("complex_vertex")
51        .where_(__.out("complex_label").count().is(P::gte(1)))
52        .to_list()?;
53
54    println!(
55        "Found {} vertices with 1 or more connected edges with label complex_label",
56        results.len()
57    );
58
59    Ok(())
60}
61
62fn create_graph(
63    g: &GraphTraversalSource<SyncTerminator>,
64) -> Result<(), Box<dyn std::error::Error>> {
65    g.v(()).has_label("complex_vertex").drop().next()?;
66    g.e(()).has_label("complex_label").drop().next()?;
67
68    let mut current_next: Option<Vertex> = None;
69    (0..10).for_each(|e| {
70        let next = g
71            .add_v("complex_vertex")
72            .property("name", format!("test{}", e))
73            .property("number", e)
74            .next()
75            .expect("failed to create vertex");
76
77        current_next.iter().zip(next.iter()).for_each(|(a, b)| {
78            g.add_e("complex_label")
79                .from(a)
80                .to(b)
81                .next()
82                .expect("failed to create edge");
83        });
84
85        current_next = next;
86    });
87
88    Ok(())
89}