Java HiveParser.TOK_SELEXPR属性代码示例

/**
 * Check if expression is answerable from fact, then push it to fact pushdown subquery
 *
 * @param node
 * @return true if expressions is used
 */
public boolean isExpressionsAnswerableFromFact(ASTNode node) {
  boolean isAnswerable = true;
  for (int i = 0; i < node.getChildCount(); i++) {
    if (node.getChild(i).getType() == HiveParser.TOK_SELEXPR) {
      int cnt = getColumnCount((ASTNode) node.getChild(i));
      if (cnt >= 2) {
        if (cnt == getNumFactTableInExpressions((ASTNode) node.getChild(i), new MutableInt(0))) {
          isAnswerable = true;
        } else {
          isAnswerable = false;
        }
      }
    }
  }
  return isAnswerable;
}
 

private ASTNode processAggregate(ASTNode astNode, ASTNode innerSelectAST,
    AliasDecider aliasDecider, boolean isSelectAst) {
  ASTNode innerSelectASTWithoutAlias = MetastoreUtil.copyAST(astNode);
  ASTNode innerSelectExprAST = new ASTNode(new CommonToken(HiveParser.TOK_SELEXPR, "TOK_SELEXPR"));
  innerSelectExprAST.addChild(innerSelectASTWithoutAlias);
  String alias = aliasDecider.decideAlias(astNode);
  ASTNode aliasNode = new ASTNode(new CommonToken(Identifier, alias));
  innerSelectExprAST.addChild(aliasNode);
  innerSelectAST.addChild(innerSelectExprAST);
  ASTNode dotAST = getDotAST(cubeql.getCube().getName(), alias);
  ASTNode outerAST = new ASTNode(new CommonToken(TOK_FUNCTION, "TOK_FUNCTION"));
  //TODO: take care or non-transitive aggregate functions
  outerAST.addChild(new ASTNode(new CommonToken(Identifier, astNode.getChild(0).getText())));
  outerAST.addChild(dotAST);
  HashableASTNode innerAST = new HashableASTNode(innerSelectASTWithoutAlias);
  if (isSelectAst && !innerToOuterSelectASTs.containsKey(innerAST)) {
    innerToOuterSelectASTs.put(innerAST, outerAST);
  } else if (!isSelectAst && !innerToOuterHavingASTs.containsKey(innerAST)) {
    innerToOuterHavingASTs.put(innerAST, outerAST);
  }
  return outerAST;
}
 

private void promoteGroupby(CubeQueryContext cubeql, List<SelectPhraseContext> selectExprs, List<String> groupByExprs)
  throws LensException {
  if (!groupbyPromotionEnabled) {
    return;
  }

  if (selectExprs.size() > 0) {
    log.info("Not promoting groupby expression to select, since there are expression projected");
    return;
  }

  int index = 0;
  for (String expr : groupByExprs) {
    if (!contains(selectExprs, expr)) {
      ASTNode exprAST = HQLParser.parseExpr(expr, cubeql.getConf());
      ASTNode parent = new ASTNode(new CommonToken(HiveParser.TOK_SELEXPR, "TOK_SELEXPR"));
      parent.addChild(exprAST);
      exprAST.setParent(parent);
      addChildAtIndex(index, cubeql.getSelectAST(), parent);
      updateSelectPhrase(cubeql, index, parent);
      index++;
    }
  }
}
 

@Override
public boolean generate(ASTNode hiveRoot, CommonTree sqlRoot, ASTNode currentHiveNode,
    CommonTree currentSqlNode, TranslateContext context) throws SqlXlateException {
  if (currentHiveNode.getChildCount() == 2&&currentHiveNode.getType()==HiveParser.TOK_SELEXPR) {

    ASTNode dot = super.newHiveASTNode(HiveParser.DOT, ".");
    dot.addChild((ASTNode) currentHiveNode.getChild(0));
    // if children count == 2 the second should only be text element
    dot.addChild((ASTNode) currentHiveNode.getChild(1).getChild(0));
    currentHiveNode.deleteChild(0);
    currentHiveNode.deleteChild(0);
    currentHiveNode.addChild(dot);
    LOG.debug("Generated Cascated Element : " + dot.toStringTree());
    return true;
  }
  return true;
}
 

/**
 * Check if subtree represents select distinct *
 *
 * @param select
 * @return
 */
public static boolean isSelectDistinctStar(ASTNode select) {
  if (select.getType() != HiveParser.TOK_SELECTDI) {
    return false;
  }
  if (select.getChildCount() > 1) {
    return false;
  }
  if (select.getChild(0).getType() != HiveParser.TOK_SELEXPR) {
    return false;
  }
  if (select.getChild(0).getChild(0).getType() != HiveParser.TOK_ALLCOLREF) {
    return false;
  }
  return true;
}
 

private ASTNode getSelectExprAST() {
  return new ASTNode(new CommonToken(HiveParser.TOK_SELEXPR, "TOK_SELEXPR"));
}