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(iceberg): Optimization iceberg source count(*) query #18067

Closed
wants to merge 8 commits into from

Conversation

xxhZs
Copy link
Contributor

@xxhZs xxhZs commented Aug 16, 2024

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

What's changed and what's your intention?

#17958

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.

@xxhZs xxhZs changed the title feat(iceberg): Optimization iceberg count(*) query feat(iceberg): Optimization iceberg source count(*) query Aug 16, 2024
@xxhZs xxhZs force-pushed the xxh/add-iceberg-count branch 2 times, most recently from 0d32982 to 2552e1f Compare August 19, 2024 06:38
@xxhZs xxhZs requested a review from chenzl25 August 19, 2024 06:39
fmt

fix ci
@xxhZs xxhZs force-pushed the xxh/add-iceberg-count branch from 2552e1f to ae95e1e Compare August 19, 2024 06:45
Comment on lines +61 to +72
enum SourceType {
UNSPECIFIED = 0;
SCAN = 1;
COUNT_STAR = 2;
}
uint32 source_id = 1;
repeated plan_common.ColumnCatalog columns = 2;
map<string, string> with_properties = 3;
repeated bytes split = 4;
catalog.StreamSourceInfo info = 5;
map<string, secret.SecretRef> secret_refs = 6;
SourceType source_type = 7;
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 this new field is too special to be a part of the SourceNode proto because it only meaningful for iceberg. Considering we will try to support predicate pushdown for iceberg scan, I think we can separate iceberg scan from source node just like file scan.

@@ -196,49 +199,81 @@ pub enum IcebergTimeTravelInfo {
}

impl IcebergSplitEnumerator {
pub async fn list_splits_batch(
pub async fn list_splits_batch_count_star(
Copy link
Contributor

Choose a reason for hiding this comment

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

If it is a count star query, we don't need to return multiple splits, i.e. we don't need to run it in a distributed way.

use crate::error::BatchError;
use crate::executor::{DataChunk, Executor};

pub struct IcebergCountStarExecutor {
Copy link
Contributor

Choose a reason for hiding this comment

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

It seems we don't even need a special iceberg count start executor, ValuesExecutor can be used instead.

@xxhZs xxhZs closed this Sep 19, 2024
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.

2 participants