diff --git common/src/java/org/apache/hadoop/hive/conf/HiveConf.java common/src/java/org/apache/hadoop/hive/conf/HiveConf.java index 41a6dc0..948d0cc 100644 --- common/src/java/org/apache/hadoop/hive/conf/HiveConf.java +++ common/src/java/org/apache/hadoop/hive/conf/HiveConf.java @@ -346,6 +346,7 @@ public class HiveConf extends Configuration { // Optimizer HIVEOPTCP("hive.optimize.cp", true), // column pruner + HIVEOPTAUTOINDEX("hive.optimize.autoindex", false), // automatically use indexes HIVEOPTPPD("hive.optimize.ppd", true), // predicate pushdown // push predicates down to storage handlers HIVEOPTPPD_STORAGE("hive.optimize.ppd.storage", true), diff --git conf/hive-default.xml conf/hive-default.xml index 7a254db..4f4c014 100644 --- conf/hive-default.xml +++ conf/hive-default.xml @@ -319,6 +319,12 @@ + hive.optimize.autoindex + false + Whether to enable automatic use of indexes + + + hive.optimize.ppd true Whether to enable predicate pushdown diff --git ql/src/java/org/apache/hadoop/hive/ql/exec/ExecDriver.java ql/src/java/org/apache/hadoop/hive/ql/exec/ExecDriver.java index 6437385..c5c8ddc 100644 --- ql/src/java/org/apache/hadoop/hive/ql/exec/ExecDriver.java +++ ql/src/java/org/apache/hadoop/hive/ql/exec/ExecDriver.java @@ -273,9 +273,13 @@ public class ExecDriver extends Task implements Serializable, Hadoop job.setNumReduceTasks(work.getNumReduceTasks().intValue()); job.setReducerClass(ExecReducer.class); + // Set hive input format, and input format file if necessary. if (work.getInputformat() != null) { HiveConf.setVar(job, HiveConf.ConfVars.HIVEINPUTFORMAT, work.getInputformat()); } + if (work.getIndexIntermediateFile() != null) { + job.set("hive.index.compact.file", work.getIndexIntermediateFile()); + } // Turn on speculative execution for reducers boolean useSpeculativeExecReducers = HiveConf.getBoolVar(job, diff --git ql/src/java/org/apache/hadoop/hive/ql/exec/TableScanOperator.java ql/src/java/org/apache/hadoop/hive/ql/exec/TableScanOperator.java index c02d90b..0fd63f5 100644 --- ql/src/java/org/apache/hadoop/hive/ql/exec/TableScanOperator.java +++ ql/src/java/org/apache/hadoop/hive/ql/exec/TableScanOperator.java @@ -22,6 +22,7 @@ import java.io.Serializable; import java.util.ArrayList; import java.util.List; +import org.apache.commons.lang.StringUtils; import org.apache.hadoop.conf.Configuration; import org.apache.hadoop.fs.Path; import org.apache.hadoop.hive.common.FileUtils; @@ -53,6 +54,8 @@ public class TableScanOperator extends Operator implements private transient String partitionSpecs; private transient boolean inputFileChanged = false; private TableDesc tableDesc; + private String indexInputFormat; + private String indexIntermediateFile; public TableDesc getTableDesc() { @@ -63,6 +66,26 @@ public class TableScanOperator extends Operator implements this.tableDesc = tableDesc; } + public String getIndexInputFormat() { + return indexInputFormat; + } + + public void setIndexInputFormat(String indexInputFormat) { + this.indexInputFormat = indexInputFormat; + } + + public String getIndexIntermediateFile() { + return indexIntermediateFile; + } + + public void setIndexIntermediateFile(String fileName) { + this.indexIntermediateFile = fileName; + } + + public boolean usesIndex() { + return StringUtils.isNotEmpty(indexInputFormat) && StringUtils.isNotEmpty(indexIntermediateFile); + } + /** * Other than gathering statistics for the ANALYZE command, the table scan operator * does not do anything special other than just forwarding the row. Since the table diff --git ql/src/java/org/apache/hadoop/hive/ql/exec/Task.java ql/src/java/org/apache/hadoop/hive/ql/exec/Task.java index d91047d..1c4be0a 100644 --- ql/src/java/org/apache/hadoop/hive/ql/exec/Task.java +++ ql/src/java/org/apache/hadoop/hive/ql/exec/Task.java @@ -246,6 +246,7 @@ public abstract class Task implements Serializable, Node } if (!getChildTasks().contains(dependent)) { ret = true; + renumberTasks(dependent, getId()); getChildTasks().add(dependent); if (dependent.getParentTasks() == null) { dependent.setParentTasks(new ArrayList>()); @@ -258,6 +259,42 @@ public abstract class Task implements Serializable, Node } /** + * Renumber the labels on the tasks + */ + void renumberTasks(Task task, String idName) + { + if (task.getId() == null) { // not all tasks have id's + return; + } + if (idName.startsWith("Stage-")) { + int base_id = Integer.parseInt(idName.substring(6)); // chop off the Stage- + int id = Integer.parseInt(task.getId().substring(6)); + int increment = base_id - id + 1; + if (increment > 0) { + LOG.info("incrementing tasks by " + increment); + renumberChildren(task, increment); + } + } + } + + /** + * recursively renumber child task labels + * @param task root task + * @param increment how much to increment the label + */ + void renumberChildren(Task task, int increment) { + String old_id = task.getId(); + int id = Integer.parseInt(task.getId().substring(6)); + task.setId("Stage-" + (id + increment)); + LOG.info("old task id: " + old_id + " new task id: " + task.getId()); + if (task.getChildTasks() != null) { + for (Task child : task.getChildTasks()) { + renumberChildren(child, increment); + } + } + } + + /** * Remove the dependent task. * * @param dependent diff --git ql/src/java/org/apache/hadoop/hive/ql/index/AbstractIndexHandler.java ql/src/java/org/apache/hadoop/hive/ql/index/AbstractIndexHandler.java index dd0186d..8e12898 100644 --- ql/src/java/org/apache/hadoop/hive/ql/index/AbstractIndexHandler.java +++ ql/src/java/org/apache/hadoop/hive/ql/index/AbstractIndexHandler.java @@ -21,7 +21,11 @@ package org.apache.hadoop.hive.ql.index; import java.util.List; import org.apache.hadoop.hive.metastore.api.FieldSchema; +import org.apache.hadoop.hive.metastore.api.Index; +import org.apache.hadoop.hive.ql.exec.Task; import org.apache.hadoop.hive.ql.metadata.HiveUtils; +import org.apache.hadoop.hive.ql.parse.ParseContext; +import org.apache.hadoop.hive.ql.plan.ExprNodeDesc; /** * Abstract base class for index handlers. This is provided as insulation @@ -42,4 +46,9 @@ public abstract class AbstractIndexHandler implements HiveIndexHandler { return sb.toString(); } + public List> generateIndexQuery(Index index, ExprNodeDesc predicate, + ParseContext pctx) { + return null; + } + } diff --git ql/src/java/org/apache/hadoop/hive/ql/index/HiveIndexHandler.java ql/src/java/org/apache/hadoop/hive/ql/index/HiveIndexHandler.java index 411b78f..1030210 100644 --- ql/src/java/org/apache/hadoop/hive/ql/index/HiveIndexHandler.java +++ ql/src/java/org/apache/hadoop/hive/ql/index/HiveIndexHandler.java @@ -22,12 +22,14 @@ import java.util.List; import java.util.Set; import org.apache.hadoop.conf.Configurable; +import org.apache.hadoop.hive.metastore.api.Index; import org.apache.hadoop.hive.ql.exec.Task; import org.apache.hadoop.hive.ql.hooks.ReadEntity; import org.apache.hadoop.hive.ql.hooks.WriteEntity; -import org.apache.hadoop.hive.ql.metadata.Hive; import org.apache.hadoop.hive.ql.metadata.HiveException; import org.apache.hadoop.hive.ql.metadata.Partition; +import org.apache.hadoop.hive.ql.parse.ParseContext; +import org.apache.hadoop.hive.ql.plan.ExprNodeDesc; /** * HiveIndexHandler defines a pluggable interface for adding new index handlers @@ -114,4 +116,12 @@ public interface HiveIndexHandler extends Configurable { Set inputs, Set outputs) throws HiveException; + /** + * + * @param index + * @param decomposedPredicate + * @return parseContext to run index table query + */ + List> generateIndexQuery(Index index, ExprNodeDesc predicate, + ParseContext pctx); } \ No newline at end of file diff --git ql/src/java/org/apache/hadoop/hive/ql/index/compact/CompactIndexHandler.java ql/src/java/org/apache/hadoop/hive/ql/index/compact/CompactIndexHandler.java index 1f01446..f784fb0 100644 --- ql/src/java/org/apache/hadoop/hive/ql/index/compact/CompactIndexHandler.java +++ ql/src/java/org/apache/hadoop/hive/ql/index/compact/CompactIndexHandler.java @@ -19,38 +19,50 @@ package org.apache.hadoop.hive.ql.index.compact; import java.util.ArrayList; -import java.util.HashSet; import java.util.Iterator; import java.util.LinkedHashMap; import java.util.List; import java.util.Set; import java.util.Map.Entry; +import org.apache.commons.logging.Log; +import org.apache.commons.logging.LogFactory; import org.apache.hadoop.conf.Configuration; import org.apache.hadoop.hive.conf.HiveConf; -import org.apache.hadoop.hive.metastore.MetaStoreUtils; import org.apache.hadoop.hive.metastore.api.FieldSchema; import org.apache.hadoop.hive.metastore.api.Index; import org.apache.hadoop.hive.metastore.api.StorageDescriptor; import org.apache.hadoop.hive.metastore.api.Table; import org.apache.hadoop.hive.ql.Driver; +import org.apache.hadoop.hive.ql.exec.TableScanOperator; import org.apache.hadoop.hive.ql.exec.Task; import org.apache.hadoop.hive.ql.exec.Utilities; import org.apache.hadoop.hive.ql.hooks.ReadEntity; import org.apache.hadoop.hive.ql.hooks.WriteEntity; import org.apache.hadoop.hive.ql.index.AbstractIndexHandler; -import org.apache.hadoop.hive.ql.metadata.Hive; +import org.apache.hadoop.hive.ql.index.IndexPredicateAnalyzer; +import org.apache.hadoop.hive.ql.index.IndexSearchCondition; import org.apache.hadoop.hive.ql.metadata.HiveException; import org.apache.hadoop.hive.ql.metadata.HiveUtils; import org.apache.hadoop.hive.ql.metadata.Partition; import org.apache.hadoop.hive.ql.metadata.VirtualColumn; +import org.apache.hadoop.hive.ql.metadata.HiveStoragePredicateHandler.DecomposedPredicate; +import org.apache.hadoop.hive.ql.parse.ParseContext; import org.apache.hadoop.hive.ql.parse.SemanticException; +import org.apache.hadoop.hive.ql.plan.ExprNodeDesc; import org.apache.hadoop.hive.ql.plan.PartitionDesc; import org.apache.hadoop.hive.ql.plan.TableDesc; +import org.apache.hadoop.hive.ql.udf.generic.GenericUDFOPEqual; +import org.apache.hadoop.hive.ql.udf.generic.GenericUDFOPEqualOrGreaterThan; +import org.apache.hadoop.hive.ql.udf.generic.GenericUDFOPEqualOrLessThan; +import org.apache.hadoop.hive.ql.udf.generic.GenericUDFOPGreaterThan; +import org.apache.hadoop.hive.ql.udf.generic.GenericUDFOPLessThan; public class CompactIndexHandler extends AbstractIndexHandler { private Configuration configuration; + private static final Log LOG = LogFactory.getLog(CompactIndexHandler.class.getName()); + @Override public void analyzeIndexDefinition(Table baseTable, Index index, @@ -201,6 +216,83 @@ public class CompactIndexHandler extends AbstractIndexHandler { } @Override + public List> generateIndexQuery(Index index, ExprNodeDesc predicate, + ParseContext pctx) { + + DecomposedPredicate decomposedPredicate = decomposePredicate(predicate, index); + + // Build reentrant QL for index query + StringBuilder qlCommand = new StringBuilder("INSERT OVERWRITE DIRECTORY "); + + String tmpFile = pctx.getContext().getMRTmpFileURI(); + qlCommand.append( "\"" + tmpFile + "\" "); // QL includes " around file name + qlCommand.append("SELECT `_bucketname` , `_offsets` FROM "); + qlCommand.append(index.getIndexTableName()); + qlCommand.append(" WHERE "); + + String predicateString = decomposedPredicate.pushedPredicate.getExprString(); + qlCommand.append(predicateString); + + // generate tasks from index query string + LOG.info("Generating tasks for re-entrant QL query: " + qlCommand.toString()); + Driver driver = new Driver(pctx.getConf()); + driver.compile(qlCommand.toString()); + + // setup TableScanOperator to change input format for original query + TableScanOperator originalTblScan = (TableScanOperator) pctx.getTopOps().get(index.getOrigTableName()); + originalTblScan.setIndexInputFormat(HiveCompactIndexInputFormat.class.getName()); + originalTblScan.setIndexIntermediateFile(tmpFile); + + Set inputs = pctx.getSemanticInputs(); + inputs.addAll(driver.getPlan().getInputs()); + return driver.getPlan().getRootTasks(); + } + + /** + * Split the predicate into the piece we can deal with (pushed), and the one we can't (residual) + * @param predicate + * @param index + * @return + */ + private DecomposedPredicate decomposePredicate(ExprNodeDesc predicate, Index index) { + IndexPredicateAnalyzer analyzer = getIndexPredicateAnalyzer(index); + List searchConditions = new ArrayList(); + // split predicate into pushed (what we can handle), and residual (what we can't handle) + ExprNodeDesc residualPredicate = analyzer.analyzePredicate(predicate, searchConditions); + + DecomposedPredicate decomposedPredicate = new DecomposedPredicate(); + decomposedPredicate.pushedPredicate = analyzer.translateSearchConditions(searchConditions); + decomposedPredicate.residualPredicate = residualPredicate; + + return decomposedPredicate; + } + + /** + * Instantiate a new predicate analyzer suitable for determining + * whether we can use an index, based on rules for indexes in + * WHERE clauses that we support + * + * @return preconfigured predicate analyzer for WHERE queries + */ + private IndexPredicateAnalyzer getIndexPredicateAnalyzer(Index index) { + IndexPredicateAnalyzer analyzer = new IndexPredicateAnalyzer(); + + analyzer.addComparisonOp(GenericUDFOPEqual.class.getName()); + analyzer.addComparisonOp(GenericUDFOPLessThan.class.getName()); + analyzer.addComparisonOp(GenericUDFOPEqualOrLessThan.class.getName()); + analyzer.addComparisonOp(GenericUDFOPGreaterThan.class.getName()); + analyzer.addComparisonOp(GenericUDFOPEqualOrGreaterThan.class.getName()); + + // only return results for columns in this index + List columnSchemas = index.getSd().getCols(); + for (FieldSchema column : columnSchemas) { + analyzer.allowColumnName(column.getName()); + } + + return analyzer; + } + + @Override public boolean usesIndexTable() { return true; } diff --git ql/src/java/org/apache/hadoop/hive/ql/metadata/Table.java ql/src/java/org/apache/hadoop/hive/ql/metadata/Table.java index 50db44c..f06a8de 100644 --- ql/src/java/org/apache/hadoop/hive/ql/metadata/Table.java +++ ql/src/java/org/apache/hadoop/hive/ql/metadata/Table.java @@ -38,6 +38,7 @@ import org.apache.hadoop.hive.metastore.MetaStoreUtils; import org.apache.hadoop.hive.metastore.ProtectMode; import org.apache.hadoop.hive.metastore.TableType; import org.apache.hadoop.hive.metastore.api.FieldSchema; +import org.apache.hadoop.hive.metastore.api.Index; import org.apache.hadoop.hive.metastore.api.MetaException; import org.apache.hadoop.hive.metastore.api.Order; import org.apache.hadoop.hive.metastore.api.SerDeInfo; @@ -813,4 +814,14 @@ public class Table implements Serializable { public String getCompleteName() { return getDbName() + "@" + getTableName(); } + + /** + * @return List containing Index Table names if there is exists indexes + * on this table + * @throws HiveException + **/ + public List getAllIndexes(short max) throws HiveException { + Hive hive = Hive.get(); + return hive.getIndexes(getTTable().getDbName(), getTTable().getTableName(), max); + } }; diff --git ql/src/java/org/apache/hadoop/hive/ql/optimizer/GenMRTableScan1.java ql/src/java/org/apache/hadoop/hive/ql/optimizer/GenMRTableScan1.java index 6162676..a9504dd 100644 --- ql/src/java/org/apache/hadoop/hive/ql/optimizer/GenMRTableScan1.java +++ ql/src/java/org/apache/hadoop/hive/ql/optimizer/GenMRTableScan1.java @@ -33,11 +33,11 @@ import org.apache.hadoop.hive.ql.lib.NodeProcessor; import org.apache.hadoop.hive.ql.lib.NodeProcessorCtx; import org.apache.hadoop.hive.ql.metadata.Partition; import org.apache.hadoop.hive.ql.optimizer.GenMRProcContext.GenMapRedCtx; -import org.apache.hadoop.hive.ql.parse.BaseSemanticAnalyzer.tableSpec; import org.apache.hadoop.hive.ql.parse.ParseContext; import org.apache.hadoop.hive.ql.parse.PrunedPartitionList; import org.apache.hadoop.hive.ql.parse.QBParseInfo; import org.apache.hadoop.hive.ql.parse.SemanticException; +import org.apache.hadoop.hive.ql.parse.BaseSemanticAnalyzer.tableSpec; import org.apache.hadoop.hive.ql.plan.MapredWork; import org.apache.hadoop.hive.ql.plan.StatsWork; /** @@ -69,6 +69,15 @@ public class GenMRTableScan1 implements NodeProcessor { ctx.setCurrTask(currTask); ctx.setCurrTopOp(currTopOp); + // Reset the inputFormat and inputFormatFile if the table scan needs a different one. + if (op.usesIndex()) { + String indexInputFormat = op.getIndexInputFormat(); + String indexIntermediateFile = op.getIndexIntermediateFile(); + ((MapredWork)currTask.getWork()).setInputformat(indexInputFormat); + ((MapredWork)currTask.getWork()).setIndexIntermediateFile(indexIntermediateFile); + } + + for (String alias : parseCtx.getTopOps().keySet()) { Operator currOp = parseCtx.getTopOps().get(alias); if (currOp == op) { diff --git ql/src/java/org/apache/hadoop/hive/ql/optimizer/physical/IndexWhereResolver.java ql/src/java/org/apache/hadoop/hive/ql/optimizer/physical/IndexWhereResolver.java new file mode 100644 index 0000000..502cdbd --- /dev/null +++ ql/src/java/org/apache/hadoop/hive/ql/optimizer/physical/IndexWhereResolver.java @@ -0,0 +1,25 @@ +package org.apache.hadoop.hive.ql.optimizer.physical; + +import java.util.ArrayList; + +import org.apache.hadoop.hive.ql.lib.DefaultGraphWalker; +import org.apache.hadoop.hive.ql.lib.Dispatcher; +import org.apache.hadoop.hive.ql.lib.GraphWalker; +import org.apache.hadoop.hive.ql.lib.Node; +import org.apache.hadoop.hive.ql.optimizer.physical.index.IndexWhereTaskDispatcher; +import org.apache.hadoop.hive.ql.parse.SemanticException; + +public class IndexWhereResolver implements PhysicalPlanResolver { + + @Override + public PhysicalContext resolve(PhysicalContext physicalContext) throws SemanticException { + //Dispatcher dispatcher = getDispatcher(physicalContext); + Dispatcher dispatcher = new IndexWhereTaskDispatcher(physicalContext); + GraphWalker opGraphWalker = new DefaultGraphWalker(dispatcher); + ArrayList topNodes = new ArrayList(); + topNodes.addAll(physicalContext.rootTasks); + opGraphWalker.startWalking(topNodes, null); + + return physicalContext; + } +} diff --git ql/src/java/org/apache/hadoop/hive/ql/optimizer/physical/PhysicalOptimizer.java ql/src/java/org/apache/hadoop/hive/ql/optimizer/physical/PhysicalOptimizer.java index 0ae9fa2..8b3db28 100644 --- ql/src/java/org/apache/hadoop/hive/ql/optimizer/physical/PhysicalOptimizer.java +++ ql/src/java/org/apache/hadoop/hive/ql/optimizer/physical/PhysicalOptimizer.java @@ -52,6 +52,9 @@ public class PhysicalOptimizer { if (hiveConf.getBoolVar(HiveConf.ConfVars.HIVECONVERTJOIN)) { resolvers.add(new CommonJoinResolver()); } + if (HiveConf.getBoolVar(hiveConf, HiveConf.ConfVars.HIVEOPTAUTOINDEX)) { + resolvers.add(new IndexWhereResolver()); + } resolvers.add(new MapJoinResolver()); } diff --git ql/src/java/org/apache/hadoop/hive/ql/optimizer/physical/index/IndexWhereProcCtx.java ql/src/java/org/apache/hadoop/hive/ql/optimizer/physical/index/IndexWhereProcCtx.java new file mode 100644 index 0000000..50852a5 --- /dev/null +++ ql/src/java/org/apache/hadoop/hive/ql/optimizer/physical/index/IndexWhereProcCtx.java @@ -0,0 +1,30 @@ +package org.apache.hadoop.hive.ql.optimizer.physical.index; + +import java.io.Serializable; + +import org.apache.commons.logging.Log; +import org.apache.commons.logging.LogFactory; +import org.apache.hadoop.hive.ql.exec.Task; +import org.apache.hadoop.hive.ql.lib.NodeProcessorCtx; +import org.apache.hadoop.hive.ql.parse.ParseContext; + +public class IndexWhereProcCtx implements NodeProcessorCtx { + + private static final Log LOG = LogFactory.getLog(IndexWhereProcCtx.class.getName()); + + private final Task currentTask; + private final ParseContext parseCtx; + + public IndexWhereProcCtx(Task task, ParseContext parseCtx) { + this.currentTask = task; + this.parseCtx = parseCtx; + } + + public ParseContext getParseContext() { + return parseCtx; + } + + public Task getCurrentTask() { + return currentTask; + } +} \ No newline at end of file diff --git ql/src/java/org/apache/hadoop/hive/ql/optimizer/physical/index/IndexWhereProcessor.java ql/src/java/org/apache/hadoop/hive/ql/optimizer/physical/index/IndexWhereProcessor.java new file mode 100644 index 0000000..c3787d0 --- /dev/null +++ ql/src/java/org/apache/hadoop/hive/ql/optimizer/physical/index/IndexWhereProcessor.java @@ -0,0 +1,110 @@ +package org.apache.hadoop.hive.ql.optimizer.physical.index; + +import java.util.ArrayList; +import java.util.List; +import java.util.Map; +import java.util.Stack; + +import org.apache.commons.logging.Log; +import org.apache.commons.logging.LogFactory; +import org.apache.hadoop.hive.metastore.api.Index; +import org.apache.hadoop.hive.ql.exec.FilterOperator; +import org.apache.hadoop.hive.ql.exec.SelectOperator; +import org.apache.hadoop.hive.ql.exec.Task; +import org.apache.hadoop.hive.ql.index.HiveIndexHandler; +import org.apache.hadoop.hive.ql.lib.Node; +import org.apache.hadoop.hive.ql.lib.NodeProcessor; +import org.apache.hadoop.hive.ql.lib.NodeProcessorCtx; +import org.apache.hadoop.hive.ql.metadata.HiveException; +import org.apache.hadoop.hive.ql.metadata.HiveUtils; +import org.apache.hadoop.hive.ql.metadata.Table; +import org.apache.hadoop.hive.ql.parse.ParseContext; +import org.apache.hadoop.hive.ql.parse.SemanticException; +import org.apache.hadoop.hive.ql.plan.ExprNodeDesc; +import org.apache.hadoop.hive.ql.plan.FilterDesc; + +/** +* +* IndexWhereProcessor. +* Processes Operator Nodes to look for WHERE queries with a predicate column +* on which we have an index. Creates an index subquery Task for these +* WHERE queries to use the index automatically. +*/ +public class IndexWhereProcessor implements NodeProcessor { + + private static final Log LOG = LogFactory.getLog(IndexWhereProcessor.class.getName()); + private final Map> indexes; + + public IndexWhereProcessor(Map> indexes) { + super(); + this.indexes = indexes; + } + + @Override + /** + * Process a node of the operator tree. This matches on the rule in IndexWhereTaskDispatcher + * We want to process the FilterOperator, but the regex matches it's child (the Select), so + * we must look at the parent operator. + */ + public Object process(Node nd, Stack stack, NodeProcessorCtx procCtx, + Object... nodeOutputs) throws SemanticException { + // Filter operator is the first parent of the Select operator + FilterOperator operator = (FilterOperator) ((SelectOperator) nd).getParentOperators().get(0); + FilterDesc operatorDesc = operator.getConf(); + ExprNodeDesc predicate = operatorDesc.getPredicate(); + + IndexWhereProcCtx context = (IndexWhereProcCtx) procCtx; + ParseContext pctx = context.getParseContext(); + + // get reentrant index queries from each index and add all to original query + List>> indexQueryTaskList = new ArrayList>>(); + for (List indexesOnTable : indexes.values()) { + for (Index index : indexesOnTable) { + List> indexQueryTasks = rewriteForIndex(predicate, index, pctx); + if (indexQueryTasks != null) + { + indexQueryTaskList.add(indexQueryTasks); + } + } + } + + // choose an index rewrite to use + if (indexQueryTaskList.size() > 0) { + // TODO This would be a good place for some sort of cost based choice? + List> chosenRewrite = indexQueryTaskList.get(0); + // add dependencies so index query runs first + Task currentTask = context.getCurrentTask(); + for (Task task : chosenRewrite) { + task.addDependentTask(currentTask); // add full scan task as child for every index query task + } + // replace the original with the index sub-query as a root task + pctx.replaceRootTask(currentTask, chosenRewrite); + } + + return null; + } + + /** + * Get a list of Tasks to activate use of indexes. + * Generate the tasks for the index query (where we store results of + * querying the index in a tmp file) inside the IndexHandler + */ + private List> rewriteForIndex(ExprNodeDesc predicate, Index index, + ParseContext pctx) + throws SemanticException { + HiveIndexHandler indexHandler; + try { + indexHandler = HiveUtils.getIndexHandler(pctx.getConf(), index.getIndexHandlerClass()); + } catch (HiveException e) { + LOG.error("Exception while loading IndexHandler: " + index.getIndexHandlerClass()); + throw new SemanticException("Failed to load indexHandler: " + index.getIndexHandlerClass(), e); + } + + // use the IndexHandler to generate the index query + List> indexQueryTasks = indexHandler.generateIndexQuery(index, predicate, pctx); + + return indexQueryTasks; + } + +} + diff --git ql/src/java/org/apache/hadoop/hive/ql/optimizer/physical/index/IndexWhereTaskDispatcher.java ql/src/java/org/apache/hadoop/hive/ql/optimizer/physical/index/IndexWhereTaskDispatcher.java new file mode 100644 index 0000000..1f7825f --- /dev/null +++ ql/src/java/org/apache/hadoop/hive/ql/optimizer/physical/index/IndexWhereTaskDispatcher.java @@ -0,0 +1,153 @@ +package org.apache.hadoop.hive.ql.optimizer.physical.index; + +import java.io.Serializable; +import java.util.ArrayList; +import java.util.Collection; +import java.util.HashMap; +import java.util.LinkedHashMap; +import java.util.List; +import java.util.Map; +import java.util.Stack; + +import org.apache.hadoop.hive.metastore.api.Index; +import org.apache.hadoop.hive.ql.exec.Task; +import org.apache.hadoop.hive.ql.index.compact.CompactIndexHandler; +import org.apache.hadoop.hive.ql.lib.DefaultGraphWalker; +import org.apache.hadoop.hive.ql.lib.DefaultRuleDispatcher; +import org.apache.hadoop.hive.ql.lib.Dispatcher; +import org.apache.hadoop.hive.ql.lib.GraphWalker; +import org.apache.hadoop.hive.ql.lib.Node; +import org.apache.hadoop.hive.ql.lib.NodeProcessor; +import org.apache.hadoop.hive.ql.lib.NodeProcessorCtx; +import org.apache.hadoop.hive.ql.lib.Rule; +import org.apache.hadoop.hive.ql.lib.RuleRegExp; +import org.apache.hadoop.hive.ql.metadata.HiveException; +import org.apache.hadoop.hive.ql.metadata.Table; +import org.apache.hadoop.hive.ql.optimizer.physical.PhysicalContext; +import org.apache.hadoop.hive.ql.parse.ParseContext; +import org.apache.hadoop.hive.ql.parse.SemanticException; + +/** + * + * IndexWhereTaskDispatcher. Walks a Task tree, and for the right kind of Task, + * walks the operator tree to create an index subquery. Then attaches the + * subquery task to the task tree. + * + */ +public class IndexWhereTaskDispatcher implements Dispatcher { + + private final PhysicalContext physicalContext; + + public IndexWhereTaskDispatcher(PhysicalContext context) { + super(); + physicalContext = context; + } + + @Override + public Object dispatch(Node nd, Stack stack, Object... nodeOutputs) + throws SemanticException { + + Task task = (Task) nd; + + ParseContext pctx = physicalContext.getParseContext(); + + // all of our index tasks are MapReduce + if (! task.isMapRedTask()) { + return null; + } + + // create the regex's so the walker can recognize our WHERE queries + Map operatorRules = createOperatorRules(pctx); + + // check for no indexes on table + if (operatorRules == null) { + return null; + } + + // create context so the walker can carry the current task with it. + IndexWhereProcCtx indexWhereOptimizeCtx = new IndexWhereProcCtx(task, pctx); + + // create the dispatcher, which fires the processor according to the rule that + // best matches + Dispatcher dispatcher = new DefaultRuleDispatcher(getDefaultProcessor(), + operatorRules, + indexWhereOptimizeCtx); + + // walk the mapper operator(not task) tree + GraphWalker ogw = new DefaultGraphWalker(dispatcher); + ArrayList topNodes = new ArrayList(); + topNodes.addAll(pctx.getTopOps().values()); + ogw.startWalking(topNodes, null); + + return null; + } + + /** + * Create a set of rules that only matches WHERE predicates on columns we have + * an index on. + * @return + */ + private Map createOperatorRules(ParseContext pctx) { + Map operatorRules = new LinkedHashMap(); + + List supportedIndexes = new ArrayList(); + supportedIndexes.add(CompactIndexHandler.class.getName()); + + // query the metastore to know what columns we have indexed + Collection topTables = pctx.getTopToTable().values(); + Map> indexes = new HashMap>(); + for (Table tbl : topTables) + { + indexes.put(tbl, getIndexes(tbl, supportedIndexes)); + } + + // quit if our table doesn't have any indexes + if (indexes.size() == 0) { + return null; + } + + // FIL% is a filter operator, a WHERE shows up as a filter on a table select operator (SEL%) + operatorRules.put(new RuleRegExp("RULEWhere", "FIL%SEL%"), new IndexWhereProcessor(indexes)); + + return operatorRules; + } + + /** + * Get a list of indexes on a table that match given types. + * Copied from HIVE-1694 patch + */ + private List getIndexes(Table baseTableMetaData, List matchIndexTypes) { + List matchingIndexes = new ArrayList(); + List indexesOnTable = null; + + try { + indexesOnTable = baseTableMetaData.getAllIndexes((short) -1); // get all indexes + + } catch (HiveException e) { + return matchingIndexes; + /* Return empty list. Trouble doing rewrite shouldn't stop regular + * query execution, if there's serious problem with metadata or anything else, + * it's assumed to be checked & handled in core hive code itself. + */ + } + + for (Index index : indexesOnTable) { + String indexType = index.getIndexHandlerClass(); + if (matchIndexTypes.contains(indexType)) { + matchingIndexes.add(index); + } + } + return matchingIndexes; + } + + private NodeProcessor getDefaultProcessor() { + return new NodeProcessor() { + @Override + public Object process(Node nd, Stack stack, NodeProcessorCtx procCtx, + Object... nodeOutputs) throws SemanticException { + return null; + } + }; + } + +} diff --git ql/src/java/org/apache/hadoop/hive/ql/parse/ParseContext.java ql/src/java/org/apache/hadoop/hive/ql/parse/ParseContext.java index 937a7b3..1451363 100644 --- ql/src/java/org/apache/hadoop/hive/ql/parse/ParseContext.java +++ ql/src/java/org/apache/hadoop/hive/ql/parse/ParseContext.java @@ -20,6 +20,7 @@ package org.apache.hadoop.hive.ql.parse; import java.io.Serializable; import java.util.HashMap; +import java.util.HashSet; import java.util.LinkedHashMap; import java.util.List; import java.util.Map; @@ -33,7 +34,9 @@ import org.apache.hadoop.hive.ql.exec.JoinOperator; import org.apache.hadoop.hive.ql.exec.MapJoinOperator; import org.apache.hadoop.hive.ql.exec.Operator; import org.apache.hadoop.hive.ql.exec.TableScanOperator; +import org.apache.hadoop.hive.ql.exec.Task; import org.apache.hadoop.hive.ql.hooks.LineageInfo; +import org.apache.hadoop.hive.ql.hooks.ReadEntity; import org.apache.hadoop.hive.ql.metadata.Table; import org.apache.hadoop.hive.ql.optimizer.unionproc.UnionProcContext; import org.apache.hadoop.hive.ql.plan.ExprNodeDesc; @@ -91,6 +94,9 @@ public class ParseContext { // a map-reduce job private boolean hasNonPartCols; + private HashSet semanticInputs; + private List> rootTasks; + public ParseContext() { } @@ -126,6 +132,8 @@ public class ParseContext { * list of map join operators with no reducer * @param opToSamplePruner * operator to sample pruner map + * @param rootTasks + * @param inputs */ public ParseContext( HiveConf conf, @@ -143,7 +151,8 @@ public class ParseContext { UnionProcContext uCtx, List> listMapJoinOpsNoReducer, Map> groupOpToInputTables, Map prunedPartitions, - HashMap opToSamplePruner) { + HashMap opToSamplePruner, + HashSet semanticInputs, List> rootTasks) { this.conf = conf; this.qb = qb; this.ast = ast; @@ -166,6 +175,8 @@ public class ParseContext { this.groupOpToInputTables = groupOpToInputTables; this.prunedPartitions = prunedPartitions; this.opToSamplePruner = opToSamplePruner; + this.semanticInputs = semanticInputs; + this.rootTasks = rootTasks; } /** @@ -486,4 +497,14 @@ public class ParseContext { public void setMapJoinContext(Map mapJoinContext) { this.mapJoinContext = mapJoinContext; } + + public HashSet getSemanticInputs() { + return semanticInputs; + } + + public void replaceRootTask(Task rootTask, + List> tasks) { + this.rootTasks.remove(rootTask); + this.rootTasks.addAll(tasks); + } } diff --git ql/src/java/org/apache/hadoop/hive/ql/parse/SemanticAnalyzer.java ql/src/java/org/apache/hadoop/hive/ql/parse/SemanticAnalyzer.java index f0aca84..8cb1dd0 100644 --- ql/src/java/org/apache/hadoop/hive/ql/parse/SemanticAnalyzer.java +++ ql/src/java/org/apache/hadoop/hive/ql/parse/SemanticAnalyzer.java @@ -258,11 +258,12 @@ public class SemanticAnalyzer extends BaseSemanticAnalyzer { } public ParseContext getParseContext() { - return new ParseContext(conf, qb, ast, opToPartPruner, opToPartList, topOps, + ParseContext pctx = new ParseContext(conf, qb, ast, opToPartPruner, opToPartList, topOps, topSelOps, opParseCtx, joinContext, topToTable, loadTableWork, loadFileWork, ctx, idToTableNameMap, destTableId, uCtx, listMapJoinOpsNoReducer, groupOpToInputTables, prunedPartitions, - opToSamplePruner); + opToSamplePruner, inputs, rootTasks); + return pctx; } @SuppressWarnings("nls") @@ -6623,7 +6624,7 @@ public class SemanticAnalyzer extends BaseSemanticAnalyzer { opToPartList, topOps, topSelOps, opParseCtx, joinContext, topToTable, loadTableWork, loadFileWork, ctx, idToTableNameMap, destTableId, uCtx, listMapJoinOpsNoReducer, groupOpToInputTables, prunedPartitions, - opToSamplePruner); + opToSamplePruner,inputs, rootTasks); Optimizer optm = new Optimizer(); optm.setPctx(pCtx); diff --git ql/src/java/org/apache/hadoop/hive/ql/plan/MapredWork.java ql/src/java/org/apache/hadoop/hive/ql/plan/MapredWork.java index 73391e9..6e661e4 100644 --- ql/src/java/org/apache/hadoop/hive/ql/plan/MapredWork.java +++ ql/src/java/org/apache/hadoop/hive/ql/plan/MapredWork.java @@ -70,6 +70,7 @@ public class MapredWork implements Serializable { private MapredLocalWork mapLocalWork; private String inputformat; + private String indexIntermediateFile; private boolean gatheringStats; private String tmpHDFSFileURI; @@ -363,10 +364,18 @@ public class MapredWork implements Serializable { return inputformat; } + public String getIndexIntermediateFile() { + return indexIntermediateFile; + } + public void setInputformat(String inputformat) { this.inputformat = inputformat; } + public void setIndexIntermediateFile(String fileName) { + this.indexIntermediateFile = fileName; + } + public void setGatheringStats(boolean gatherStats) { this.gatheringStats = gatherStats; } diff --git ql/src/test/queries/clientpositive/index_opt_where.q ql/src/test/queries/clientpositive/index_opt_where.q new file mode 100644 index 0000000..0b92520 --- /dev/null +++ ql/src/test/queries/clientpositive/index_opt_where.q @@ -0,0 +1,22 @@ +-- try the query without indexing, with manual indexing, and with automatic indexing +SELECT * FROM src WHERE key > 80 AND key < 100 ORDER BY key; + +CREATE INDEX src_index ON TABLE src(key) as 'COMPACT' WITH DEFERRED REBUILD; +ALTER INDEX src_index ON src REBUILD; + +SET hive.input.format=org.apache.hadoop.hive.ql.io.HiveInputFormat; +INSERT OVERWRITE DIRECTORY "/tmp/index_where" SELECT `_bucketname` , `_offsets` FROM default__src_src_index__ WHERE key > 80 AND key < 100; +SET hive.index.compact.file=/tmp/index_where; +SET hive.optimize.autoindex=false; +SET hive.input.format=org.apache.hadoop.hive.ql.index.compact.HiveCompactIndexInputFormat; + +EXPLAIN SELECT key, value FROM src WHERE key > 80 AND key < 100 ORDER BY key; +SELECT key, value FROM src WHERE key > 80 AND key < 100 ORDER BY key; + +SET hive.input.format=org.apache.hadoop.hive.ql.io.HiveInputFormat; +SET hive.optimize.autoindex=true; + +-- test automatic usage of index in query +SELECT * FROM src WHERE key > 80 AND key < 100 ORDER BY key; + +DROP INDEX src_index on src; \ No newline at end of file diff --git ql/src/test/queries/clientpositive/index_opt_where_simple.q ql/src/test/queries/clientpositive/index_opt_where_simple.q new file mode 100644 index 0000000..e04fe22 --- /dev/null +++ ql/src/test/queries/clientpositive/index_opt_where_simple.q @@ -0,0 +1,26 @@ +-- test automatic use of index +CREATE INDEX src_index ON TABLE src(key) as 'COMPACT' WITH DEFERRED REBUILD; +ALTER INDEX src_index ON src REBUILD; + +SET hive.input.format=org.apache.hadoop.hive.ql.io.HiveInputFormat; +SET hive.optimize.autoindex=false; +INSERT OVERWRITE DIRECTORY "/tmp/index_result_where1" SELECT `_bucketname` , `_offsets` FROM default__src_src_index__ WHERE key=86; +SET hive.index.compact.file=/tmp/index_result_where1; +SET hive.input.format=org.apache.hadoop.hive.ql.index.compact.HiveCompactIndexInputFormat; + +EXPLAIN SELECT key, value FROM src WHERE key=86 ORDER BY key; +SELECT key, value FROM src WHERE key=86 ORDER BY key; + +SET hive.input.format=org.apache.hadoop.hive.ql.io.HiveInputFormat; +SET hive.optimize.autoindex=true; + +EXPLAIN SELECT key, value FROM src WHERE key=86 ORDER BY key; +SELECT key, value FROM src WHERE key=86 ORDER BY key; + +SET hive.input.format=org.apache.hadoop.hive.ql.io.CombineHiveInputFormat; +SET hive.optimize.autoindex=true; + +EXPLAIN SELECT key, value FROM src WHERE key=86 ORDER BY key; +SELECT key, value FROM src WHERE key=86 ORDER BY key; + +DROP INDEX src_index on src; \ No newline at end of file diff --git ql/src/test/results/clientpositive/index_opt_where.q.out ql/src/test/results/clientpositive/index_opt_where.q.out new file mode 100644 index 0000000..4b274dc --- /dev/null +++ ql/src/test/results/clientpositive/index_opt_where.q.out @@ -0,0 +1,187 @@ +PREHOOK: query: -- try the query without indexing, with manual indexing, and with automatic indexing +SELECT * FROM src WHERE key > 80 AND key < 100 ORDER BY key +PREHOOK: type: QUERY +PREHOOK: Input: default@src +PREHOOK: Output: file:/tmp/rmelick/hive_2011-03-17_18-28-31_122_6222207918662299980/-mr-10000 +POSTHOOK: query: -- try the query without indexing, with manual indexing, and with automatic indexing +SELECT * FROM src WHERE key > 80 AND key < 100 ORDER BY key +POSTHOOK: type: QUERY +POSTHOOK: Input: default@src +POSTHOOK: Output: file:/tmp/rmelick/hive_2011-03-17_18-28-31_122_6222207918662299980/-mr-10000 +82 val_82 +83 val_83 +83 val_83 +84 val_84 +84 val_84 +85 val_85 +86 val_86 +87 val_87 +90 val_90 +90 val_90 +90 val_90 +92 val_92 +95 val_95 +95 val_95 +96 val_96 +97 val_97 +97 val_97 +98 val_98 +98 val_98 +PREHOOK: query: CREATE INDEX src_index ON TABLE src(key) as 'COMPACT' WITH DEFERRED REBUILD +PREHOOK: type: CREATEINDEX +POSTHOOK: query: CREATE INDEX src_index ON TABLE src(key) as 'COMPACT' WITH DEFERRED REBUILD +POSTHOOK: type: CREATEINDEX +PREHOOK: query: ALTER INDEX src_index ON src REBUILD +PREHOOK: type: QUERY +PREHOOK: Input: default@src +PREHOOK: Output: default@default__src_src_index__ +POSTHOOK: query: ALTER INDEX src_index ON src REBUILD +POSTHOOK: type: QUERY +POSTHOOK: Input: default@src +POSTHOOK: Output: default@default__src_src_index__ +POSTHOOK: Lineage: default__src_src_index__._bucketname SIMPLE [(src)src.FieldSchema(name:INPUT__FILE__NAME, type:string, comment:), ] +POSTHOOK: Lineage: default__src_src_index__._offsets EXPRESSION [(src)src.FieldSchema(name:BLOCK__OFFSET__INSIDE__FILE, type:bigint, comment:), ] +POSTHOOK: Lineage: default__src_src_index__.key SIMPLE [(src)src.FieldSchema(name:key, type:string, comment:default), ] +PREHOOK: query: INSERT OVERWRITE DIRECTORY "/tmp/index_where" SELECT `_bucketname` , `_offsets` FROM default__src_src_index__ WHERE key > 80 AND key < 100 +PREHOOK: type: QUERY +PREHOOK: Input: default@default__src_src_index__ +PREHOOK: Output: /tmp/index_where +POSTHOOK: query: INSERT OVERWRITE DIRECTORY "/tmp/index_where" SELECT `_bucketname` , `_offsets` FROM default__src_src_index__ WHERE key > 80 AND key < 100 +POSTHOOK: type: QUERY +POSTHOOK: Input: default@default__src_src_index__ +POSTHOOK: Output: /tmp/index_where +POSTHOOK: Lineage: default__src_src_index__._bucketname SIMPLE [(src)src.FieldSchema(name:INPUT__FILE__NAME, type:string, comment:), ] +POSTHOOK: Lineage: default__src_src_index__._offsets EXPRESSION [(src)src.FieldSchema(name:BLOCK__OFFSET__INSIDE__FILE, type:bigint, comment:), ] +POSTHOOK: Lineage: default__src_src_index__.key SIMPLE [(src)src.FieldSchema(name:key, type:string, comment:default), ] +PREHOOK: query: EXPLAIN SELECT key, value FROM src WHERE key > 80 AND key < 100 ORDER BY key +PREHOOK: type: QUERY +POSTHOOK: query: EXPLAIN SELECT key, value FROM src WHERE key > 80 AND key < 100 ORDER BY key +POSTHOOK: type: QUERY +POSTHOOK: Lineage: default__src_src_index__._bucketname SIMPLE [(src)src.FieldSchema(name:INPUT__FILE__NAME, type:string, comment:), ] +POSTHOOK: Lineage: default__src_src_index__._offsets EXPRESSION [(src)src.FieldSchema(name:BLOCK__OFFSET__INSIDE__FILE, type:bigint, comment:), ] +POSTHOOK: Lineage: default__src_src_index__.key SIMPLE [(src)src.FieldSchema(name:key, type:string, comment:default), ] +ABSTRACT SYNTAX TREE: + (TOK_QUERY (TOK_FROM (TOK_TABREF (TOK_TABNAME src))) (TOK_INSERT (TOK_DESTINATION (TOK_DIR TOK_TMP_FILE)) (TOK_SELECT (TOK_SELEXPR (TOK_TABLE_OR_COL key)) (TOK_SELEXPR (TOK_TABLE_OR_COL value))) (TOK_WHERE (AND (> (TOK_TABLE_OR_COL key) 80) (< (TOK_TABLE_OR_COL key) 100))) (TOK_ORDERBY (TOK_TABSORTCOLNAMEASC (TOK_TABLE_OR_COL key))))) + +STAGE DEPENDENCIES: + Stage-1 is a root stage + Stage-0 is a root stage + +STAGE PLANS: + Stage: Stage-1 + Map Reduce + Alias -> Map Operator Tree: + src + TableScan + alias: src + Filter Operator + predicate: + expr: ((key > 80) and (key < 100)) + type: boolean + Filter Operator + predicate: + expr: ((key > 80) and (key < 100)) + type: boolean + Select Operator + expressions: + expr: key + type: string + expr: value + type: string + outputColumnNames: _col0, _col1 + Reduce Output Operator + key expressions: + expr: _col0 + type: string + sort order: + + tag: -1 + value expressions: + expr: _col0 + type: string + expr: _col1 + type: string + Reduce Operator Tree: + Extract + File Output Operator + compressed: false + GlobalTableId: 0 + table: + input format: org.apache.hadoop.mapred.TextInputFormat + output format: org.apache.hadoop.hive.ql.io.HiveIgnoreKeyTextOutputFormat + + Stage: Stage-0 + Fetch Operator + limit: -1 + + +PREHOOK: query: SELECT key, value FROM src WHERE key > 80 AND key < 100 ORDER BY key +PREHOOK: type: QUERY +PREHOOK: Input: default@src +PREHOOK: Output: file:/tmp/rmelick/hive_2011-03-17_18-28-53_815_2544877942858526416/-mr-10000 +POSTHOOK: query: SELECT key, value FROM src WHERE key > 80 AND key < 100 ORDER BY key +POSTHOOK: type: QUERY +POSTHOOK: Input: default@src +POSTHOOK: Output: file:/tmp/rmelick/hive_2011-03-17_18-28-53_815_2544877942858526416/-mr-10000 +POSTHOOK: Lineage: default__src_src_index__._bucketname SIMPLE [(src)src.FieldSchema(name:INPUT__FILE__NAME, type:string, comment:), ] +POSTHOOK: Lineage: default__src_src_index__._offsets EXPRESSION [(src)src.FieldSchema(name:BLOCK__OFFSET__INSIDE__FILE, type:bigint, comment:), ] +POSTHOOK: Lineage: default__src_src_index__.key SIMPLE [(src)src.FieldSchema(name:key, type:string, comment:default), ] +82 val_82 +83 val_83 +83 val_83 +84 val_84 +84 val_84 +85 val_85 +86 val_86 +87 val_87 +90 val_90 +90 val_90 +90 val_90 +92 val_92 +95 val_95 +95 val_95 +96 val_96 +97 val_97 +97 val_97 +98 val_98 +98 val_98 +PREHOOK: query: -- test automatic usage of index in query +SELECT * FROM src WHERE key > 80 AND key < 100 ORDER BY key +PREHOOK: type: QUERY +PREHOOK: Input: default@default__src_src_index__ +PREHOOK: Input: default@src +PREHOOK: Output: file:/tmp/rmelick/hive_2011-03-17_18-28-57_322_4749087031650695280/-mr-10000 +POSTHOOK: query: -- test automatic usage of index in query +SELECT * FROM src WHERE key > 80 AND key < 100 ORDER BY key +POSTHOOK: type: QUERY +POSTHOOK: Input: default@default__src_src_index__ +POSTHOOK: Input: default@src +POSTHOOK: Output: file:/tmp/rmelick/hive_2011-03-17_18-28-57_322_4749087031650695280/-mr-10000 +POSTHOOK: Lineage: default__src_src_index__._bucketname SIMPLE [(src)src.FieldSchema(name:INPUT__FILE__NAME, type:string, comment:), ] +POSTHOOK: Lineage: default__src_src_index__._offsets EXPRESSION [(src)src.FieldSchema(name:BLOCK__OFFSET__INSIDE__FILE, type:bigint, comment:), ] +POSTHOOK: Lineage: default__src_src_index__.key SIMPLE [(src)src.FieldSchema(name:key, type:string, comment:default), ] +82 val_82 +83 val_83 +83 val_83 +84 val_84 +84 val_84 +85 val_85 +86 val_86 +87 val_87 +90 val_90 +90 val_90 +90 val_90 +92 val_92 +95 val_95 +95 val_95 +96 val_96 +97 val_97 +97 val_97 +98 val_98 +98 val_98 +PREHOOK: query: DROP INDEX src_index on src +PREHOOK: type: DROPINDEX +POSTHOOK: query: DROP INDEX src_index on src +POSTHOOK: type: DROPINDEX +POSTHOOK: Lineage: default__src_src_index__._bucketname SIMPLE [(src)src.FieldSchema(name:INPUT__FILE__NAME, type:string, comment:), ] +POSTHOOK: Lineage: default__src_src_index__._offsets EXPRESSION [(src)src.FieldSchema(name:BLOCK__OFFSET__INSIDE__FILE, type:bigint, comment:), ] +POSTHOOK: Lineage: default__src_src_index__.key SIMPLE [(src)src.FieldSchema(name:key, type:string, comment:default), ] diff --git ql/src/test/results/clientpositive/index_opt_where_simple.q.out ql/src/test/results/clientpositive/index_opt_where_simple.q.out new file mode 100644 index 0000000..8cbdd4c --- /dev/null +++ ql/src/test/results/clientpositive/index_opt_where_simple.q.out @@ -0,0 +1,376 @@ +PREHOOK: query: -- test automatic use of index +CREATE INDEX src_index ON TABLE src(key) as 'COMPACT' WITH DEFERRED REBUILD +PREHOOK: type: CREATEINDEX +POSTHOOK: query: -- test automatic use of index +CREATE INDEX src_index ON TABLE src(key) as 'COMPACT' WITH DEFERRED REBUILD +POSTHOOK: type: CREATEINDEX +PREHOOK: query: ALTER INDEX src_index ON src REBUILD +PREHOOK: type: QUERY +PREHOOK: Input: default@src +PREHOOK: Output: default@default__src_src_index__ +POSTHOOK: query: ALTER INDEX src_index ON src REBUILD +POSTHOOK: type: QUERY +POSTHOOK: Input: default@src +POSTHOOK: Output: default@default__src_src_index__ +POSTHOOK: Lineage: default__src_src_index__._bucketname SIMPLE [(src)src.FieldSchema(name:INPUT__FILE__NAME, type:string, comment:), ] +POSTHOOK: Lineage: default__src_src_index__._offsets EXPRESSION [(src)src.FieldSchema(name:BLOCK__OFFSET__INSIDE__FILE, type:bigint, comment:), ] +POSTHOOK: Lineage: default__src_src_index__.key SIMPLE [(src)src.FieldSchema(name:key, type:string, comment:default), ] +PREHOOK: query: INSERT OVERWRITE DIRECTORY "/tmp/index_result_where1" SELECT `_bucketname` , `_offsets` FROM default__src_src_index__ WHERE key=86 +PREHOOK: type: QUERY +PREHOOK: Input: default@default__src_src_index__ +PREHOOK: Output: /tmp/index_result_where1 +POSTHOOK: query: INSERT OVERWRITE DIRECTORY "/tmp/index_result_where1" SELECT `_bucketname` , `_offsets` FROM default__src_src_index__ WHERE key=86 +POSTHOOK: type: QUERY +POSTHOOK: Input: default@default__src_src_index__ +POSTHOOK: Output: /tmp/index_result_where1 +POSTHOOK: Lineage: default__src_src_index__._bucketname SIMPLE [(src)src.FieldSchema(name:INPUT__FILE__NAME, type:string, comment:), ] +POSTHOOK: Lineage: default__src_src_index__._offsets EXPRESSION [(src)src.FieldSchema(name:BLOCK__OFFSET__INSIDE__FILE, type:bigint, comment:), ] +POSTHOOK: Lineage: default__src_src_index__.key SIMPLE [(src)src.FieldSchema(name:key, type:string, comment:default), ] +PREHOOK: query: EXPLAIN SELECT key, value FROM src WHERE key=86 ORDER BY key +PREHOOK: type: QUERY +POSTHOOK: query: EXPLAIN SELECT key, value FROM src WHERE key=86 ORDER BY key +POSTHOOK: type: QUERY +POSTHOOK: Lineage: default__src_src_index__._bucketname SIMPLE [(src)src.FieldSchema(name:INPUT__FILE__NAME, type:string, comment:), ] +POSTHOOK: Lineage: default__src_src_index__._offsets EXPRESSION [(src)src.FieldSchema(name:BLOCK__OFFSET__INSIDE__FILE, type:bigint, comment:), ] +POSTHOOK: Lineage: default__src_src_index__.key SIMPLE [(src)src.FieldSchema(name:key, type:string, comment:default), ] +ABSTRACT SYNTAX TREE: + (TOK_QUERY (TOK_FROM (TOK_TABREF (TOK_TABNAME src))) (TOK_INSERT (TOK_DESTINATION (TOK_DIR TOK_TMP_FILE)) (TOK_SELECT (TOK_SELEXPR (TOK_TABLE_OR_COL key)) (TOK_SELEXPR (TOK_TABLE_OR_COL value))) (TOK_WHERE (= (TOK_TABLE_OR_COL key) 86)) (TOK_ORDERBY (TOK_TABSORTCOLNAMEASC (TOK_TABLE_OR_COL key))))) + +STAGE DEPENDENCIES: + Stage-1 is a root stage + Stage-0 is a root stage + +STAGE PLANS: + Stage: Stage-1 + Map Reduce + Alias -> Map Operator Tree: + src + TableScan + alias: src + Filter Operator + predicate: + expr: (key = 86) + type: boolean + Filter Operator + predicate: + expr: (key = 86) + type: boolean + Select Operator + expressions: + expr: key + type: string + expr: value + type: string + outputColumnNames: _col0, _col1 + Reduce Output Operator + key expressions: + expr: _col0 + type: string + sort order: + + tag: -1 + value expressions: + expr: _col0 + type: string + expr: _col1 + type: string + Reduce Operator Tree: + Extract + File Output Operator + compressed: false + GlobalTableId: 0 + table: + input format: org.apache.hadoop.mapred.TextInputFormat + output format: org.apache.hadoop.hive.ql.io.HiveIgnoreKeyTextOutputFormat + + Stage: Stage-0 + Fetch Operator + limit: -1 + + +PREHOOK: query: SELECT key, value FROM src WHERE key=86 ORDER BY key +PREHOOK: type: QUERY +PREHOOK: Input: default@src +PREHOOK: Output: file:/tmp/rmelick/hive_2011-03-22_22-44-41_443_7188950341223053879/-mr-10000 +POSTHOOK: query: SELECT key, value FROM src WHERE key=86 ORDER BY key +POSTHOOK: type: QUERY +POSTHOOK: Input: default@src +POSTHOOK: Output: file:/tmp/rmelick/hive_2011-03-22_22-44-41_443_7188950341223053879/-mr-10000 +POSTHOOK: Lineage: default__src_src_index__._bucketname SIMPLE [(src)src.FieldSchema(name:INPUT__FILE__NAME, type:string, comment:), ] +POSTHOOK: Lineage: default__src_src_index__._offsets EXPRESSION [(src)src.FieldSchema(name:BLOCK__OFFSET__INSIDE__FILE, type:bigint, comment:), ] +POSTHOOK: Lineage: default__src_src_index__.key SIMPLE [(src)src.FieldSchema(name:key, type:string, comment:default), ] +86 val_86 +PREHOOK: query: EXPLAIN SELECT key, value FROM src WHERE key=86 ORDER BY key +PREHOOK: type: QUERY +POSTHOOK: query: EXPLAIN SELECT key, value FROM src WHERE key=86 ORDER BY key +POSTHOOK: type: QUERY +POSTHOOK: Lineage: default__src_src_index__._bucketname SIMPLE [(src)src.FieldSchema(name:INPUT__FILE__NAME, type:string, comment:), ] +POSTHOOK: Lineage: default__src_src_index__._offsets EXPRESSION [(src)src.FieldSchema(name:BLOCK__OFFSET__INSIDE__FILE, type:bigint, comment:), ] +POSTHOOK: Lineage: default__src_src_index__.key SIMPLE [(src)src.FieldSchema(name:key, type:string, comment:default), ] +ABSTRACT SYNTAX TREE: + (TOK_QUERY (TOK_FROM (TOK_TABREF (TOK_TABNAME src))) (TOK_INSERT (TOK_DESTINATION (TOK_DIR TOK_TMP_FILE)) (TOK_SELECT (TOK_SELEXPR (TOK_TABLE_OR_COL key)) (TOK_SELEXPR (TOK_TABLE_OR_COL value))) (TOK_WHERE (= (TOK_TABLE_OR_COL key) 86)) (TOK_ORDERBY (TOK_TABSORTCOLNAMEASC (TOK_TABLE_OR_COL key))))) + +STAGE DEPENDENCIES: + Stage-1 is a root stage + Stage-4 depends on stages: Stage-1 , consists of Stage-3, Stage-2 + Stage-3 + Stage-4 depends on stages: Stage-3, Stage-2 + Stage-2 + Stage-2 depends on stages: Stage-1 + Stage-0 is a root stage + +STAGE PLANS: + Stage: Stage-1 + Map Reduce + Alias -> Map Operator Tree: + default__src_src_index__ + TableScan + alias: default__src_src_index__ + Filter Operator + predicate: + expr: (key = 86) + type: boolean + Filter Operator + predicate: + expr: (key = 86) + type: boolean + Select Operator + expressions: + expr: _bucketname + type: string + expr: _offsets + type: array + outputColumnNames: _col0, _col1 + File Output Operator + compressed: false + GlobalTableId: 1 + table: + input format: org.apache.hadoop.mapred.TextInputFormat + output format: org.apache.hadoop.hive.ql.io.HiveIgnoreKeyTextOutputFormat + + Stage: Stage-4 + Conditional Operator + + Stage: Stage-3 + Move Operator + files: + hdfs directory: true + destination: file:/home/rmelick/hive/build/ql/scratchdir/hive_2011-03-22_22-44-44_716_6816948650187221192/-ext-10000 + + Stage: Stage-4 + Move Operator + files: + hdfs directory: true + destination: file:/tmp/rmelick/hive_2011-03-22_22-44-44_664_5142464855693043492/-mr-10002 + + Stage: Stage-2 + Map Reduce + Alias -> Map Operator Tree: + file:/home/rmelick/hive/build/ql/scratchdir/hive_2011-03-22_22-44-44_716_6816948650187221192/-ext-10001 + File Output Operator + compressed: false + GlobalTableId: 0 + table: + input format: org.apache.hadoop.mapred.TextInputFormat + output format: org.apache.hadoop.hive.ql.io.HiveIgnoreKeyTextOutputFormat + + Stage: Stage-2 + Map Reduce + Alias -> Map Operator Tree: + src + TableScan + alias: src + Filter Operator + predicate: + expr: (key = 86) + type: boolean + Filter Operator + predicate: + expr: (key = 86) + type: boolean + Select Operator + expressions: + expr: key + type: string + expr: value + type: string + outputColumnNames: _col0, _col1 + Reduce Output Operator + key expressions: + expr: _col0 + type: string + sort order: + + tag: -1 + value expressions: + expr: _col0 + type: string + expr: _col1 + type: string + Reduce Operator Tree: + Extract + File Output Operator + compressed: false + GlobalTableId: 0 + table: + input format: org.apache.hadoop.mapred.TextInputFormat + output format: org.apache.hadoop.hive.ql.io.HiveIgnoreKeyTextOutputFormat + + Stage: Stage-0 + Fetch Operator + limit: -1 + + +PREHOOK: query: SELECT key, value FROM src WHERE key=86 ORDER BY key +PREHOOK: type: QUERY +PREHOOK: Input: default@default__src_src_index__ +PREHOOK: Input: default@src +PREHOOK: Output: file:/tmp/rmelick/hive_2011-03-22_22-44-44_793_3409503229115496623/-mr-10000 +POSTHOOK: query: SELECT key, value FROM src WHERE key=86 ORDER BY key +POSTHOOK: type: QUERY +POSTHOOK: Input: default@default__src_src_index__ +POSTHOOK: Input: default@src +POSTHOOK: Output: file:/tmp/rmelick/hive_2011-03-22_22-44-44_793_3409503229115496623/-mr-10000 +POSTHOOK: Lineage: default__src_src_index__._bucketname SIMPLE [(src)src.FieldSchema(name:INPUT__FILE__NAME, type:string, comment:), ] +POSTHOOK: Lineage: default__src_src_index__._offsets EXPRESSION [(src)src.FieldSchema(name:BLOCK__OFFSET__INSIDE__FILE, type:bigint, comment:), ] +POSTHOOK: Lineage: default__src_src_index__.key SIMPLE [(src)src.FieldSchema(name:key, type:string, comment:default), ] +86 val_86 +PREHOOK: query: EXPLAIN SELECT key, value FROM src WHERE key=86 ORDER BY key +PREHOOK: type: QUERY +POSTHOOK: query: EXPLAIN SELECT key, value FROM src WHERE key=86 ORDER BY key +POSTHOOK: type: QUERY +POSTHOOK: Lineage: default__src_src_index__._bucketname SIMPLE [(src)src.FieldSchema(name:INPUT__FILE__NAME, type:string, comment:), ] +POSTHOOK: Lineage: default__src_src_index__._offsets EXPRESSION [(src)src.FieldSchema(name:BLOCK__OFFSET__INSIDE__FILE, type:bigint, comment:), ] +POSTHOOK: Lineage: default__src_src_index__.key SIMPLE [(src)src.FieldSchema(name:key, type:string, comment:default), ] +ABSTRACT SYNTAX TREE: + (TOK_QUERY (TOK_FROM (TOK_TABREF (TOK_TABNAME src))) (TOK_INSERT (TOK_DESTINATION (TOK_DIR TOK_TMP_FILE)) (TOK_SELECT (TOK_SELEXPR (TOK_TABLE_OR_COL key)) (TOK_SELEXPR (TOK_TABLE_OR_COL value))) (TOK_WHERE (= (TOK_TABLE_OR_COL key) 86)) (TOK_ORDERBY (TOK_TABSORTCOLNAMEASC (TOK_TABLE_OR_COL key))))) + +STAGE DEPENDENCIES: + Stage-1 is a root stage + Stage-4 depends on stages: Stage-1 , consists of Stage-3, Stage-2 + Stage-3 + Stage-4 depends on stages: Stage-3, Stage-2 + Stage-2 + Stage-2 depends on stages: Stage-1 + Stage-0 is a root stage + +STAGE PLANS: + Stage: Stage-1 + Map Reduce + Alias -> Map Operator Tree: + default__src_src_index__ + TableScan + alias: default__src_src_index__ + Filter Operator + predicate: + expr: (key = 86) + type: boolean + Filter Operator + predicate: + expr: (key = 86) + type: boolean + Select Operator + expressions: + expr: _bucketname + type: string + expr: _offsets + type: array + outputColumnNames: _col0, _col1 + File Output Operator + compressed: false + GlobalTableId: 1 + table: + input format: org.apache.hadoop.mapred.TextInputFormat + output format: org.apache.hadoop.hive.ql.io.HiveIgnoreKeyTextOutputFormat + + Stage: Stage-4 + Conditional Operator + + Stage: Stage-3 + Move Operator + files: + hdfs directory: true + destination: file:/home/rmelick/hive/build/ql/scratchdir/hive_2011-03-22_22-44-50_943_6637016649403342257/-ext-10000 + + Stage: Stage-4 + Move Operator + files: + hdfs directory: true + destination: file:/tmp/rmelick/hive_2011-03-22_22-44-50_906_6637852083732386798/-mr-10002 + + Stage: Stage-2 + Map Reduce + Alias -> Map Operator Tree: + file:/home/rmelick/hive/build/ql/scratchdir/hive_2011-03-22_22-44-50_943_6637016649403342257/-ext-10001 + File Output Operator + compressed: false + GlobalTableId: 0 + table: + input format: org.apache.hadoop.mapred.TextInputFormat + output format: org.apache.hadoop.hive.ql.io.HiveIgnoreKeyTextOutputFormat + + Stage: Stage-2 + Map Reduce + Alias -> Map Operator Tree: + src + TableScan + alias: src + Filter Operator + predicate: + expr: (key = 86) + type: boolean + Filter Operator + predicate: + expr: (key = 86) + type: boolean + Select Operator + expressions: + expr: key + type: string + expr: value + type: string + outputColumnNames: _col0, _col1 + Reduce Output Operator + key expressions: + expr: _col0 + type: string + sort order: + + tag: -1 + value expressions: + expr: _col0 + type: string + expr: _col1 + type: string + Reduce Operator Tree: + Extract + File Output Operator + compressed: false + GlobalTableId: 0 + table: + input format: org.apache.hadoop.mapred.TextInputFormat + output format: org.apache.hadoop.hive.ql.io.HiveIgnoreKeyTextOutputFormat + + Stage: Stage-0 + Fetch Operator + limit: -1 + + +PREHOOK: query: SELECT key, value FROM src WHERE key=86 ORDER BY key +PREHOOK: type: QUERY +PREHOOK: Input: default@default__src_src_index__ +PREHOOK: Input: default@src +PREHOOK: Output: file:/tmp/rmelick/hive_2011-03-22_22-44-51_016_3038767398565713192/-mr-10000 +POSTHOOK: query: SELECT key, value FROM src WHERE key=86 ORDER BY key +POSTHOOK: type: QUERY +POSTHOOK: Input: default@default__src_src_index__ +POSTHOOK: Input: default@src +POSTHOOK: Output: file:/tmp/rmelick/hive_2011-03-22_22-44-51_016_3038767398565713192/-mr-10000 +POSTHOOK: Lineage: default__src_src_index__._bucketname SIMPLE [(src)src.FieldSchema(name:INPUT__FILE__NAME, type:string, comment:), ] +POSTHOOK: Lineage: default__src_src_index__._offsets EXPRESSION [(src)src.FieldSchema(name:BLOCK__OFFSET__INSIDE__FILE, type:bigint, comment:), ] +POSTHOOK: Lineage: default__src_src_index__.key SIMPLE [(src)src.FieldSchema(name:key, type:string, comment:default), ] +86 val_86 +PREHOOK: query: DROP INDEX src_index on src +PREHOOK: type: DROPINDEX +POSTHOOK: query: DROP INDEX src_index on src +POSTHOOK: type: DROPINDEX +POSTHOOK: Lineage: default__src_src_index__._bucketname SIMPLE [(src)src.FieldSchema(name:INPUT__FILE__NAME, type:string, comment:), ] +POSTHOOK: Lineage: default__src_src_index__._offsets EXPRESSION [(src)src.FieldSchema(name:BLOCK__OFFSET__INSIDE__FILE, type:bigint, comment:), ] +POSTHOOK: Lineage: default__src_src_index__.key SIMPLE [(src)src.FieldSchema(name:key, type:string, comment:default), ]