gqlite 1.8.0 → 1.8.2

data/ext/gqlitedb/src/tests/templates/ast.rs

@@ -9,6 +9,7 @@ fn return_statement(var_id: VariableIdentifier) -> Statement
       expression: Variable { identifier: var_id }.into(),
     }],
     modifiers: Modifiers {
+      distinct: false,
       skip: None,
       limit: None,
       order_by: None,
@@ -26,6 +27,8 @@ pub(crate) fn simple_create_node() -> Statements
         variable: None,
         labels: LabelExpression::None,
         properties: None,
+
+        predicate: None,
       })],
     }
     .into(),
@@ -53,6 +56,8 @@ pub(crate) fn create_named_node() -> Statements
           }
           .into(),
         ),
+
+        predicate: None,
       })],
     }
     .into(),
@@ -70,6 +75,7 @@ pub(crate) fn create_named_node() -> Statements
         .into(),
       }],
       modifiers: Modifiers {
+        distinct: false,
         skip: None,
         limit: None,
         order_by: None,
@@ -104,6 +110,8 @@ pub(crate) fn create_named_node_double_return() -> Statements
           }
           .into(),
         ),
+
+        predicate: None,
       })],
     }
     .into(),
@@ -134,6 +142,7 @@ pub(crate) fn create_named_node_double_return() -> Statements
         },
       ],
       modifiers: Modifiers {
+        distinct: false,
         skip: None,
         limit: None,
         order_by: None,
@@ -162,6 +171,7 @@ pub(crate) fn double_with_return() -> Statements
         },
       ],
       modifiers: Modifiers {
+        distinct: false,
         skip: None,
         limit: None,
         order_by: None,
@@ -188,6 +198,7 @@ pub(crate) fn double_with_return() -> Statements
         },
       ],
       modifiers: Modifiers {
+        distinct: false,
         skip: None,
         limit: None,
         order_by: None,
@@ -228,16 +239,19 @@ pub(crate) fn match_loop() -> Statements
           variable: Some(var_ids.create_variable_from_name("n")),
           labels: LabelExpression::None,
           properties: None,
+          predicate: None,
         },
         destination: NodePattern {
           variable: Some(var_ids.create_variable_from_name("n")),
           labels: LabelExpression::None,
           properties: None,
+          predicate: None,
         },
         labels: LabelExpression::None,
         properties: None,
         directivity: graphcore::EdgeDirectivity::Directed,
-        recursive_range: None,
+
+        predicate: None,
       })],
       where_expression: None,
       optional: false,
@@ -258,6 +272,8 @@ pub(crate) fn optional_match() -> Statements
         variable: Some(var_ids.create_variable_from_name("a")),
         labels: LabelExpression::None,
         properties: None,
+
+        predicate: None,
       })],
       where_expression: None,
       optional: true,
@@ -278,6 +294,8 @@ pub(crate) fn match_count() -> Statements
         variable: Some(var_ids.create_variable_from_name("a")),
         labels: LabelExpression::None,
         properties: None,
+
+        predicate: None,
       })],
       where_expression: None,
       optional: false,
@@ -291,10 +309,12 @@ pub(crate) fn match_count() -> Statements
         expression: FunctionCall {
           name: "count".into(),
           arguments: vec![Value { value: 0.into() }.into()],
+          distinct: false,
         }
         .into(),
       }],
       modifiers: Modifiers {
+        distinct: false,
         skip: None,
         limit: None,
         order_by: None,
@@ -315,6 +335,8 @@ pub(crate) fn aggregation() -> Statements
         variable: Some(var_ids.create_variable_from_name("n")),
         labels: LabelExpression::None,
         properties: None,
+
+        predicate: None,
       })],
       where_expression: None,
       optional: false,
@@ -349,11 +371,13 @@ pub(crate) fn aggregation() -> Statements
               }
               .into(),
             ],
+            distinct: false,
           }
           .into(),
         },
       ],
       modifiers: Modifiers {
+        distinct: false,
         skip: None,
         limit: None,
         order_by: None,
@@ -391,6 +415,8 @@ pub(crate) fn match_with_vector_search(index_name: &str) -> Statements
         variable: Some(n.clone()),
         labels: LabelExpression::String("Doc".to_string()),
         properties: None,
+
+        predicate: None,
       })],
       where_expression: None,
       optional: false,
@@ -421,6 +447,273 @@ pub(crate) fn match_with_vector_search(index_name: &str) -> Statements
         .into(),
       }],
       modifiers: Modifiers {
+        distinct: false,
+        skip: None,
+        limit: None,
+        order_by: None,
+      },
+      where_expression: None,
+    }
+    .into(),
+  ]
+}
+
+/// AST for `MATCH (a) ((n)-[:R]->(m)){1,3} (b) RETURN a, b`
+/// AST for `MATCH (a) ((n)-[:R]->(m)){1,3} (b) RETURN a, b`
+pub(crate) fn group_single_edge_range() -> Statements
+{
+  let var_ids = VariableIdentifiers::default();
+  let a = var_ids.create_variable_from_name("a");
+  let n = var_ids.create_variable_from_name("n");
+  let m = var_ids.create_variable_from_name("m");
+  let b = var_ids.create_variable_from_name("b");
+  let edge_anon = var_ids.create_variable_from_name("anonymous_4");
+  vec![
+    Match {
+      patterns: vec![Pattern::QuantifiedPath(QuantifiedPathPattern {
+        variable: None,
+        source: NodePattern {
+          variable: Some(a.clone()),
+          labels: LabelExpression::None,
+          properties: None,
+          predicate: None,
+        },
+        group: vec![Pattern::Edge(EdgePattern {
+          variable: Some(edge_anon),
+          source: NodePattern {
+            variable: Some(n),
+            labels: LabelExpression::None,
+            properties: None,
+            predicate: None,
+          },
+          destination: NodePattern {
+            variable: Some(m),
+            labels: LabelExpression::None,
+            properties: None,
+            predicate: None,
+          },
+          directivity: graphcore::EdgeDirectivity::Directed,
+          labels: LabelExpression::String("R".to_string()),
+          properties: None,
+          predicate: None,
+        })],
+        group_where: None,
+        path_where: None,
+        quantifier: Some(1..3),
+        destination: NodePattern {
+          variable: Some(b.clone()),
+          labels: LabelExpression::None,
+          properties: None,
+          predicate: None,
+        },
+        selector: PathSelector::All,
+      })],
+      where_expression: None,
+      optional: false,
+      search: None,
+    }
+    .into(),
+    Return {
+      all: false,
+      expressions: vec![
+        NamedExpression {
+          identifier: a.clone(),
+          expression: Variable { identifier: a }.into(),
+        },
+        NamedExpression {
+          identifier: b.clone(),
+          expression: Variable { identifier: b }.into(),
+        },
+      ],
+      modifiers: Modifiers {
+        distinct: false,
+        skip: None,
+        limit: None,
+        order_by: None,
+      },
+      where_expression: None,
+    }
+    .into(),
+  ]
+}
+
+/// AST for `MATCH (a) ((n)-[:R1]->(x)-[:R2]->(m)){1,3} (b) RETURN a, b`
+pub(crate) fn group_two_edges_range() -> Statements
+{
+  let var_ids = VariableIdentifiers::default();
+  let a = var_ids.create_variable_from_name("a");
+  let n = var_ids.create_variable_from_name("n");
+  let x = var_ids.create_variable_from_name("x");
+  let m = var_ids.create_variable_from_name("m");
+  let b = var_ids.create_variable_from_name("b");
+  let edge1_anon = var_ids.create_variable_from_name("anonymous_5");
+  let edge2_anon = var_ids.create_variable_from_name("anonymous_6");
+  vec![
+    Match {
+      patterns: vec![Pattern::QuantifiedPath(QuantifiedPathPattern {
+        variable: None,
+        source: NodePattern {
+          variable: Some(a.clone()),
+          labels: LabelExpression::None,
+          properties: None,
+          predicate: None,
+        },
+        group: vec![
+          Pattern::Edge(EdgePattern {
+            variable: Some(edge1_anon),
+            source: NodePattern {
+              variable: Some(n),
+              labels: LabelExpression::None,
+              properties: None,
+              predicate: None,
+            },
+            destination: NodePattern {
+              variable: Some(x.clone()),
+              labels: LabelExpression::None,
+              properties: None,
+              predicate: None,
+            },
+            directivity: graphcore::EdgeDirectivity::Directed,
+            labels: LabelExpression::String("R1".to_string()),
+            properties: None,
+            predicate: None,
+          }),
+          Pattern::Edge(EdgePattern {
+            variable: Some(edge2_anon),
+            source: NodePattern {
+              variable: Some(x),
+              labels: LabelExpression::None,
+              properties: None,
+              predicate: None,
+            },
+            destination: NodePattern {
+              variable: Some(m),
+              labels: LabelExpression::None,
+              properties: None,
+              predicate: None,
+            },
+            directivity: graphcore::EdgeDirectivity::Directed,
+            labels: LabelExpression::String("R2".to_string()),
+            properties: None,
+            predicate: None,
+          }),
+        ],
+        group_where: None,
+        path_where: None,
+        quantifier: Some(1..3),
+        destination: NodePattern {
+          variable: Some(b.clone()),
+          labels: LabelExpression::None,
+          properties: None,
+          predicate: None,
+        },
+        selector: PathSelector::All,
+      })],
+      where_expression: None,
+      optional: false,
+      search: None,
+    }
+    .into(),
+    Return {
+      all: false,
+      expressions: vec![
+        NamedExpression {
+          identifier: a.clone(),
+          expression: Variable { identifier: a }.into(),
+        },
+        NamedExpression {
+          identifier: b.clone(),
+          expression: Variable { identifier: b }.into(),
+        },
+      ],
+      modifiers: Modifiers {
+        distinct: false,
+        skip: None,
+        limit: None,
+        order_by: None,
+      },
+      where_expression: None,
+    }
+    .into(),
+  ]
+}
+
+/// AST for `MATCH p = SHORTEST 1 (a)-[:R]-+(b) RETURN a, b, length(p)`
+pub(crate) fn shortest_single_edge_full_return() -> Statements
+{
+  let var_ids = VariableIdentifiers::default();
+  let a = var_ids.create_variable_from_name("a");
+  let b = var_ids.create_variable_from_name("b");
+  let edge_anon = var_ids.create_variable_from_name("anonymous_2");
+  let p = var_ids.create_variable_from_name("p");
+  vec![
+    Match {
+      patterns: vec![Pattern::QuantifiedPath(QuantifiedPathPattern {
+        variable: Some(p.clone()),
+        source: NodePattern {
+          variable: Some(a.clone()),
+          labels: LabelExpression::None,
+          properties: None,
+          predicate: None,
+        },
+        group: vec![Pattern::Edge(EdgePattern {
+          variable: Some(edge_anon),
+          source: NodePattern {
+            variable: Some(a.clone()),
+            labels: LabelExpression::None,
+            properties: None,
+            predicate: None,
+          },
+          destination: NodePattern {
+            variable: Some(b.clone()),
+            labels: LabelExpression::None,
+            properties: None,
+            predicate: None,
+          },
+          directivity: graphcore::EdgeDirectivity::Undirected,
+          labels: LabelExpression::String("R".to_string()),
+          properties: None,
+          predicate: None,
+        })],
+        group_where: None,
+        path_where: None,
+        quantifier: Some(1..usize::MAX),
+        destination: NodePattern {
+          variable: Some(b.clone()),
+          labels: LabelExpression::None,
+          properties: None,
+          predicate: None,
+        },
+        selector: PathSelector::Shortest(Value { value: 1.into() }.into()),
+      })],
+      where_expression: None,
+      optional: false,
+      search: None,
+    }
+    .into(),
+    Return {
+      all: false,
+      expressions: vec![
+        NamedExpression {
+          identifier: a.clone(),
+          expression: Variable { identifier: a }.into(),
+        },
+        NamedExpression {
+          identifier: b.clone(),
+          expression: Variable { identifier: b }.into(),
+        },
+        NamedExpression {
+          identifier: var_ids.create_variable_from_name("length(p)"),
+          expression: FunctionCall {
+            name: "length".into(),
+            arguments: vec![Variable { identifier: p }.into()],
+            distinct: false,
+          }
+          .into(),
+        },
+      ],
+      modifiers: Modifiers {
+        distinct: false,
         skip: None,
         limit: None,
         order_by: None,