Control Flow Statements
The GSQL Query Language includes a comprehensive set of control flow statements to empower sophisticated graph traversal and data computation with conditional (IF/ELSE, CASE) and iterative (WHILE, and FOREACH) execution.
You may also be interested in CASE expressions.
Differences in Block Syntax
Note that any of these statements can be used as a query-body statement or as a DML-sub level statement.
If the control flow statement is at the query-body level, then its block(s) of statements are query-body statements ( queryBodyStmts ). In a queryBodyStmts block, each individual statement ends with a semicolon, so there is always a semicolon at the end.
If the control flow statement is at the DML-sub level, then its block(s) of statements are DML-sub statements ( dmlSubStmtList ). In a dmlSubStmtList block, a comma separates statements, but there is no punctuation at the end.
For more detailed general examples of the difference between query-body statements and DML-sub statements, see statement types.
IF Statement
The IF statement provides conditional branching: execute a block of statements ( queryBodyStmts or dmlSubStmtList ) only if a given condition is true.
The IF statement allows for zero or more ELSE-IF clauses, followed by an optional ELSE clause.
The IF statement can be used either at the query-body level or at the DML-sub-statement level. (See the note about differences in block syntax.)
Syntax
IF syntaxqueryBodyIfStmt := IF condition THEN queryBodyStmts
[ELSE IF condition THEN queryBodyStmts ]*
[ELSE queryBodyStmts ] END
dmlSubIfStmt := IF condition THEN dmlSubStmtList
[ELSE IF condition THEN dmlSubStmtList ]*
[ELSE dmlSubStmtList ] END
If an IF condition is not true, then the flow proceeds to the next ELSE IF condition.
When a true condition is encountered, its corresponding block of statements is executed, and then the IF statement terminates (skipping any remaining ELSE-IF or ELSE clauses).
If an ELSE clause is present, its block of statements are executed if none of the preceding conditions are true.
Overall, the functionality can be summarized as "execute the first block of statements whose conditional test is true."
// if then
IF x == 5 THEN y = 10; END; # y is assigned to 10 only if x is 5.
// if then else
IF x == 5 THEN y = 10; # y is 10 only if x is 5.
ELSE y = 20; END; # y is 20 only if x is NOT 5.
// if with ELSE IF
IF x == 5 THEN y = 10; # y is 10 only if x is 5.
ELSE IF x == 7 THEN y = 5; # y is 5 only if x is 7.
ELSE y = 20; END; # y is 20 only if x is NOT 5 and NOT 7.
Examples
This query counts the number of friends a person has, and optionally include coworkers in that count
CREATE QUERY count_friends_of2(VERTEX<Person> seed, BOOL include_coworkers) FOR GRAPH Friend_Net
{
SumAccum<INT> @@num_friends = 0;
start = {seed};
IF include_coworkers THEN
friends = SELECT v FROM start -((Friend | Coworker):e)- :v
ACCUM @@num_friends +=1;
ELSE
friends = SELECT v FROM start -(Friend:e)- :v
ACCUM @@num_friends +=1;
END;
PRINT @@num_friends, include_coworkers;
}
RUN QUERY count_friends_of2("person2", false)
{
"error": false,
"message": "",
"version": {
"schema": 0,
"edition": "enterprise",
"api": "v2"
},
"results": [{
"@@num_friends": 2,
"include_coworkers": false
}]
}
This query uses a more advanced activity calculation, taking into account number of posts and number of likes that a user made.
CREATE QUERY calculate_in_depth_activity(VERTEX<Person> seed) FOR GRAPH Social_Net
{
SumAccum<INT> @@number_posts = 0;
SumAccum<INT> @@number_likes = 0;
start = {seed};
result = SELECT post_vertex FROM start -(Posted>:e)- :post_vertex
ACCUM @@number_posts += 1;
result = SELECT liked_post FROM start -(Liked>:e)- :liked_post
ACCUM @@number_likes += 1;
IF @@number_posts < 2 THEN
IF @@number_likes < 1 THEN
PRINT "Not very active";
ELSE
PRINT "Semi-active";
END;
ELSE IF @@number_posts < 3 THEN
IF @@number_likes < 2 THEN
PRINT "Semi-active";
ELSE
PRINT "Active";
END;
ELSE
PRINT "Very active";
END;
}
RUN QUERY calculate_in_depth_activity("person1")
{
"error": false,
"message": "",
"version": {
"edition": "developer",
"schema": 0,
"api": "v2"
},
"results": [{"Semi-active": "Semi-active"}]
}
CASE Statement
The CASE statement provides conditional branching: execute a block of statements only if a given condition is true.
CASE statements can be used as query-body statements or DML-sub-statements. (See note about differences in block syntax.)
Syntax
CASE syntaxqueryBodyCaseStmt := CASE (WHEN condition THEN queryBodyStmts)+ [ELSE queryBodyStmts] END
| CASE expr (WHEN constant THEN queryBodyStmts)+ [ELSE queryBodyStmts] END
dmlSubCaseStmt := CASE (WHEN condition THEN dmlSubStmtList)+ [ELSE dmlSubStmtList] END
| CASE expr (WHEN constant THEN dmlSubStmtList)+ [ELSE dmlSubStmtList] END
One CASE statement contains one or more WHEN-THEN clauses, each WHEN presenting one expression.
The CASE statement may also have one ELSE clause whose statements are executed if none of the preceding conditions are true.
There are two syntax of the CASE statement: one equivalent to an if-else statement, and the other is structured like a switch statement.
The if-else version evaluates the boolean condition within each WHEN clause and executes the first block of statements whose condition is true.
The optional concluding ELSE clause is executed only if all WHEN clause conditions are false.
The switch version evaluates the expression following the keyword WHEN and compares its value to the expression immediately following the keyword CASE.
These expressions do not need to be boolean; the CASE statement compares pairs of expressions to see if their values are equal.
The first WHEN-THEN clause to have an expression value equal to the CASE expression value is executed; the remaining clauses are skipped. The optional ELSE clause is executed only if no WHEN clause expression has a value matching the CASE value.
STRING drink = "Juice";
// CASE statement: if-else version
CASE
WHEN drink == "Juice" THEN @@calories += 50
WHEN drink == "Soda" THEN @@calories += 120
...
ELSE @@calories = 0 // Optional else-clause
END
// Since drink = "Juice", 50 will be added to calories
// CASE statement: switch version
CASE drink
WHEN "Juice" THEN @@calories += 50
WHEN "Soda" THEN @@calories += 120
...
ELSE @@calories = 0 // Optional else-clause
END
// Since drink = "Juice", 50 will be added to calories
Examples
This query tallies male and female friends of the starting vertex.
CREATE QUERY count_gender_of_friends(VERTEX<Person> seed) FOR GRAPH Social_Net{
SumAccum<INT> @@males = 0;
SumAccum<INT> @@females = 0;
SumAccum<INT> @@unknown = 0;
starting_vertex = {seed};
people = SELECT v FROM starting_vertex -(Friend:e)-:v
ACCUM
CASE v.gender
WHEN "Male" THEN @@males += 1
WHEN "Female" THEN @@females +=1
ELSE @@unknown += 1
END;
PRINT @@males, @@females, @@unknown;
}
RUN QUERY countGenderOfFriends("person4")
{
"error": false,
"message": "",
"version": {
"edition": "developer",
"schema": 0,
"api": "v2"
},
"results": [{
"@@males": 2,
"@@unknown": 0,
"@@females": 1
}]
}
This query gives each user post a rating based on the subject and how many likes it has.
CREATE QUERY rate_posts() FOR GRAPH Social_Net API("v2") {
SumAccum<INT> @rating = 0;
all_people = {Person.*};
results = SELECT v FROM all_people -(:e)- Post:v
ACCUM CASE e.type
WHEN "Posted" THEN
CASE
WHEN v.subject == "cats" THEN v.@rating += -1
WHEN v.subject == "Graphs" THEN v.@rating += 2
WHEN v.subject == "tigergraph" THEN v.@rating += 10
END
WHEN "Liked" THEN v.@rating += 3 END;
PRINT results[results.@rating];
}
GSQL > RUN QUERY rate_posts()
{
"error": false,
"message": "",
"version": {
"edition": "developer",
"schema": 0,
"api": "v2"
},
"results": [{"results": [
{
"v_id": "0",
"attributes": {"results.@rating": 11},
"v_type": "post"
},
{
"v_id": "10",
"attributes": {"results.@rating": 2},
"v_type": "post"
},
{
"v_id": "2",
"attributes": {"results.@rating": 0},
"v_type": "post"
},
{
"v_id": "4",
"attributes": {"results.@rating": 6},
"v_type": "post"
},
{
"v_id": "9",
"attributes": {"results.@rating": -1},
"v_type": "post"
},
{
"v_id": "3",
"attributes": {"results.@rating": 2},
"v_type": "post"
},
{
"v_id": "5",
"attributes": {"results.@rating": 10},
"v_type": "post"
},
{
"v_id": "7",
"attributes": {"results.@rating": 2},
"v_type": "post"
},
{
"v_id": "1",
"attributes": {"results.@rating": 10},
"v_type": "post"
},
{
"v_id": "11",
"attributes": {"results.@rating": -1},
"v_type": "post"
},
{
"v_id": "8",
"attributes": {"results.@rating": 2},
"v_type": "post"
},
{
"v_id": "6",
"attributes": {"results.@rating": 13},
"v_type": "post"
}
]}]
}
WHILE Statement
The WHILE statement provides unbounded iteration over a block of statements. WHILE statements can be used as query-body statements or DML-sub-statements. (See the note about differences in block syntax.)
Syntax
queryBodyWhileStmt := WHILE condition [LIMIT simpleSize] DO queryBodyStmts END
dmlSubWhileStmt := WHILE condition [LIMIT simpleSize] DO dmlSubStmtList END
simpleSize := integer | varName | paramName
The WHILE statement iterates over its body ( queryBodyStmts or dmlSubStmtList ) until the condition evaluates to false or until the iteration limit is met. A condition is any expression that evaluates to a boolean.
The condition is evaluated before each iteration. CONTINUE statements can be used to change the control flow within the while block. BREAK statements can be used to exit the while loop.
A WHILE statement may have an optional LIMIT clause.
LIMIT clauses has a constant positive integer value or integer variable to constrain the maximum number of loop iterations.
The example below demonstrates how the LIMIT behaves.
|
If a limit value is not specified, it is possible for a WHILE loop to iterate infinitely. It is the responsibility of the query author to design the condition logic so that it is guaranteed to eventually be true (or to set a limit). |
The following three WHILE statements behave the same.
Each terminates when v.size == 0 or after 5 iterations of the loop.
WHILE v.size() !=0 LIMIT 5 DO
// Statements
END;
INT iter = 0;
WHILE (v.size() !=0) AND (iter < 5) DO
// Statements
iter = iter + 1;
END;
INT iter = 0;
WHILE v.size() !=0 DO
IF iter == 5 THEN BREAK; END;
// Statements
iter = iter + 1;
END;
Examples
Below are a number of examples that demonstrate the use of WHILE statements.
The following query finds all vertices which are reachable from a starting seed vertex (i.e., breadth-first search)
CREATE QUERY reachable(VERTEX<Person> seed) FOR GRAPH Work_Net
{
OrAccum @visited;
reachable_vertices = {};
visited_vertices (ANY) = {seed};
// Loop terminates when all neighbors are visited
WHILE visited_vertices.size() !=0 DO
// s is all neighbors of visited_vertices which have not been visited
visited_vertices = SELECT s
FROM visited_vertices-(:e)-:s
WHERE s.@visited == FALSE
POST-ACCUM
s.@visited = true;
reachable_vertices = reachable_vertices UNION visited_vertices;
END;
PRINT reachable_vertices;
}
RUN QUERY reachable("person1")
{
"error": false,
"message": "",
"version": {
"edition": "developer",
"schema": 0,
"api": "v2"
},
"results": [{"reachableVertices": [
{
"v_id": "person3",
"attributes": {
"interestList": ["teaching"],
"skillSet": [ 6, 1, 4 ],
"skillList": [ 4, 1, 6 ],
"locationId": "jp",
"interestSet": ["teaching"],
"@visited": true,
"id": "person3"
},
"v_type": "person"
},
{
"v_id": "person9",
"attributes": {
"interestList": [ "financial", "teaching" ],
"skillSet": [ 2, 7, 4 ],
"skillList": [ 4, 7, 2 ],
"locationId": "us",
"interestSet": [ "teaching", "financial" ],
"@visited": true,
"id": "person9"
},
"v_type": "person"
},
{
"v_id": "person4",
"attributes": {
"interestList": ["football"],
"skillSet": [ 10, 1, 4 ],
"skillList": [ 4, 1, 10 ],
"locationId": "us",
"interestSet": ["football"],
"@visited": true,
"id": "person4"
},
"v_type": "person"
},
{
"v_id": "person7",
"attributes": {
"interestList": [ "art", "sport" ],
"skillSet": [ 6, 8 ],
"skillList": [ 8, 6 ],
"locationId": "us",
"interestSet": [ "sport", "art" ],
"@visited": true,
"id": "person7"
},
"v_type": "person"
},
{
"v_id": "person1",
"attributes": {
"interestList": [ "management", "financial" ],
"skillSet": [ 3, 2, 1 ],
"skillList": [ 1, 2, 3 ],
"locationId": "us",
"interestSet": [ "financial", "management" ],
"@visited": true,
"id": "person1"
},
"v_type": "person"
},
{
"v_id": "person5",
"attributes": {
"interestList": [ "sport", "financial", "engineering" ],
"skillSet": [ 5, 2, 8 ],
"skillList": [ 8, 2, 5 ],
"locationId": "can",
"interestSet": [ "engineering", "financial", "sport" ],
"@visited": true,
"id": "person5"
},
"v_type": "person"
},
{
"v_id": "person6",
"attributes": {
"interestList": [ "music", "art" ],
"skillSet": [ 10, 7 ],
"skillList": [ 7, 10 ],
"locationId": "jp",
"interestSet": [ "art", "music" ],
"@visited": true,
"id": "person6"
},
"v_type": "person"
},
{
"v_id": "person2",
"attributes": {
"interestList": ["engineering"],
"skillSet": [ 6, 5, 3, 2 ],
"skillList": [ 2, 3, 5, 6 ],
"locationId": "chn",
"interestSet": ["engineering"],
"@visited": true,
"id": "person2"
},
"v_type": "person"
},
{
"v_id": "person8",
"attributes": {
"interestList": ["management"],
"skillSet": [ 2, 5, 1 ],
"skillList": [ 1, 5, 2 ],
"locationId": "chn",
"interestSet": ["management"],
"@visited": true,
"id": "person8"
},
"v_type": "person"
},
{
"v_id": "company3",
"attributes": {
"country": "jp",
"@visited": true,
"id": "company3"
},
"v_type": "company"
},
{
"v_id": "company2",
"attributes": {
"country": "chn",
"@visited": true,
"id": "company2"
},
"v_type": "company"
},
{
"v_id": "company1",
"attributes": {
"country": "us",
"@visited": true,
"id": "company1"
},
"v_type": "company"
},
{
"v_id": "person10",
"attributes": {
"interestList": [ "football", "sport" ],
"skillSet": [3],
"skillList": [3],
"locationId": "us",
"interestSet": [ "sport", "football" ],
"@visited": true,
"id": "person10"
},
"v_type": "person"
}
]}]
}
This query finds all vertices reachable within two hops from a starting seed vertex (i.e., breadth-first search)
CREATE QUERY reachable_within_two(VERTEX<Person> seed) FOR GRAPH Work_Net
{
OrAccum @visited;
reachable_vertices = {};
visited_vertices (ANY) = {seed};
// loop terminates when all neighbors within 2-hops of the seed vertex are visited
WHILE visited_vertices.size() !=0 LIMIT 2 DO
// s is all neighbors of visited_vertices which have not been visited
visited_vertices = SELECT s
FROM visited_vertices-(:e)-:s
WHERE s.@visited == false
POST-ACCUM
s.@visited = true;
reachable_vertices = reachable_vertices UNION visited_vertices;
END;
PRINT reachable_vertices;
}
RUN QUERY reachable_within_two("person1")
[
{
"reachable_vertices": [
{
"attributes": {
"@visited": true,
"id": "person6",
"interest_list": [
"music",
"art"
],
"interest_set": [
"music",
"art"
],
"location_id": "jp",
"skill_list": [
7,
10
],
"skill_set": [
7,
10
]
},
"v_id": "person6",
"v_type": "Person"
},
{
"attributes": {
"@visited": true,
"country": "chn",
"id": "company2"
},
"v_id": "company2",
"v_type": "Company"
},
{
"attributes": {
"@visited": true,
"country": "us",
"id": "company1"
},
"v_id": "company1",
"v_type": "Company"
},
{
"attributes": {
"@visited": true,
"id": "person10",
"interest_list": [
"football",
"sport"
],
"interest_set": [
"football",
"sport"
],
"location_id": "us",
"skill_list": [
3
],
"skill_set": [
3
]
},
"v_id": "person10",
"v_type": "Person"
},
{
"attributes": {
"@visited": true,
"id": "person4",
"interest_list": [
"football"
],
"interest_set": [
"football"
],
"location_id": "us",
"skill_list": [
4,
1,
10
],
"skill_set": [
4,
1,
10
]
},
"v_id": "person4",
"v_type": "Person"
},
{
"attributes": {
"@visited": true,
"id": "person3",
"interest_list": [
"teaching"
],
"interest_set": [
"teaching"
],
"location_id": "jp",
"skill_list": [
4,
1,
6
],
"skill_set": [
4,
1,
6
]
},
"v_id": "person3",
"v_type": "Person"
},
{
"attributes": {
"@visited": true,
"id": "person7",
"interest_list": [
"art",
"sport"
],
"interest_set": [
"art",
"sport"
],
"location_id": "us",
"skill_list": [
8,
6
],
"skill_set": [
8,
6
]
},
"v_id": "person7",
"v_type": "Person"
},
{
"attributes": {
"@visited": true,
"id": "person2",
"interest_list": [
"engineering"
],
"interest_set": [
"engineering"
],
"location_id": "chn",
"skill_list": [
2,
3,
5,
6
],
"skill_set": [
2,
3,
5,
6
]
},
"v_id": "person2",
"v_type": "Person"
},
{
"attributes": {
"@visited": true,
"id": "person1",
"interest_list": [
"management",
"financial"
],
"interest_set": [
"management",
"financial"
],
"location_id": "us",
"skill_list": [
1,
2,
3
],
"skill_set": [
1,
2,
3
]
},
"v_id": "person1",
"v_type": "Person"
},
{
"attributes": {
"@visited": true,
"id": "person5",
"interest_list": [
"sport",
"financial",
"engineering"
],
"interest_set": [
"sport",
"financial",
"engineering"
],
"location_id": "can",
"skill_list": [
8,
2,
5
],
"skill_set": [
8,
2,
5
]
},
"v_id": "person5",
"v_type": "Person"
},
{
"attributes": {
"@visited": true,
"id": "person8",
"interest_list": [
"management"
],
"interest_set": [
"management"
],
"location_id": "chn",
"skill_list": [
1,
5,
2
],
"skill_set": [
1,
5,
2
]
},
"v_id": "person8",
"v_type": "Person"
},
{
"attributes": {
"@visited": true,
"id": "person9",
"interest_list": [
"financial",
"teaching"
],
"interest_set": [
"financial",
"teaching"
],
"location_id": "us",
"skill_list": [
4,
7,
2
],
"skill_set": [
4,
7,
2
]
},
"v_id": "person9",
"v_type": "Person"
}
]
}
]
FOREACH Statement
The FOREACH statement provides bounded iteration over a block of statements. FOREACH statements can be used as query-body statements or DML-sub-statements.
See the note about differences in block syntax.
Syntax
FOREACH syntaxqueryBodyForEachStmt := FOREACH forEachControl DO queryBodyStmts END
dmlSubForEachStmt := FOREACH forEachControl DO dmlSubStmtList END
forEachControl := ( iterationVar | "(" keyVar ("," valueVar)+ ")") (IN | ":") setBagExpr
| iterationVar IN RANGE "[" expr "," expr"]" ["." STEP(" expr ")"]
iterationVar := name
keyVar := name
valueVar := name
The formal syntax for forEachControl appears can be broken down into the following cases:
-
name IN setBagExpr -
(key, value) pair IN setBagExpr// because it’s a Map -
name IN RANGE [ expr, expr ] -
name IN RANGE [ expr, expr ].STEP ( expr )
Note that setBagExpr includes container accumulators and explicit sets.
Limitations
The FOREACH statement has the following restrictions:
-
In a DML-sub level
FOREACH, it is never permissible to update the loop variable (the variable declared beforeIN, e.g.,varin “FOREACH var IN setBagExpr”). -
In a query-body level
FOREACH, in most cases it is not permissible to update the loop variable. The following exceptions apply:-
If the iteration is over a
ListAccum, its values can be updated. -
If the iteration is over a
MapAccum, its values can be updated, but its keys cannot.
-
-
If the iteration is over a set of vertices, it is not permissible to access (read or write) their vertex-attached accumulators.
-
A query-body-level
FOREACHcannot iterate over a set or bag of constants. For example,FOREACHi in (1,2,3) is not supported. However, DML-subFOREACHdoes support this.
FOREACH … IN RANGE
The FOREACH statement has an optional RANGE clause RANGE[expr, expr], which can be used to define the iteration collection.
Optionally, the range may specify a step size:
RANGE[expr, expr].STEP(expr)
Each expr must evaluate to an integer.
Any of the integers may be negative, but the step expr may not be 0.
The clause RANGE[a,b].STEP(c) produces the sequence of integers from a to b, inclusive, with step size c.
That is,
\((a, a+c, a+2*c, a+3*c, ... a+k*c)\), where k = the largest integer such that \(|k*c| ≤ |b-a|\).
If the .STEP method is not given, then the step size c = 1.
Example with the Traditional Method
FOREACH i IN RANGE[0, 5].STEP(2) DO
PRINT "Loop 1: index=" + to_string(i);
END;
Alternative Method (Using range())
As of TigerGraph 4.1, FOREACH also supports using the range() function as an alternative. The syntax for this is:
FOREACH i IN range(0, 5, 2) DO
PRINT "Loop 2: index=" + to_string(i);
END;
|
FOREACH IN RANGE with MapAccumCREATE QUERY foreach_range_ex() FOR GRAPH Social_Net {
ListAccum<INT> @@t;
Start = {Person.*};
FOREACH i IN RANGE[0, 2] DO
@@t += i;
L = SELECT Start
FROM Start
WHERE Start.id == "person1"
ACCUM
FOREACH j IN RANGE[0, i] DO
@@t += j
END;
END;
PRINT @@t;
}
GSQL > RUN QUERY foreach_range_ex()
{
"error": false,
"message": "",
"version": {
"edition": "developer",
"schema": 0,
"api": "v2"
},
"results": [{"@@t": [ 0, 0, 1, 0, 1, 2, 0, 1, 2 ]}]
}
The following FOREACH statement uses a specified step value for iterating through the range.
CREATE QUERY foreach_range_step(INT a, INT b, INT c) FOR GRAPH Minimal_Net {
ListAccum<INT> @@t;
FOREACH i IN RANGE[a,b].step(c) DO
@@t += i;
END;
PRINT @@t;
}
The step value can be positive for an ascending range or negative for a descending range. If the step has the wrong polarity, then the loop has zero iterations; that is, the exit condition is already satisfied.
RUN QUERY foreach_range_step(100,0,-9)
{
"error": false,
"message": "",
"version": {
"schema": 0,
"edition": "enterprise",
"api": "v2"
},
"results": [{"@@t": [
100,
91,
82,
73,
64,
55,
46,
37,
28,
19,
10,
1
]}]
}
Query-body-level FOREACH Examples
This query counts the number of companies whose country matches the provided string.
CREATE QUERY company_count(STRING country_name) FOR GRAPH Work_Net {
ListAccum<STRING> @@company_list;
INT country_count;
start = {ANY};
// Select all vertices with type "Company", and append country attribute from all company vertices to list accum
s = SELECT v FROM start:v
WHERE v.type == "Company"
ACCUM @@company_list += v.country;
// Iterate the ListAccum and compare each element to the country_name parameter
FOREACH item in @@company_list DO
IF item == country_name THEN
country_count = country_count + 1;
END;
END;
PRINT country_count;
}
RUN QUERY company_count("us")
{
"error": false,
"message": "",
"version": {
"schema": 0,
"edition": "enterprise",
"api": "v2"
},
"results": [{"country_count": 2}]
}
This query counts the number of employees from a given country and list their IDs.
CREATE QUERY employee_by_country(STRING country_name) FOR GRAPH Work_Net {
MapAccum <STRING, ListAccum<STRING>> @@employees;
// Start will have a set of all person type vertices
start = {Person.*};
// Build a map using person location_id as a key and a list of strings to hold multiple person ids
s = SELECT v FROM start:v
ACCUM @@employees += (v.location_id -> v.id);
// Iterate the map using (key,value) pairs
FOREACH (key,val) IN @@employees DO
IF key == country_name THEN
PRINT val.size();
// Nested foreach to iterate over the list of person ids
FOREACH employee in val DO
PRINT employee;
END;
// MapAccum keys are unique so we can BREAK out of the loop
BREAK;
END;
END;
}
RUN QUERY employee_by_country("us")
{
"error": false,
"message": "",
"version": {
"schema": 0,
"edition": "enterprise",
"api": "v2"
},
"results": [
{"val.size()": 5},
{"employee": "person7"},
{"employee": "person1"},
{"employee": "person4"},
{"employee": "person9"},
{"employee": "person10"}
]
}
DML-sub FOREACH Examples
This query shows post topics liked by users and show total likes per topic.
CREATE QUERY topic_likes() FOR GRAPH Social_Net {
SetAccum<STRING> @@person_osts;
SumAccum<INT> @post_likes;
MapAccum<STRING,INT> @@likes_by_topic;
start = {Person.*};
// Find all user posts and generate a set of post topics
// (set has no duplicates)
posts = SELECT g FROM start - (Posted>) - :g
ACCUM @@person_posts += g.subject;
// Use set of topics to increment how many times a specific
// post is liked by other users
liked_posts = SELECT f FROM start - (Liked>) - :f
ACCUM FOREACH x in @@person_posts DO
CASE WHEN (f.subject == x) THEN
f.@post_likes += 1
END
END
// Aggregate all liked totals by topic
POST-ACCUM @@likes_by_topic += (f.subject -> f.@post_likes);
// Display the number of likes per topic
PRINT @@likes_by_topic;
}
GSQL > RUN QUERY topic_likes()
{
"error": false,
"message": "",
"version": {
"edition": "developer",
"schema": 0,
"api": "v2"
},
"results": [{"@@likesByTopic": {
"cats": 3,
"coffee": 2,
"Graphs": 3,
"tigergraph": 1
}}]
}
This query shows a summary of the number of friends all persons have by gender.
CREATE QUERY friend_gender() FOR GRAPH Social_Net {
ListAccum<STRING> @friend_gender;
SumAccum<INT> @@male_gender_count;
SumAccum<INT> @@female_gender_count;
start = {Person.*};
// Record a list showing each friend's gender
social_members = SELECT s FROM start:s -(Friend)- :g
ACCUM s.@friend_gender += (g.gender)
// Loop over each list of genders and total them
POST-ACCUM
FOREACH a in s.@friend_gender DO
CASE WHEN (a == "Male") THEN
@@male_gender_count += 1
ELSE
@@female_gender_count += 1
END
END;
PRINT @@male_gender_count;
PRINT @@female_gender_count;
}
GSQL > RUN QUERY friend_gender()
{
"error": false,
"message": "",
"version": {
"edition": "developer",
"schema": 0,
"api": "v2"
},
"results": [
{"@@maleGenderCount": 11},
{"@@femaleGenderCount": 7}
]
}
CONTINUE and BREAK Statements
The CONTINUE and BREAK statements can only be used within a block of a WHILE or FOREACH statement.
The CONTINUE statement branches control flow to the end of the loop, skipping any remaining statements in the current iteration, and proceeding to the next iteration.
That is, everything in the loop block after the CONTINUE statement will be skipped, and then the loop will continue as normal.
The BREAK statement branches control flow out of the loop, i.e., it will exit the loop and stop iteration.
Examples
Below are a number of examples that demonstrate the use of BREAK and CONTINUE.
// While loop with continue statement
INT i = 0;
INT nCount = 0;
WHILE i < 10 DO
i = i + 1;
IF (i % 2 == 0) { CONTINUE; }
nCount = nCount + 1;
END;
// i is 10, nCount is 5 (skips the increment for every even i).
// While loop with a break statement
i = 0;
WHILE i < 10 DO
IF (i == 5) { BREAK; } # When i is 5 the loop is exited
i = i + 1;
END;
// i is now 5
This query finds posts of a given person, and post of friends of that person, friends of friends, etc. until a post about cats is found. The number of friend-hops to reach is the 'degree' of cats.
CREATE QUERY find_degree_of_cats(VERTEX<Person> seed) FOR GRAPH Social_Net
{
SumAccum<INT> @@degree = 0;
OrAccum @@found_cat_post = FALSE;
OrAccum @visited = FALSE;
friends (ANY) = {seed};
WHILE @@found_cat_post != true AND friends.size() > 0 DO
posts = SELECT v FROM friends-(Posted>:e)-:v
ACCUM CASE WHEN v.subject == "cats" THEN
@@found_cat_post += true
END;
IF @@found_cat_post THEN
BREAK;
END;
friends = SELECT v FROM friends-(Friend:e)-:v
WHERE v.@visited == false
ACCUM v.@visited = true;
@@degree += 1;
END;
PRINT @@degree;
}
RUN QUERY find_degree_of_cats("person2")
{
"error": false,
"message": "",
"version": {
"schema": 0,
"edition": "enterprise",
"api": "v2"
},
"results": [{"@@degree": 2}]
}
This query finds all 3-hop friends of a starting vertex. Count coworkers as friends if there are not enough friends
CREATE QUERY find_enough_friends(VERTEX<Person> seed) FOR GRAPH Friend_Net
{
// Keep track of the distance from the seed
SumAccum<INT> @@distance = 0;
OrAccum @visited = false;
visited_vertices = {seed};
WHILE true LIMIT 3 DO
@@distance += 1;
// Traverse from visited_vertices to its friends
friends = SELECT v
FROM visited_vertices -(Friend:e)- :v
WHERE v.@visited == false
POST-ACCUM v.@visited = true;
PRINT @@distance, friends;
// If number of friends at this level is sufficient, finish this iteration
IF visited_vertices.size() >= 2 THEN
visited_vertices = friends;
CONTINUE;
END;
// If fewer than 4 friends, add in coworkers
coworkers = SELECT v
FROM visited_vertices -(Coworker:e)- :v
WHERE v.@visited == false
POST-ACCUM v.@visited = true;
visited_vertices = friends UNION coworkers;
PRINT @@distance, coworkers;
END;
}
RUN QUERY find_enough_friends("person1")
{
"error": false,
"message": "",
"version": {
"schema": 0,
"edition": "enterprise",
"api": "v2"
},
"results": [
{
"@@distance": 1,
"friends": [
{
"v_id": "person4",
"attributes": {
"@visited": true,
"id": "person4"
},
"v_type": "Person"
},
{
"v_id": "person2",
"attributes": {
"@visited": true,
"id": "person2"
},
"v_type": "Person"
},
{
"v_id": "person3",
"attributes": {
"@visited": true,
"id": "person3"
},
"v_type": "Person"
}
]
},
{
"coworkers": [
{
"v_id": "person5",
"attributes": {
"@visited": true,
"id": "person5"
},
"v_type": "Person"
},
{
"v_id": "person6",
"attributes": {
"@visited": true,
"id": "person6"
},
"v_type": "Person"
}
],
"@@distance": 1
},
{
"@@distance": 2,
"friends": [
{
"v_id": "person1",
"attributes": {
"@visited": true,
"id": "person1"
},
"v_type": "Person"
},
{
"v_id": "person8",
"attributes": {
"@visited": true,
"id": "person8"
},
"v_type": "Person"
},
{
"v_id": "person9",
"attributes": {
"@visited": true,
"id": "person9"
},
"v_type": "Person"
}
]
},
{
"@@distance": 3,
"friends": [
{
"v_id": "person7",
"attributes": {
"@visited": true,
"id": "person7"
},
"v_type": "Person"
},
{
"v_id": "person10",
"attributes": {
"@visited": true,
"id": "person10"
},
"v_type": "Person"
},
{
"v_id": "person12",
"attributes": {
"@visited": true,
"id": "person12"
},
"v_type": "Person"
}
]
}
]
}