org.apache.hadoop.hive.ql.parse.QB Maven / Gradle / Ivy
The newest version!
/**
* 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.parse;
import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.Set;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.apache.hadoop.fs.Path;
import org.apache.hadoop.hive.ql.metadata.Table;
import org.apache.hadoop.hive.ql.plan.CreateTableDesc;
/**
* Implementation of the query block.
*
**/
public class QB {
private static final Log LOG = LogFactory.getLog("hive.ql.parse.QB");
private final int numJoins = 0;
private final int numGbys = 0;
private int numSels = 0;
private int numSelDi = 0;
private HashMap aliasToTabs;
private HashMap aliasToSubq;
private HashMap> aliasToProps;
private List aliases;
private QBParseInfo qbp;
private QBMetaData qbm;
private QBJoinTree qbjoin;
private String id;
private boolean isQuery;
private boolean isAnalyzeRewrite;
private CreateTableDesc tblDesc = null; // table descriptor of the final
private CreateTableDesc directoryDesc = null ;
private List encryptedTargetTablePaths;
// used by PTFs
/*
* This map maintains the PTFInvocationSpec for each PTF chain invocation in this QB.
*/
private HashMap ptfNodeToSpec;
/*
* the WindowingSpec used for windowing clauses in this QB.
*/
private HashMap destToWindowingSpec;
/*
* If this QB represents a SubQuery predicate then this will point to the SubQuery object.
*/
private QBSubQuery subQueryPredicateDef;
/*
* used to give a unique name to each SubQuery QB Currently there can be at
* most 2 SubQueries in a Query: 1 in the Where clause, and 1 in the Having
* clause.
*/
private int numSubQueryPredicates;
/*
* for now a top level QB can have 1 where clause SQ predicate.
*/
private QBSubQuery whereClauseSubQueryPredicate;
/*
* for now a top level QB can have 1 where clause SQ predicate.
*/
private QBSubQuery havingClauseSubQueryPredicate;
// results
public void print(String msg) {
LOG.info(msg + "alias=" + qbp.getAlias());
for (String alias : getSubqAliases()) {
QBExpr qbexpr = getSubqForAlias(alias);
LOG.info(msg + "start subquery " + alias);
qbexpr.print(msg + " ");
LOG.info(msg + "end subquery " + alias);
}
}
public QB() {
}
public QB(String outer_id, String alias, boolean isSubQ) {
// Must be deterministic order maps - see HIVE-8707
aliasToTabs = new LinkedHashMap();
aliasToSubq = new LinkedHashMap();
aliasToProps = new LinkedHashMap>();
aliases = new ArrayList();
if (alias != null) {
alias = alias.toLowerCase();
}
qbp = new QBParseInfo(alias, isSubQ);
qbm = new QBMetaData();
// Must be deterministic order maps - see HIVE-8707
ptfNodeToSpec = new LinkedHashMap();
destToWindowingSpec = new LinkedHashMap();
id = getAppendedAliasFromId(outer_id, alias);
}
// For sub-queries, the id. and alias should be appended since same aliases can be re-used
// within different sub-queries.
// For a query like:
// select ...
// (select * from T1 a where ...) subq1
// join
// (select * from T2 a where ...) subq2
// ..
// the alias is modified to subq1:a and subq2:a from a, to identify the right sub-query.
public static String getAppendedAliasFromId(String outer_id, String alias) {
return (outer_id == null ? alias : outer_id + ":" + alias);
}
public String getAlias() {
return qbp.getAlias();
}
public QBParseInfo getParseInfo() {
return qbp;
}
public QBMetaData getMetaData() {
return qbm;
}
public void setQBParseInfo(QBParseInfo qbp) {
this.qbp = qbp;
}
public void countSelDi() {
numSelDi++;
}
public void countSel() {
numSels++;
}
public boolean exists(String alias) {
alias = alias.toLowerCase();
if (aliasToTabs.get(alias) != null || aliasToSubq.get(alias) != null) {
return true;
}
return false;
}
public void setTabAlias(String alias, String tabName) {
aliasToTabs.put(alias.toLowerCase(), tabName);
}
public void setSubqAlias(String alias, QBExpr qbexpr) {
aliasToSubq.put(alias.toLowerCase(), qbexpr);
}
public void setTabProps(String alias, Map props) {
aliasToProps.put(alias.toLowerCase(), props);
}
public void addAlias(String alias) {
if (!aliases.contains(alias.toLowerCase())) {
aliases.add(alias.toLowerCase());
}
}
public String getId() {
return id;
}
public int getNumGbys() {
return numGbys;
}
public int getNumSelDi() {
return numSelDi;
}
public int getNumSels() {
return numSels;
}
public int getNumJoins() {
return numJoins;
}
public Set getSubqAliases() {
return aliasToSubq.keySet();
}
public Set getTabAliases() {
return aliasToTabs.keySet();
}
public List getAliases() {
return aliases;
}
public QBExpr getSubqForAlias(String alias) {
return aliasToSubq.get(alias.toLowerCase());
}
public String getTabNameForAlias(String alias) {
return aliasToTabs.get(alias.toLowerCase());
}
public Map getTabPropsForAlias(String alias) {
return aliasToProps.get(alias.toLowerCase());
}
public void rewriteViewToSubq(String alias, String viewName, QBExpr qbexpr) {
alias = alias.toLowerCase();
String tableName = aliasToTabs.remove(alias);
assert (viewName.equals(tableName));
aliasToSubq.put(alias, qbexpr);
}
public void rewriteCTEToSubq(String alias, String cteName, QBExpr qbexpr) {
rewriteViewToSubq(alias, cteName, qbexpr);
}
public QBJoinTree getQbJoinTree() {
return qbjoin;
}
public void setQbJoinTree(QBJoinTree qbjoin) {
this.qbjoin = qbjoin;
}
public void setIsQuery(boolean isQuery) {
this.isQuery = isQuery;
}
public boolean getIsQuery() {
return isQuery;
}
// to decide whether to rewrite RR of subquery
public boolean isTopLevelSelectStarQuery() {
return !isCTAS() && qbp.isTopLevelSimpleSelectStarQuery();
}
// to find target for fetch task conversion optimizer (not allows subqueries)
public boolean isSimpleSelectQuery() {
if (!qbp.isSimpleSelectQuery() || isCTAS() || qbp.isAnalyzeCommand()) {
return false;
}
for (QBExpr qbexpr : aliasToSubq.values()) {
if (!qbexpr.isSimpleSelectQuery()) {
return false;
}
}
return true;
}
public boolean hasTableSample(String alias) {
return qbp.getTabSample(alias) != null;
}
public CreateTableDesc getTableDesc() {
return tblDesc;
}
public void setTableDesc(CreateTableDesc desc) {
tblDesc = desc;
}
public CreateTableDesc getDirectoryDesc() {
return directoryDesc;
}
public void setDirectoryDesc(CreateTableDesc directoryDesc) {
this.directoryDesc = directoryDesc;
}
/**
* Whether this QB is for a CREATE-TABLE-AS-SELECT.
*/
public boolean isCTAS() {
return tblDesc != null;
}
/**
* Retrieve skewed column name for a table.
* @param alias table alias
* @return
*/
public List getSkewedColumnNames(String alias) {
List skewedColNames = null;
if (null != qbm &&
null != qbm.getAliasToTable() &&
qbm.getAliasToTable().size() > 0) {
Table tbl = getMetaData().getTableForAlias(alias);
skewedColNames = tbl.getSkewedColNames();
}
return skewedColNames;
}
public boolean isAnalyzeRewrite() {
return isAnalyzeRewrite;
}
public void setAnalyzeRewrite(boolean isAnalyzeRewrite) {
this.isAnalyzeRewrite = isAnalyzeRewrite;
}
public PTFInvocationSpec getPTFInvocationSpec(ASTNode node) {
return ptfNodeToSpec == null ? null : ptfNodeToSpec.get(node);
}
public void addPTFNodeToSpec(ASTNode node, PTFInvocationSpec spec) {
// Must be deterministic order map - see HIVE-8707
ptfNodeToSpec = ptfNodeToSpec == null ? new LinkedHashMap() : ptfNodeToSpec;
ptfNodeToSpec.put(node, spec);
}
public HashMap getPTFNodeToSpec() {
return ptfNodeToSpec;
}
public WindowingSpec getWindowingSpec(String dest) {
return destToWindowingSpec.get(dest);
}
public void addDestToWindowingSpec(String dest, WindowingSpec windowingSpec) {
destToWindowingSpec.put(dest, windowingSpec);
}
public boolean hasWindowingSpec(String dest) {
return destToWindowingSpec.get(dest) != null;
}
public HashMap getAllWindowingSpecs() {
return destToWindowingSpec;
}
protected void setSubQueryDef(QBSubQuery subQueryPredicateDef) {
this.subQueryPredicateDef = subQueryPredicateDef;
}
protected QBSubQuery getSubQueryPredicateDef() {
return subQueryPredicateDef;
}
protected int getNumSubQueryPredicates() {
return numSubQueryPredicates;
}
protected int incrNumSubQueryPredicates() {
return ++numSubQueryPredicates;
}
void setWhereClauseSubQueryPredicate(QBSubQuery sq) {
whereClauseSubQueryPredicate = sq;
}
public QBSubQuery getWhereClauseSubQueryPredicate() {
return whereClauseSubQueryPredicate;
}
void setHavingClauseSubQueryPredicate(QBSubQuery sq) {
havingClauseSubQueryPredicate = sq;
}
public QBSubQuery getHavingClauseSubQueryPredicate() {
return havingClauseSubQueryPredicate;
}
void addEncryptedTargetTablePath(Path p) {
if(encryptedTargetTablePaths == null) {
encryptedTargetTablePaths = new ArrayList<>();
}
encryptedTargetTablePaths.add(p);
}
/**
* List of dbName.tblName of encrypted target tables of insert statement
* Used to support Insert ... values(...)
*/
List getEncryptedTargetTablePaths() {
if(encryptedTargetTablePaths == null) {
return Collections.emptyList();
}
return encryptedTargetTablePaths;
}
}