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

feat(frontend): ban jsonb in aggregation stream key #14442

Merged
merged 5 commits into from
Jan 10, 2024

Conversation

yuhao-su
Copy link
Contributor

@yuhao-su yuhao-su commented Jan 9, 2024

I hereby agree to the terms of the RisingWave Labs, Inc. Contributor License Agreement.

What's changed and what's your intention?

#14314

Checklist

  • I have written necessary rustdoc comments
  • I have added necessary unit tests and integration tests
  • I have added test labels as necessary. See details.
  • I have added fuzzing tests or opened an issue to track them. (Optional, recommended for new SQL features Sqlsmith: Sql feature generation #7934).
  • My PR contains breaking changes. (If it deprecates some features, please create a tracking issue to remove them in the future).
  • All checks passed in ./risedev check (or alias, ./risedev c)
  • My PR changes performance-critical code. (Please run macro/micro-benchmarks and show the results.)
  • My PR contains critical fixes that are necessary to be merged into the latest release. (Please check out the details)

Documentation

  • My PR needs documentation updates. (Please use the Release note section below to summarize the impact on users)

Release note

If this PR includes changes that directly affect users or other significant modifications relevant to the community, kindly draft a release note to provide a concise summary of these changes. Please prioritize highlighting the impact these changes will have on users.

Comment on lines 379 to 384
if let Some(err) = StreamKeyChecker.visit(self.plan.clone()) {
return Err(ErrorCode::NotSupported(
err,
"Use a large Jsonb as part of the stream key is unexpected. If you are sure your Jsonb is small, use `set XXX true`".to_string(),
).into());
}
Copy link
Contributor

Choose a reason for hiding this comment

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

I think we should check the stream key at least after the logical optimization phase because after column pruning, some unnecessary columns check could be skipped.

Copy link
Contributor Author

Choose a reason for hiding this comment

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

The plan nodes after optimization can be different from the plan after binding and cause inaccurate error messages.

The columns checked are group by order by partition by, which is unlikely to be pruned.

Copy link
Contributor

Choose a reason for hiding this comment

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

But for union, it seems hard to say? e.g. select a, b, c from ( select * from t1 union all select * from t2)

Copy link
Contributor

@st1page st1page Jan 9, 2024

Choose a reason for hiding this comment

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

you mean `union` here ? actually I am not sure about the behavior here... can we prune the column here? in other words, I consider they are not equalevant ```SQL SELECT a,b FROM ( SELECT distinct a,b,c FROM t ) t;

SELECT distinct a,b FROM FROM t;

Copy link
Contributor Author

Choose a reason for hiding this comment

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

fixed an issue about union all

Copy link
Contributor

Choose a reason for hiding this comment

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

Agg can be created during subquery unnesting. See the sample query here: #7981 (comment)

Btw is join key a concern as well?

Copy link
Contributor Author

Choose a reason for hiding this comment

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

Yes, this or only handles user explicitly write agg queries. We need a fallback to handle other cases.

Yes, join can be a concern.

Copy link
Member

@fuyufjh fuyufjh left a comment

Choose a reason for hiding this comment

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

LGTM except the error messages.

@yuhao-su yuhao-su added this pull request to the merge queue Jan 10, 2024
Merged via the queue into main with commit 3b3d0de Jan 10, 2024
26 of 27 checks passed
@yuhao-su yuhao-su deleted the yuhao/ban_jsonb_in_stream_key branch January 10, 2024 10:29
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.

5 participants