Skip to content
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.

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

[Enhancement] eliminate constant cte #53286

Open
wants to merge 1 commit into
base: main
Choose a base branch
from

Conversation

stephen-shelby
Copy link
Contributor

@stephen-shelby stephen-shelby commented Nov 27, 2024

Why I'm doing:

If all leaf nodes of cte are constant, we can directly inline this cte.After inline, we can perform more optimizations, such as
with cte as (select 111) select * from cte join (select * from (select * from t1 join cte) s1) s2;
we can eliminate join node by EliminateJoinWithConstantRule to prevent error join order.

image

What I'm doing:

Fixes #issue

What type of PR is this:

  • BugFix
  • Feature
  • Enhancement
  • Refactor
  • UT
  • Doc
  • Tool

Does this PR entail a change in behavior?

  • Yes, this PR will result in a change in behavior.
  • No, this PR will not result in a change in behavior.

If yes, please specify the type of change:

  • Interface/UI changes: syntax, type conversion, expression evaluation, display information
  • Parameter changes: default values, similar parameters but with different default values
  • Policy changes: use new policy to replace old one, functionality automatically enabled
  • Feature removed
  • Miscellaneous: upgrade & downgrade compatibility, etc.

Checklist:

  • I have added test cases for my bug fix or my new feature
  • This pr needs user documentation (for new or modified features or behaviors)
    • I have added documentation for my new feature or new function
  • This is a backport pr

Bugfix cherry-pick branch check:

  • I have checked the version labels which the pr will be auto-backported to the target branch
    • 3.4
    • 3.3
    • 3.2
    • 3.1
    • 3.0

@stephen-shelby stephen-shelby requested a review from a team as a code owner November 27, 2024 14:09
return true;
}

}
Copy link

Choose a reason for hiding this comment

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

The most risky bug in this code is:
The transformation logic assumes all children of the CTE are leaves, but if a non-leaf is encountered, it prematurely stops by returning only direct inputs instead of transforming deeper structures. This can skip necessary transformations on nested structures.

You can modify the code like this:

@Override
public List<OptExpression> transform(OptExpression input, OptimizerContext context) {
    return flattenCTE(input);
}

private List<OptExpression> flattenCTE(OptExpression root) {
    List<OptExpression> flattened = new ArrayList<>();
    for (OptExpression optExpression : root.getInputs()) {
        if (areAllLeafNodesConstants(optExpression)) {
            flattened.addAll(optExpression.getInputs());
        } else {
            flattened.add(optExpression);
        }
    }
    return flattened;
}

This modifies the transform method to directly flatten CTEs that consist entirely of constant expressions by further expanding into their inputs while preserving the structure when not all nodes are constants.

@stephen-shelby stephen-shelby changed the title [Enhancement] eliminate constant cte [UT] eliminate constant cte Nov 27, 2024
Copy link

[FE Incremental Coverage Report]

pass : 1 / 1 (100.00%)

file detail

path covered_line new_line coverage not_covered_line_detail
🔵 com/starrocks/sql/optimizer/Optimizer.java 1 1 100.00% []

@stephen-shelby stephen-shelby changed the title [UT] eliminate constant cte [Enhancement] eliminate constant cte Nov 27, 2024
Copy link

sonarcloud bot commented Nov 27, 2024

Copy link

[Java-Extensions Incremental Coverage Report]

pass : 0 / 0 (0%)

Copy link

[BE Incremental Coverage Report]

pass : 0 / 0 (0%)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant