zabetak commented on a change in pull request #1325: URL: https://github.com/apache/hive/pull/1325#discussion_r467318976
########## File path: ql/src/java/org/apache/hadoop/hive/ql/optimizer/SemiJoinReductionMerge.java ########## @@ -0,0 +1,399 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one + * or more contributor license agreements. See the NOTICE file + * distributed with this work for additional information + * regarding copyright ownership. The ASF licenses this file + * to you under the Apache License, Version 2.0 (the + * "License"); you may not use this file except in compliance + * with the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +package org.apache.hadoop.hive.ql.optimizer; + +import org.apache.hadoop.hive.conf.HiveConf; +import org.apache.hadoop.hive.ql.exec.ColumnInfo; +import org.apache.hadoop.hive.ql.exec.FilterOperator; +import org.apache.hadoop.hive.ql.exec.GroupByOperator; +import org.apache.hadoop.hive.ql.exec.Operator; +import org.apache.hadoop.hive.ql.exec.OperatorFactory; +import org.apache.hadoop.hive.ql.exec.OperatorUtils; +import org.apache.hadoop.hive.ql.exec.ReduceSinkOperator; +import org.apache.hadoop.hive.ql.exec.RowSchema; +import org.apache.hadoop.hive.ql.exec.SelectOperator; +import org.apache.hadoop.hive.ql.exec.TableScanOperator; +import org.apache.hadoop.hive.ql.exec.Utilities; +import org.apache.hadoop.hive.ql.io.AcidUtils; +import org.apache.hadoop.hive.ql.parse.GenTezUtils; +import org.apache.hadoop.hive.ql.parse.ParseContext; +import org.apache.hadoop.hive.ql.parse.RuntimeValuesInfo; +import org.apache.hadoop.hive.ql.parse.SemanticAnalyzer; +import org.apache.hadoop.hive.ql.parse.SemanticException; +import org.apache.hadoop.hive.ql.parse.SemiJoinBranchInfo; +import org.apache.hadoop.hive.ql.plan.AggregationDesc; +import org.apache.hadoop.hive.ql.plan.DynamicValue; +import org.apache.hadoop.hive.ql.plan.ExprNodeColumnDesc; +import org.apache.hadoop.hive.ql.plan.ExprNodeConstantDesc; +import org.apache.hadoop.hive.ql.plan.ExprNodeDesc; +import org.apache.hadoop.hive.ql.plan.ExprNodeDynamicValueDesc; +import org.apache.hadoop.hive.ql.plan.ExprNodeGenericFuncDesc; +import org.apache.hadoop.hive.ql.plan.FilterDesc; +import org.apache.hadoop.hive.ql.plan.GroupByDesc; +import org.apache.hadoop.hive.ql.plan.PlanUtils; +import org.apache.hadoop.hive.ql.plan.ReduceSinkDesc; +import org.apache.hadoop.hive.ql.plan.SelectDesc; +import org.apache.hadoop.hive.ql.plan.TableDesc; +import org.apache.hadoop.hive.ql.udf.generic.GenericUDAFBloomFilter; +import org.apache.hadoop.hive.ql.udf.generic.GenericUDAFEvaluator; +import org.apache.hadoop.hive.ql.udf.generic.GenericUDAFMax; +import org.apache.hadoop.hive.ql.udf.generic.GenericUDAFMin; +import org.apache.hadoop.hive.ql.udf.generic.GenericUDFBetween; +import org.apache.hadoop.hive.ql.udf.generic.GenericUDFInBloomFilter; +import org.apache.hadoop.hive.ql.udf.generic.GenericUDFMurmurHash; +import org.apache.hadoop.hive.ql.udf.generic.GenericUDFOPAnd; +import org.apache.hadoop.hive.ql.util.NullOrdering; +import org.apache.hadoop.hive.serde2.typeinfo.TypeInfo; +import org.apache.hadoop.hive.serde2.typeinfo.TypeInfoFactory; + +import java.util.ArrayDeque; +import java.util.ArrayList; +import java.util.Arrays; +import java.util.Collections; +import java.util.Comparator; +import java.util.Deque; +import java.util.EnumSet; +import java.util.HashMap; +import java.util.List; +import java.util.Map; +import java.util.SortedMap; +import java.util.TreeMap; + +public class SemiJoinReductionMerge extends Transform { + + public ParseContext transform(ParseContext parseContext) throws SemanticException { + Map<ReduceSinkOperator, SemiJoinBranchInfo> map = parseContext.getRsToSemiJoinBranchInfo(); + if (map.isEmpty()) { + return parseContext; + } + HiveConf hiveConf = parseContext.getConf(); + + // Order does not really matter but it is necessary to keep plans stable + SortedMap<SJSourceTarget, List<ReduceSinkOperator>> sameTableSJ = + new TreeMap<>(Comparator.comparing(SJSourceTarget::toString)); + for (Map.Entry<ReduceSinkOperator, SemiJoinBranchInfo> smjEntry : map.entrySet()) { + TableScanOperator ts = smjEntry.getValue().getTsOp(); + // Semijoin optimization branch should look like <Parent>-SEL-GB1-RS1-GB2-RS2 + SelectOperator selOp = OperatorUtils.ancestor(smjEntry.getKey(), SelectOperator.class, 0, 0, 0, 0); + assert selOp != null; + assert selOp.getParentOperators().size() == 1; + Operator<?> source = selOp.getParentOperators().get(0); + SJSourceTarget sjKey = new SJSourceTarget(source, ts); + List<ReduceSinkOperator> ops = sameTableSJ.computeIfAbsent(sjKey, tableScanOperator -> new ArrayList<>()); + ops.add(smjEntry.getKey()); + } + for (Map.Entry<SJSourceTarget, List<ReduceSinkOperator>> sjMergeCandidate : sameTableSJ.entrySet()) { + final List<ReduceSinkOperator> sjBrances = sjMergeCandidate.getValue(); + if (sjBrances.size() < 2) { + continue; + } + // Order does not really matter but it is necessary to keep plans stable + sjBrances.sort(Comparator.comparing(Operator::getIdentifier)); + + List<SelectOperator> selOps = new ArrayList<>(sjBrances.size()); + for (ReduceSinkOperator rs : sjBrances) { + selOps.add(OperatorUtils.ancestor(rs, SelectOperator.class, 0, 0, 0, 0)); + } + SelectOperator selectOp = mergeSelectOps(sjMergeCandidate.getKey().source, selOps); + + GroupByOperator gbPartialOp = createGroupBy(selectOp, selectOp, GroupByDesc.Mode.HASH, hiveConf); + + ReduceSinkOperator rsPartialOp = createReduceSink(gbPartialOp, NullOrdering.defaultNullOrder(hiveConf)); + rsPartialOp.getConf().setReducerTraits(EnumSet.of(ReduceSinkDesc.ReducerTraits.QUICKSTART)); + + GroupByOperator gbCompleteOp = createGroupBy(selectOp, rsPartialOp, GroupByDesc.Mode.FINAL, hiveConf); + + ReduceSinkOperator rsCompleteOp = createReduceSink(gbCompleteOp, NullOrdering.defaultNullOrder(hiveConf)); + + final TableScanOperator sjTargetTable = sjMergeCandidate.getKey().target; + SemiJoinBranchInfo sjInfo = new SemiJoinBranchInfo(sjTargetTable, false); + parseContext.getRsToSemiJoinBranchInfo().put(rsCompleteOp, sjInfo); + + // Save the info that is required at query time to resolve dynamic/runtime values. + RuntimeValuesInfo valuesInfo = createRuntimeValuesInfo(rsCompleteOp, sjBrances, parseContext); + parseContext.getRsToRuntimeValuesInfoMap().put(rsCompleteOp, valuesInfo); + + ExprNodeGenericFuncDesc sjPredicate = createSemiJoinPredicate(sjBrances, valuesInfo, parseContext); + + // Update filter operators with the new semi-join predicate + for (Operator<?> op : sjTargetTable.getChildOperators()) { + if (op instanceof FilterOperator) { + FilterDesc filter = ((FilterOperator) op).getConf(); + filter.setPredicate(and(Arrays.asList(filter.getPredicate(), sjPredicate))); + } + } + // Update tableScan with the new semi-join predicate + sjTargetTable.getConf().setFilterExpr(and(Arrays.asList(sjTargetTable.getConf().getFilterExpr(), sjPredicate))); + + for (ReduceSinkOperator rs : sjBrances) { + GenTezUtils.removeSemiJoinOperator(parseContext, rs, sjTargetTable); + GenTezUtils.removeBranch(rs); + } + + // TODO How to associate multi-cols with gb ? + // parseContext.getColExprToGBMap().put(key, gb); + } + return parseContext; + } + + private static ExprNodeGenericFuncDesc createSemiJoinPredicate(List<ReduceSinkOperator> sjBranches, + RuntimeValuesInfo sjValueInfo, ParseContext context) { + // Performance note: To speed-up evaluation 'BETWEEN' predicates should come before the 'IN_BLOOM_FILTER' + Deque<String> dynamicIds = new ArrayDeque<>(sjValueInfo.getDynamicValueIDs()); + List<ExprNodeDesc> sjPredicates = new ArrayList<>(); + List<ExprNodeDesc> hashArgs = new ArrayList<>(); + for (ReduceSinkOperator rs : sjBranches) { + RuntimeValuesInfo info = context.getRsToRuntimeValuesInfoMap().get(rs); + assert info.getTargetColumns().size() == 1; + final ExprNodeDesc targetColumn = info.getTargetColumns().get(0); + TypeInfo typeInfo = targetColumn.getTypeInfo(); + DynamicValue minDynamic = new DynamicValue(dynamicIds.poll(), typeInfo); + DynamicValue maxDynamic = new DynamicValue(dynamicIds.poll(), typeInfo); + + List<ExprNodeDesc> betweenArgs = Arrays.asList( + // Use false to not invert between result + new ExprNodeConstantDesc(Boolean.FALSE), + targetColumn, + new ExprNodeDynamicValueDesc(minDynamic), + new ExprNodeDynamicValueDesc(maxDynamic)); + ExprNodeDesc betweenExp = + new ExprNodeGenericFuncDesc(TypeInfoFactory.booleanTypeInfo, new GenericUDFBetween(), "between", betweenArgs); + sjPredicates.add(betweenExp); + hashArgs.add(targetColumn); + } + + ExprNodeDesc hashExp = + new ExprNodeGenericFuncDesc(TypeInfoFactory.intTypeInfo, new GenericUDFMurmurHash(), "hash", hashArgs); + + assert dynamicIds.size() == 1 : "There should be one column left untreated the one with the bloom filter"; + DynamicValue bloomDynamic = new DynamicValue(dynamicIds.poll(), TypeInfoFactory.binaryTypeInfo); + sjPredicates.add( + new ExprNodeGenericFuncDesc(TypeInfoFactory.booleanTypeInfo, new GenericUDFInBloomFilter(), "in_bloom_filter", + Arrays.asList(hashExp, new ExprNodeDynamicValueDesc(bloomDynamic)))); + return and(sjPredicates); + } + + private static RuntimeValuesInfo createRuntimeValuesInfo(ReduceSinkOperator rs, List<ReduceSinkOperator> sjBranches, + ParseContext parseContext) { + List<ExprNodeDesc> valueCols = rs.getConf().getValueCols(); + RuntimeValuesInfo info = new RuntimeValuesInfo(); + TableDesc rsFinalTableDesc = + PlanUtils.getReduceValueTableDesc(PlanUtils.getFieldSchemasFromColumnList(valueCols, "_col")); + List<String> dynamicValueIDs = new ArrayList<>(); + for (ExprNodeDesc rsCol : valueCols) { + dynamicValueIDs.add(rs.toString() + rsCol.getExprString()); + } + + info.setTableDesc(rsFinalTableDesc); + info.setDynamicValueIDs(dynamicValueIDs); + info.setColExprs(valueCols); + List<ExprNodeDesc> targetTableExpressions = new ArrayList<>(); + for (ReduceSinkOperator sjBranch : sjBranches) { + RuntimeValuesInfo sjInfo = parseContext.getRsToRuntimeValuesInfoMap().get(sjBranch); + assert sjInfo.getTargetColumns().size() == 1; + targetTableExpressions.add(sjInfo.getTargetColumns().get(0)); + } + info.setTargetColumns(targetTableExpressions); + return info; + } + + private static SelectOperator mergeSelectOps(Operator<?> parent, List<SelectOperator> selectOperators) { + List<String> colNames = new ArrayList<>(); + List<ExprNodeDesc> colDescs = new ArrayList<>(); + List<ColumnInfo> columnInfos = new ArrayList<>(); + Map<String, ExprNodeDesc> selectColumnExprMap = new HashMap<>(); + for (SelectOperator sel : selectOperators) { + for (ExprNodeDesc col : sel.getConf().getColList()) { + String colName = HiveConf.getColumnInternalName(colDescs.size()); + colNames.add(colName); + columnInfos.add(new ColumnInfo(colName, col.getTypeInfo(), "", false)); + colDescs.add(col); + selectColumnExprMap.put(colName, col); + } + } + ExprNodeDesc hashExp = + new ExprNodeGenericFuncDesc(TypeInfoFactory.intTypeInfo, new GenericUDFMurmurHash(), "hash", colDescs); + String hashName = HiveConf.getColumnInternalName(colDescs.size() + 1); + colNames.add(hashName); + columnInfos.add(new ColumnInfo(hashName, hashExp.getTypeInfo(), "", false)); + + List<ExprNodeDesc> selDescs = new ArrayList<>(colDescs); + selDescs.add(hashExp); + + SelectDesc select = new SelectDesc(selDescs, colNames); + SelectOperator selectOp = + (SelectOperator) OperatorFactory.getAndMakeChild(select, new RowSchema(columnInfos), parent); + selectOp.setColumnExprMap(selectColumnExprMap); + return selectOp; + } + + private static ReduceSinkOperator createReduceSink(Operator<?> parentOp, NullOrdering nullOrder) + throws SemanticException { + List<ExprNodeDesc> valueCols = new ArrayList<>(); + RowSchema parentSchema = parentOp.getSchema(); + List<String> outColNames = new ArrayList<>(); + for (int i = 0; i < parentSchema.getSignature().size(); i++) { + ColumnInfo colInfo = parentSchema.getSignature().get(i); + ExprNodeColumnDesc colExpr = new ExprNodeColumnDesc(colInfo.getType(), colInfo.getInternalName(), "", false); + valueCols.add(colExpr); + outColNames.add(SemanticAnalyzer.getColumnInternalName(i)); + } + + ReduceSinkDesc rsDesc = PlanUtils + .getReduceSinkDesc(Collections.emptyList(), valueCols, outColNames, false, -1, 0, 1, + AcidUtils.Operation.NOT_ACID, nullOrder); + rsDesc.setColumnExprMap(Collections.emptyMap()); + return (ReduceSinkOperator) OperatorFactory.getAndMakeChild(rsDesc, new RowSchema(parentSchema), parentOp); + } + + private static GroupByOperator createGroupBy(SelectOperator selectOp, Operator<?> parentOp, GroupByDesc.Mode gbMode, + HiveConf hiveConf) { + + final List<ExprNodeDesc> params; + final GenericUDAFEvaluator.Mode udafMode = SemanticAnalyzer.groupByDescModeToUDAFMode(gbMode, false); + switch (gbMode) { + case FINAL: + params = createGroupByAggregationParameters((ReduceSinkOperator) parentOp); + break; + case HASH: + params = createGroupByAggregationParameters(selectOp); + break; + default: + throw new AssertionError(gbMode.toString() + " is not supported"); + } + + List<AggregationDesc> gbAggs = new ArrayList<>(); + Deque<ExprNodeDesc> paramsCopy = new ArrayDeque<>(params); + while (paramsCopy.size() > 1) { + gbAggs.add(minAggregation(udafMode, paramsCopy.poll())); + gbAggs.add(maxAggregation(udafMode, paramsCopy.poll())); + } + gbAggs.add(bloomFilterAggregation(udafMode, paramsCopy.poll(), selectOp, hiveConf)); + assert paramsCopy.size() == 0; + + List<String> gbOutputNames = new ArrayList<>(gbAggs.size()); + List<ColumnInfo> gbColInfos = new ArrayList<>(gbAggs.size()); + for (int i = 0; i < params.size(); i++) { + String colName = HiveConf.getColumnInternalName(i); + gbOutputNames.add(colName); + final TypeInfo colType; + if (i == params.size() - 1) { + colType = TypeInfoFactory.binaryTypeInfo; // Bloom type + } else { + colType = params.get(i).getTypeInfo(); // Min/Max type + } + gbColInfos.add(new ColumnInfo(colName, colType, "", false)); + } + + float groupByMemoryUsage = HiveConf.getFloatVar(hiveConf, HiveConf.ConfVars.HIVEMAPAGGRHASHMEMORY); + float memoryThreshold = HiveConf.getFloatVar(hiveConf, HiveConf.ConfVars.HIVEMAPAGGRMEMORYTHRESHOLD); + float minReductionHashAggr = HiveConf.getFloatVar(hiveConf, HiveConf.ConfVars.HIVEMAPAGGRHASHMINREDUCTION); + GroupByDesc groupBy = + new GroupByDesc(gbMode, gbOutputNames, Collections.emptyList(), gbAggs, false, groupByMemoryUsage, + memoryThreshold, minReductionHashAggr, null, false, -1, false); + groupBy.setColumnExprMap(Collections.emptyMap()); + return (GroupByOperator) OperatorFactory.getAndMakeChild(groupBy, new RowSchema(gbColInfos), parentOp); + } + + private static List<ExprNodeDesc> createGroupByAggregationParameters(SelectOperator selectOp) { + List<ExprNodeDesc> params = new ArrayList<>(); + // The first n-1 cols are used as parameters for min & max so we need two expressions + for (ColumnInfo c : selectOp.getSchema().getSignature()) { + String name = c.getInternalName(); + ExprNodeColumnDesc p = new ExprNodeColumnDesc(new ColumnInfo(name, c.getType(), "", false)); + params.add(p); + params.add(p); + } + // The last col is used as parameter for bloom so we need only one expression + params.remove(params.size() - 1); + return params; + } + + private static List<ExprNodeDesc> createGroupByAggregationParameters(ReduceSinkOperator reduceOp) { + List<ExprNodeDesc> params = new ArrayList<>(); + // There is a 1-1 mapping between columns and parameters for the aggregation functions min, max, bloom + for (ColumnInfo c : reduceOp.getSchema().getSignature()) { + String name = Utilities.ReduceField.VALUE + "." + c.getInternalName(); + params.add(new ExprNodeColumnDesc(new ColumnInfo(name, c.getType(), "", false))); + } + return params; + } + + private static ExprNodeGenericFuncDesc and(List<ExprNodeDesc> args) { + return new ExprNodeGenericFuncDesc(TypeInfoFactory.booleanTypeInfo, new GenericUDFOPAnd(), "and", args); + } + + private static AggregationDesc minAggregation(GenericUDAFEvaluator.Mode mode, ExprNodeDesc col) { + List<ExprNodeDesc> p = Collections.singletonList(col); + return new AggregationDesc("min", new GenericUDAFMin.GenericUDAFMinEvaluator(), p, false, mode); + } + + private static AggregationDesc maxAggregation(GenericUDAFEvaluator.Mode mode, ExprNodeDesc col) { + List<ExprNodeDesc> p = Collections.singletonList(col); + return new AggregationDesc("max", new GenericUDAFMax.GenericUDAFMaxEvaluator(), p, false, mode); + } + + private static AggregationDesc bloomFilterAggregation(GenericUDAFEvaluator.Mode mode, ExprNodeDesc col, + SelectOperator source, HiveConf conf) { + GenericUDAFBloomFilter.GenericUDAFBloomFilterEvaluator bloomFilterEval = + new GenericUDAFBloomFilter.GenericUDAFBloomFilterEvaluator(); + bloomFilterEval.setSourceOperator(source); + bloomFilterEval.setMaxEntries(conf.getLongVar(HiveConf.ConfVars.TEZ_MAX_BLOOM_FILTER_ENTRIES)); + bloomFilterEval.setMinEntries(conf.getLongVar(HiveConf.ConfVars.TEZ_MIN_BLOOM_FILTER_ENTRIES)); + bloomFilterEval.setFactor(conf.getFloatVar(HiveConf.ConfVars.TEZ_BLOOM_FILTER_FACTOR)); + // TODO Setup hints + List<ExprNodeDesc> p = Collections.singletonList(col); + AggregationDesc bloom = new AggregationDesc("bloom_filter", bloomFilterEval, p, false, mode); + // TODO Why do we need to set it explicitly? Review comment: Removed in https://github.com/apache/hive/pull/1325/commits/bf5cad3aa0861eef811565214231e5dec87cd3e6 and created https://issues.apache.org/jira/browse/HIVE-24018 if I ever feel like investigating further. ---------------------------------------------------------------- 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 --------------------------------------------------------------------- To unsubscribe, e-mail: gitbox-unsubscr...@hive.apache.org For additional commands, e-mail: gitbox-h...@hive.apache.org
