- Notifications
You must be signed in to change notification settings - Fork30
Provide groupByKey shortcuts for groupBy.as#213
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 ourterms of service andprivacy statement. We’ll occasionally send you account related emails.
Already on GitHub?Sign in to your account
Merged
Merged
Uh oh!
There was an error while loading.Please reload this page.
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.Learn more about bidirectional Unicode characters
github-actionsbot commentedDec 8, 2023 • edited
Loading Uh oh!
There was an error while loading.Please reload this page.
edited
Uh oh!
There was an error while loading.Please reload this page.
Test Results 566 files ± 0 566 suites ±0 1h 29m 12s⏱️ +11s Results for commit411afe8. ± Comparison against base commit8314de4. This pull requestremoves 28 andadds 30 tests.Note that renamed tests count towards both.♻️ This comment has been updated with latest results. |
Sign up for freeto join this conversation on GitHub. Already have an account?Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
This provides shortcuts for
groupBy(...).as[...]that make it easier to use column-basedgroupByKey.Calling
Dataset.groupBy(...).as[K, T]should be preferred over callingDataset.groupByKey(...)whenever possible. The former allows Catalyst to exploit existing partitioning and ordering of the Dataset, while the latter hides from Catalyst which columns are used to create the keys.When the dataset is already partitioned and ordered by the grouping columns,
Dataset.groupByKey(...)will repartition and order the entire dataset again.Example:
Calling
ds.groupByKey(_.id)hides from Catalyst that columnidis the grouping key, whileds.groupBy($"id").as[Int, V]tells Catalyst thatdsis to be grouped by (partitioned and ordered by) columnid.The new column-based
groupByKeymethods make it easier for users to find a way to express the grouping by expressions. Looking at theDatasetAPI, the user findsgroupByKeywithColumn. The existinggroupBymethod returns aRelationalGroupedDataset, which provides theas[K, V]method, which allows for the same semantics, but is difficult to find.The new column-based
groupByKeymethods further do not require the user to specify the typeVof the originalDataset[V], asgroupByKeyhas access to the type / encoder:vs.