-
Notifications
You must be signed in to change notification settings - Fork 412
Add delete file index to pyiceberg and support equality delete reads #2255
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
base: main
Are you sure you want to change the base?
Conversation
|
I noticed that this PR addresses the same issue/feature as the one I was working on in here. However, your implementation is more complete (by supporting reading equality deletes and deletion vectors), so I think it makes sense to move forward with this one instead. (cc: @sungwy, since you reviewed my PR) |
|
oops, sorry @gabeiglio, I was searching for positional deletes in github search and i didnt see that you were already working on it in that PR. Looks like there are some parts of the PR that is still super useful to get merged, like the validates. |
|
Yea exactly, should have been more clear on my message, my implementation for DeleteFileIndex was a scope creep to achieve the validation. so now that PR can be only for the validation instead of partition maps, delete file index, etc. :) @kevinjqliu |
sungwy
left a comment
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Hi @geruh - thanks for working on this PR, and sorry for the delayed review.
I've added some review feedback. Let me know your thoughts!
|
@sungwy Thanks a lot! I have done the suggested changes, could you take another look at it? |
sungwy
left a comment
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Hi @rutb327 thank you for continuing to work on the PR!
I've added a few more suggestions after taking longer time reading your implementation and the test suite. Hope you find this helpful!
| if self.dv: | ||
| if not self.dv_sorted: | ||
| self.dv_values = sorted(self.dv.values(), key=lambda x: x[1]) | ||
| self.dv_sorted = True |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Instead of needing to track whether we are sorted here is there a better data structure that just is sorted that we could use?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
We went with the lazy sort pattern here to follow the Java implementation. This allows us to add each file in O(1), then sort once in n log n. We technically could use sorted containers SortedList for roughly the same performance. WDYT
|
Also I will test, but will this DeleteFileIndex class properly map deletes to data files? In the current implementation pyiceberg maps dangling delete files |
|
Hi, I hope this PR can be included in 0.11.0.
It seems that position delete files with If this approach makes sense, would you consider adding a similar change to this PR? |
Co-authored-by: Sung Yun <[email protected]>
Closes #1210
Summary
This work was primarily done by @rutb327 while I provided guidance!
This PR adds equality delete read support to PyIceberg by implementing the delete file indexing system that matches delete files to data files, mimicking the behavior found in Iceberg Core. With this implementation we are able to index files and now read equality deletes during table scans.
Design details
Delete File Index
The new
DeleteFileIndexclass centralizes handling of all delete file types: positional deletes, equality deletes, and deletion vectors. It organizes deletes by type (equality vs. positional), partition (usingPartitionMapfor spec-aware grouping), and path (for path-specific positional deletes). This enables efficient lookup during table scans, reducing unnecessary delete file processing.Equality Delete support
Equality delete files are loaded as PyArrow Tables with their respective equality ids for the schema and for each we are grouping tables with the same set equality id's to reduce anti join operations.
Testing
Added tests from the core iceberg DeleteFileIndex test suite and added some tests with dummy files. As well as some manual testing with a flink setup.
Are there any user-facing changes?
Yes can read tables with equality deletes