Skip to content

Refactor rewrite star projection #3136

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
nafraf wants to merge 24 commits into
base: master
Choose a base branch
from
Open

Refactor rewrite star projection #3136

Changes from 1 commit
Commits
Show all changes
24 commits
Select commit Hold shift + click to select a range
57ad9dc
WIP: Refactoring AST_RewriteStarProjections()
nafraf Jul 5, 2023
1137b34
collect_call_projections(): Remove yield_count validation
nafraf Jul 5, 2023
b6620f5
Reorganize shared code
nafraf Jul 5, 2023
ced298c
WIP: _rewrite_call_subquery_star_projections()
nafraf Jul 6, 2023
6cd3c3f
Test CALL{} using CREATE, MERGE, UNWIND, CALL
nafraf Jul 6, 2023
9d47264
replace_clause(): remove unused var scope_start
nafraf Jul 7, 2023
5906b06
Fix intermediate WITH * including new variable
nafraf Jul 7, 2023
fa6756d
Add pointer casting
nafraf Jul 7, 2023
ee1b813
Merge branch 'master' into rewrite_star_projection
nafraf Jul 7, 2023
2b9b226
Remove changes of PR #3139
nafraf Jul 10, 2023
f2e5b19
_Validate_RETURN_Clause(): Check RETURN * without vars in scope
nafraf Jul 11, 2023
4b82445
Merge branch 'master' into rewrite_star_projection
nafraf Jul 11, 2023
4b29940
Check if vars of RETURN * are part of outer scope
nafraf Jul 11, 2023
a86f7d5
Remove raxClone()
nafraf Jul 11, 2023
37f02b4
Add test CALL { ... RETURN *,var}
nafraf Jul 12, 2023
bd20572
Use vctx->ignore_identifiers in _Validate_RETURN_Clause
nafraf Jul 12, 2023
66aee9f
Add collect_non_star_projections()
nafraf Jul 12, 2023
7a3470c
Remove unneedded case from collect_return_projections()
nafraf Jul 12, 2023
63e8b06
Merge branch 'master' into rewrite_star_projection
nafraf Jul 17, 2023
70d09a9
Remove unneeded cast
nafraf Jul 17, 2023
ba88672
Fix query having WITH * after UNION
nafraf Jul 17, 2023
a622b67
Fix collect projection in _rewrite_call_subquery_star_projections()
nafraf Jul 18, 2023
acf9d3f
Add missing replace_clause in AST_RewriteStarProjections()
nafraf Jul 18, 2023
6382406
Remove unneeded casting
nafraf Jul 18, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
replace_clause(): remove unused var scope_start
  • Loading branch information
@nafraf
nafraf committed Jul 7, 2023
commit 9d47264380870aade2cd92a36c9f9729789aeccb
47 changes: 11 additions & 36 deletions src/ast/ast_rewrite_star_projections.c
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,6 @@ static void replace_clause
(
cypher_astnode_t *root, // ast root
cypher_astnode_t *clause, // clause being replaced
int scope_start, // beginning of scope
int scope_end, // ending of scope
rax *identifiers // bound vars
) {
Expand Down Expand Up @@ -145,7 +144,6 @@ static void replace_clause
// update `nprojections` to actual number of projections
// value might be reduced due to duplicates
nprojections = proj_idx;
//raxFree(identifiers);

// prepare arguments for new return clause node
bool distinct = false ;
Expand Down Expand Up @@ -219,7 +217,6 @@ static void replace_clause
static bool _rewrite_call_subquery_star_projections
(
const cypher_astnode_t *wrapping_clause, // wrapping clause
uint scope_start, // start scope in wrapping clause
uint idx, // index of the call subquery
rax *identifiers // bound vars
) {
Expand All @@ -236,7 +233,6 @@ static bool _rewrite_call_subquery_star_projections

Copy link
Copy Markdown
Collaborator

@raz-mon raz-mon Jul 16, 2023

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

No need for casting here (can also remove the new-line).

Copy link
Copy Markdown
Contributor Author

@nafraf nafraf Jul 18, 2023

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

OK. Casting removed

uint n_clauses = cypher_ast_query_nclauses(query);

uint first_in_scope = 0;
// initialize `last_is_union` to true to rewrite the first importing `WITH`
bool last_is_union = true;
for(uint i = 0; i < n_clauses; i++) {
Expand All @@ -248,25 +244,21 @@ static bool _rewrite_call_subquery_star_projections
cypher_astnode_t *inner_query =
cypher_ast_call_subquery_get_query(call_subquery);
rewritten |= AST_RewriteStarProjections(inner_query);
last_is_union = false;
} else if(t == CYPHER_AST_MATCH) {
// the MATCH clause contains one pattern of N paths
const cypher_astnode_t *pattern =
cypher_ast_match_get_pattern(clause);
collect_aliases_in_pattern(pattern, local_identifiers);
last_is_union = false;
} else if(t == CYPHER_AST_CREATE) {
// the CREATE clause contains one pattern of N paths
const cypher_astnode_t *pattern =
cypher_ast_create_get_pattern(clause);
collect_aliases_in_pattern(pattern, local_identifiers);
last_is_union = false;
} else if(t == CYPHER_AST_MERGE) {
// the MERGE clause contains one path
const cypher_astnode_t *path =
cypher_ast_merge_get_pattern_path(clause);
collect_aliases_in_path(path, local_identifiers);
last_is_union = false;
} else if(t == CYPHER_AST_UNWIND) {
// the UNWIND clause introduces one alias
const cypher_astnode_t *unwind_alias =
Expand All @@ -275,10 +267,8 @@ static bool _rewrite_call_subquery_star_projections
cypher_ast_identifier_get_name(unwind_alias);
raxTryInsert(local_identifiers, (unsigned char *)identifier,
strlen(identifier), (void *)unwind_alias, NULL);
last_is_union = false;
} else if(t == CYPHER_AST_CALL) {
collect_call_projections(clause, local_identifiers);
last_is_union = false;
} else if(t == CYPHER_AST_WITH || t == CYPHER_AST_RETURN) {
// check whether the clause contains a star projection
bool has_star = (t == CYPHER_AST_WITH) ?
Expand All @@ -292,26 +282,18 @@ static bool _rewrite_call_subquery_star_projections
if(has_star) {
if(last_is_union && t == CYPHER_AST_WITH) {
// importing `WITH` clause, import vars from wrapping clause
replace_clause(query, clause, first_in_scope, i,
clone_identifiers);

replace_clause(query, clause, i, clone_identifiers);
const cypher_astnode_t *newclause =
cypher_ast_query_get_clause(query, i);

if(t == CYPHER_AST_WITH) {
collect_with_projections(newclause, local_identifiers);
} else {
collect_return_projections(newclause, local_identifiers);
}
collect_with_projections(newclause, local_identifiers);
} else {
// intermediate `WITH` or `RETURN` clause
if(t == CYPHER_AST_WITH) {
collect_with_projections(clause, local_identifiers);
} else {
collect_return_projections(clause, local_identifiers);
}
replace_clause(query, clause, first_in_scope, i,
local_identifiers);
replace_clause(query, clause, i, local_identifiers);
}
rewritten = true;
} else {
Expand All @@ -323,10 +305,10 @@ static bool _rewrite_call_subquery_star_projections
collect_return_projections(clause, local_identifiers);
}
}
first_in_scope = i;
last_is_union = false;
} else if(t == CYPHER_AST_UNION) {
first_in_scope = i + 1;
}

// update last_is_union
if(t == CYPHER_AST_UNION) {
last_is_union = true;
} else {
last_is_union = false;
Expand All @@ -349,7 +331,6 @@ bool AST_RewriteStarProjections
}

// rewrite all WITH * / RETURN * clauses to include all aliases
uint scope_start = 0;
uint clause_count = cypher_ast_query_nclauses(root);
rax *identifiers = raxNew();

Expand All @@ -358,11 +339,9 @@ bool AST_RewriteStarProjections
cypher_astnode_type_t type = cypher_astnode_type(clause);

if(type == CYPHER_AST_CALL_SUBQUERY) {
// collect_call_subquery_projections(clause, identifiers);
// rewrite the CALL{} before collecting the identifiers
rewritten |=
_rewrite_call_subquery_star_projections(root, scope_start, i,
identifiers);

_rewrite_call_subquery_star_projections(root, i, identifiers);
clause = cypher_ast_query_get_clause(root, i);
collect_call_subquery_projections(clause, identifiers);
} else if(type == CYPHER_AST_MATCH) {
Expand Down Expand Up @@ -394,30 +373,26 @@ bool AST_RewriteStarProjections
if(cypher_ast_with_has_include_existing(clause)) {
// clause contains a star projection, replace it
replace_clause((cypher_astnode_t *)root,
(cypher_astnode_t *)clause, scope_start, i, identifiers);
(cypher_astnode_t *)clause, i, identifiers);
clause = cypher_ast_query_get_clause(root, i);
rewritten = true;
}
// update new scope identifiers
raxFree(identifiers);
identifiers = raxNew();
collect_with_projections(clause, identifiers);
// update scope start
scope_start = i;
} else if(type == CYPHER_AST_RETURN) {
if(cypher_ast_return_has_include_existing(clause)) {
// clause contains a star projection, replace it
replace_clause((cypher_astnode_t *)root,
(cypher_astnode_t *)clause, scope_start, i, identifiers);
(cypher_astnode_t *)clause, i, identifiers);
clause = cypher_ast_query_get_clause(root, i);
rewritten = true;
}
// update new scope identifiers
raxFree(identifiers);
identifiers = raxNew();
collect_return_projections(clause, identifiers);
// update scope start
scope_start = i;
}
}
raxFree(identifiers);
Expand Down