[ https://issues.apache.org/jira/browse/HIVE-24084?focusedWorklogId=482134&page=com.atlassian.jira.plugin.system.issuetabpanels:worklog-tabpanel#worklog-482134 ]
ASF GitHub Bot logged work on HIVE-24084: ----------------------------------------- Author: ASF GitHub Bot Created on: 11/Sep/20 15:43 Start Date: 11/Sep/20 15:43 Worklog Time Spent: 10m Work Description: jcamachor commented on a change in pull request #1439: URL: https://github.com/apache/hive/pull/1439#discussion_r487130820 ########## File path: ql/src/java/org/apache/hadoop/hive/ql/optimizer/calcite/rules/HiveAggregateJoinTransposeRule.java ########## @@ -303,6 +305,90 @@ public void onMatch(RelOptRuleCall call) { } } + /** + * Determines weather the give grouping is unique. + * + * Consider a join which might produce non-unique rows; but later the results are aggregated again. + * This method determines if there are sufficient columns in the grouping which have been present previously as unique column(s). + */ + private boolean isGroupingUnique(RelNode input, ImmutableBitSet groups) { + if (groups.isEmpty()) { + return false; + } + RelMetadataQuery mq = input.getCluster().getMetadataQuery(); + Set<ImmutableBitSet> uKeys = mq.getUniqueKeys(input); + for (ImmutableBitSet u : uKeys) { + if (groups.contains(u)) { + return true; + } + } + if (input instanceof Join) { + Join join = (Join) input; + RexBuilder rexBuilder = input.getCluster().getRexBuilder(); + SimpleConditionInfo cond = new SimpleConditionInfo(join.getCondition(), rexBuilder); + + if (cond.valid) { + ImmutableBitSet newGroup = groups.intersect(ImmutableBitSet.fromBitSet(cond.fields)); + RelNode l = join.getLeft(); + RelNode r = join.getRight(); + + int joinFieldCount = join.getRowType().getFieldCount(); + int lFieldCount = l.getRowType().getFieldCount(); + + ImmutableBitSet groupL = newGroup.get(0, lFieldCount); + ImmutableBitSet groupR = newGroup.get(lFieldCount, joinFieldCount).shift(-lFieldCount); + + if (isGroupingUnique(l, groupL)) { Review comment: OK. If it turns out there are many changes and it may need some time to be fixed, feel free to defer to follow-up JIRA and let's merge this one. ---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: us...@infra.apache.org Issue Time Tracking ------------------- Worklog Id: (was: 482134) Time Spent: 2h 50m (was: 2h 40m) > Push Aggregates thru joins in case it re-groups previously unique columns > ------------------------------------------------------------------------- > > Key: HIVE-24084 > URL: https://issues.apache.org/jira/browse/HIVE-24084 > Project: Hive > Issue Type: Improvement > Reporter: Zoltan Haindrich > Assignee: Zoltan Haindrich > Priority: Major > Labels: pull-request-available > Time Spent: 2h 50m > Remaining Estimate: 0h > -- This message was sent by Atlassian Jira (v8.3.4#803005)