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(batch,optimizer): support column pruning for iceberg source #16429

Merged
merged 2 commits into from
Apr 22, 2024

Conversation

chenzl25
Copy link
Contributor

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

What's changed and what's your intention?

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.

Copy link
Contributor

@st1page st1page left a comment

Choose a reason for hiding this comment

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

Rest LGTM

Comment on lines 86 to 89
if required_cols.is_empty() {
let mapping = ColIndexMapping::with_remaining_columns(required_cols, self.schema().len());
// If reuqiured_cols is empty, we use the first column of iceberg to avoid the empty schema.
LogicalProject::with_mapping(self.clone_with_required_cols(&[0]).into() , mapping).into()
Copy link
Contributor

Choose a reason for hiding this comment

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

So iceberg scan does not support scan no fields? But the RW scan can do it (only the row count is meaningful in this case)

dev=> explain select 1 from t;
                QUERY PLAN
-------------------------------------------
 BatchExchange { order: [], dist: Single }
 └─BatchProject { exprs: [1:Int32] }
   └─BatchScan { table: t, columns: [] }
(3 rows)

Copy link
Contributor

Choose a reason for hiding this comment

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

This is a very corner case, but I am wondering if Iceberg supports creating a table without any columns. It should be possible in PG

postgres=# create table t();
CREATE TABLE
postgres=# \d+ t;
                                          Table "public.t"
 Column | Type | Collation | Nullable | Default | Storage | Compression | Stats target | Description
--------+------+-----------+----------+---------+---------+-------------+--------------+-------------
Access method: heap

Copy link
Contributor Author

@chenzl25 chenzl25 Apr 22, 2024

Choose a reason for hiding this comment

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

So iceberg scan does not support scan no fields? But the RW scan can do it (only the row count is meaningful in this case)

Actually, the underlying sdk icelake doesn't handle the empty schema well and it will treat empty schema as all columns are required, so we need to make a trick here.

This is a very corner case, but I am wondering if Iceberg supports creating a table without any columns. It should be possible in PG

Iceberg doesn't support creating a table without any columns.

@chenzl25 chenzl25 added this pull request to the merge queue Apr 22, 2024
Merged via the queue into main with commit cf2dda6 Apr 22, 2024
27 of 28 checks passed
@chenzl25 chenzl25 deleted the dylan/support_column_pruning_for_iceberg branch April 22, 2024 05:43
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