org.apache.calcite.rex.RexBuilder.makeAbstractCast()方法的使用及代码示例

x33g5p2x  于2022-01-29 转载在 其他  
字(10.7k)|赞(0)|评价(0)|浏览(79)

本文整理了Java中org.apache.calcite.rex.RexBuilder.makeAbstractCast方法的一些代码示例，展示了RexBuilder.makeAbstractCast的具体用法。这些代码示例主要来源于Github/Stackoverflow/Maven等平台，是从一些精选项目中提取出来的代码，具有较强的参考意义，能在一定程度帮忙到你。RexBuilder.makeAbstractCast方法的具体详情如下：
包路径：org.apache.calcite.rex.RexBuilder
类名称：RexBuilder
方法名：makeAbstractCast

RexBuilder.makeAbstractCast介绍

[英]Creates a call to the CAST operator.
[中]创建对CAST操作符的调用。

代码示例

代码示例来源：origin: apache/hive

case 2:
 node = rexBuilder.makeCall(SqlStdOperatorTable.MULTIPLY, merges);
 node = rexBuilder.makeAbstractCast(aggregateCall.type, node);
 break;
default:

代码示例来源：origin: apache/hive

private RexNode handleExplicitCast(ExprNodeGenericFuncDesc func, List<RexNode> childRexNodeLst)
  throws CalciteSemanticException {
 RexNode castExpr = null;
 if (childRexNodeLst != null && childRexNodeLst.size() == 1) {
  GenericUDF udf = func.getGenericUDF();
  if ((udf instanceof GenericUDFToChar) || (udf instanceof GenericUDFToVarchar)
    || (udf instanceof GenericUDFToString)
    || (udf instanceof GenericUDFToDecimal) || (udf instanceof GenericUDFToDate)
    || (udf instanceof GenericUDFTimestamp) || (udf instanceof GenericUDFToTimestampLocalTZ)
    || (udf instanceof GenericUDFToBinary) || castExprUsingUDFBridge(udf)) {
   castExpr = cluster.getRexBuilder().makeAbstractCast(
     TypeConverter.convert(func.getTypeInfo(), cluster.getTypeFactory()),
     childRexNodeLst.get(0));
  }
 }
 return castExpr;
}

代码示例来源：origin: apache/drill

case 2:
 node = rexBuilder.makeCall(SqlStdOperatorTable.MULTIPLY, merges);
 node = rexBuilder.makeAbstractCast(aggregateCall.type, node);
 break;
default:

代码示例来源：origin: apache/hive

@Override
public ASTNode visitLiteral(RexLiteral literal) {
 if (RexUtil.isNull(literal) && literal.getType().getSqlTypeName() != SqlTypeName.NULL
   && rexBuilder != null) {
  // It is NULL value with different type, we need to introduce a CAST
  // to keep it
  if(nullLiteralMap.containsKey(literal)) {
   return ASTBuilder.literal(literal, useTypeQualInLiteral);
  }
  nullLiteralMap.put(literal, true);
  RexNode r = rexBuilder.makeAbstractCast(literal.getType(), literal);
  return r.accept(this);
 }
 return ASTBuilder.literal(literal, useTypeQualInLiteral);
}

代码示例来源：origin: apache/drill

private RexNode handleExplicitCast(ExprNodeGenericFuncDesc func, List<RexNode> childRexNodeLst)
  throws CalciteSemanticException {
 RexNode castExpr = null;
 if (childRexNodeLst != null && childRexNodeLst.size() == 1) {
  GenericUDF udf = func.getGenericUDF();
  if ((udf instanceof GenericUDFToChar) || (udf instanceof GenericUDFToVarchar)
    || (udf instanceof GenericUDFToDecimal) || (udf instanceof GenericUDFToDate)
    // Calcite can not specify the scale for timestamp. As a result, all
    // the millisecond part will be lost
    || (udf instanceof GenericUDFTimestamp)
    || (udf instanceof GenericUDFToBinary) || castExprUsingUDFBridge(udf)) {
   castExpr = cluster.getRexBuilder().makeAbstractCast(
     TypeConverter.convert(func.getTypeInfo(), cluster.getTypeFactory()),
     childRexNodeLst.get(0));
  }
 }
 return castExpr;
}

代码示例来源：origin: apache/incubator-druid

@Test
public void testCastFromDate()
{
 testExpression(
   rexBuilder.makeAbstractCast(
     typeFactory.createSqlType(SqlTypeName.VARCHAR),
     rexBuilder.makeAbstractCast(
       typeFactory.createSqlType(SqlTypeName.DATE),
       inputRef("t")
     )
   ),
   DruidExpression.fromExpression(
     "timestamp_format(timestamp_floor(\"t\",'P1D',null,'UTC'),'yyyy-MM-dd','UTC')"
   ),
   "2000-02-03"
 );
 testExpression(
   rexBuilder.makeAbstractCast(
     typeFactory.createSqlType(SqlTypeName.BIGINT),
     rexBuilder.makeAbstractCast(
       typeFactory.createSqlType(SqlTypeName.DATE),
       inputRef("t")
     )
   ),
   DruidExpression.fromExpression("timestamp_floor(\"t\",'P1D',null,'UTC')"),
   DateTimes.of("2000-02-03").getMillis()
 );
}

代码示例来源：origin: apache/incubator-druid

rexBuilder.makeAbstractCast(
  typeFactory.createSqlType(SqlTypeName.VARCHAR),
  rexBuilder.makeAbstractCast(
    typeFactory.createSqlType(SqlTypeName.TIMESTAMP),
    inputRef("t")
rexBuilder.makeAbstractCast(
  typeFactory.createSqlType(SqlTypeName.BIGINT),
  rexBuilder.makeAbstractCast(
    typeFactory.createSqlType(SqlTypeName.TIMESTAMP),
    inputRef("t")

代码示例来源：origin: apache/incubator-druid

@Test
public void testCastAsDate()
{
 testExpression(
   rexBuilder.makeAbstractCast(
     typeFactory.createSqlType(SqlTypeName.DATE),
     inputRef("t")
   ),
   DruidExpression.fromExpression("timestamp_floor(\"t\",'P1D',null,'UTC')"),
   DateTimes.of("2000-02-03").getMillis()
 );
 testExpression(
   rexBuilder.makeAbstractCast(
     typeFactory.createSqlType(SqlTypeName.DATE),
     inputRef("dstr")
   ),
   DruidExpression.fromExpression(
     "timestamp_floor(timestamp_parse(\"dstr\",null,'UTC'),'P1D',null,'UTC')"
   ),
   DateTimes.of("2000-02-03").getMillis()
 );
}

代码示例来源：origin: apache/incubator-druid

@Test
public void testCastAsTimestamp()
{
 testExpression(
   rexBuilder.makeAbstractCast(
     typeFactory.createSqlType(SqlTypeName.TIMESTAMP),
     inputRef("t")
   ),
   DruidExpression.of(
     SimpleExtraction.of("t", null),
     "\"t\""
   ),
   DateTimes.of("2000-02-03T04:05:06Z").getMillis()
 );
 testExpression(
   rexBuilder.makeAbstractCast(
     typeFactory.createSqlType(SqlTypeName.TIMESTAMP),
     inputRef("tstr")
   ),
   DruidExpression.of(
     null,
     "timestamp_parse(\"tstr\",null,'UTC')"
   ),
   DateTimes.of("2000-02-03T04:05:06Z").getMillis()
 );
}

代码示例来源：origin: apache/drill

r = select.getCluster().getRexBuilder().makeAbstractCast(r.getType(), r);
r = select.getCluster().getRexBuilder().makeAbstractCast(r.getType(), r);

代码示例来源：origin: Qihoo360/Quicksql

/**
 * Creates a call to the CAST operator.
 *
 * <p>This method enables to create {@code CAST(42 nullable int)} expressions.</p>
 *
 * @param e input node
 * @param type type to cast to
 * @return call to CAST operator
 */
protected RexNode abstractCast(RexNode e, RelDataType type) {
 return rexBuilder.makeAbstractCast(type, e);
}

代码示例来源：origin: dremio/dremio-oss

@Override
public RexNode makeAbstractCast(
 RelDataType type,
 RexNode exp) {
 if (exp.getType().equals(type)) {
  return exp;
 }
 return super.makeAbstractCast(type, exp);
}

代码示例来源：origin: org.apache.calcite/calcite-core

/**
 * Creates a call to the CAST operator.
 *
 * <p>This method enables to create {@code CAST(42 nullable int)} expressions.</p>
 *
 * @param e input node
 * @param type type to cast to
 * @return call to CAST operator
 */
protected RexNode abstractCast(RexNode e, RelDataType type) {
 return rexBuilder.makeAbstractCast(type, e);
}

代码示例来源：origin: Qihoo360/Quicksql

private RexNode visit(final RexNode call) {
  int i = reducibleExps.indexOf(call);
  if (i == -1) {
   return null;
  }
  RexNode replacement = reducedValues.get(i);
  if (addCasts.get(i)
    && (replacement.getType() != call.getType())) {
   // Handle change from nullable to NOT NULL by claiming
   // that the result is still nullable, even though
   // we know it isn't.
   //
   // Also, we cannot reduce CAST('abc' AS VARCHAR(4)) to 'abc'.
   // If we make 'abc' of type VARCHAR(4), we may later encounter
   // the same expression in a Project's digest where it has
   // type VARCHAR(3), and that's wrong.
   replacement =
     simplify.rexBuilder.makeAbstractCast(call.getType(), replacement);
  }
  return replacement;
 }
}

代码示例来源：origin: org.apache.calcite/calcite-core

private RexNode visit(final RexNode call) {
  int i = reducibleExps.indexOf(call);
  if (i == -1) {
   return null;
  }
  RexNode replacement = reducedValues.get(i);
  if (addCasts.get(i)
    && (replacement.getType() != call.getType())) {
   // Handle change from nullable to NOT NULL by claiming
   // that the result is still nullable, even though
   // we know it isn't.
   //
   // Also, we cannot reduce CAST('abc' AS VARCHAR(4)) to 'abc'.
   // If we make 'abc' of type VARCHAR(4), we may later encounter
   // the same expression in a Project's digest where it has
   // type VARCHAR(3), and that's wrong.
   replacement =
     simplify.rexBuilder.makeAbstractCast(call.getType(), replacement);
  }
  return replacement;
 }
}

代码示例来源：origin: Qihoo360/Quicksql

/**
 * Makes a cast of a value to NOT NULL;
 * no-op if the type already has NOT NULL.
 */
public RexNode makeNotNull(RexNode exp) {
 final RelDataType type = exp.getType();
 if (!type.isNullable()) {
  return exp;
 }
 final RelDataType notNullType =
   typeFactory.createTypeWithNullability(type, false);
 return makeAbstractCast(notNullType, exp);
}

代码示例来源：origin: org.apache.calcite/calcite-core

/**
 * Makes a cast of a value to NOT NULL;
 * no-op if the type already has NOT NULL.
 */
public RexNode makeNotNull(RexNode exp) {
 final RelDataType type = exp.getType();
 if (!type.isNullable()) {
  return exp;
 }
 final RelDataType notNullType =
   typeFactory.createTypeWithNullability(type, false);
 return makeAbstractCast(notNullType, exp);
}

代码示例来源：origin: com.facebook.presto.hive/hive-apache

private RexNode handleExplicitCast(ExprNodeGenericFuncDesc func, List<RexNode> childRexNodeLst)
  throws CalciteSemanticException {
 RexNode castExpr = null;
 if (childRexNodeLst != null && childRexNodeLst.size() == 1) {
  GenericUDF udf = func.getGenericUDF();
  if ((udf instanceof GenericUDFToChar) || (udf instanceof GenericUDFToVarchar)
    || (udf instanceof GenericUDFToDecimal) || (udf instanceof GenericUDFToDate)
    || (udf instanceof GenericUDFToBinary) || castExprUsingUDFBridge(udf)) {
   castExpr = cluster.getRexBuilder().makeAbstractCast(
     TypeConverter.convert(func.getTypeInfo(), cluster.getTypeFactory()),
     childRexNodeLst.get(0));
  }
 }
 return castExpr;
}

代码示例来源：origin: Qihoo360/Quicksql

public AggregateCall topSplit(RexBuilder rexBuilder,
  Registry<RexNode> extra, int offset, RelDataType inputRowType,
  AggregateCall aggregateCall, int leftSubTotal, int rightSubTotal) {
 final List<RexNode> merges = new ArrayList<>();
 final List<RelDataTypeField> fieldList = inputRowType.getFieldList();
 if (leftSubTotal >= 0) {
  final RelDataType type = fieldList.get(leftSubTotal).getType();
  merges.add(rexBuilder.makeInputRef(type, leftSubTotal));
 }
 if (rightSubTotal >= 0) {
  final RelDataType type = fieldList.get(rightSubTotal).getType();
  merges.add(rexBuilder.makeInputRef(type, rightSubTotal));
 }
 RexNode node;
 switch (merges.size()) {
 case 1:
  node = merges.get(0);
  break;
 case 2:
  node = rexBuilder.makeCall(SqlStdOperatorTable.MULTIPLY, merges);
  node = rexBuilder.makeAbstractCast(aggregateCall.type, node);
  break;
 default:
  throw new AssertionError("unexpected count " + merges);
 }
 int ordinal = extra.register(node);
 return AggregateCall.create(getMergeAggFunctionOfTopSplit(), false, false,
   ImmutableList.of(ordinal), -1, aggregateCall.type,
   aggregateCall.name);
}

代码示例来源：origin: org.apache.calcite/calcite-core

private static RexNode simplifyBooleanCase(RexBuilder rexBuilder,
  List<CaseBranch> inputBranches, RexUnknownAs unknownAs, RelDataType branchType) {
 RexNode result;
 // prepare all condition/branches for boolean interpretation
 // It's done here make these interpretation changes available to case2or simplifications
 // but not interfere with the normal simplifcation recursion
 List<CaseBranch> branches = new ArrayList<>();
 for (CaseBranch branch : inputBranches) {
  if (!isSafeExpression(branch.cond) || !isSafeExpression(branch.value)) {
   return null;
  }
  RexNode cond;
  RexNode value;
  if (branch.cond.getType().isNullable()) {
   cond = rexBuilder.makeCall(SqlStdOperatorTable.IS_TRUE, branch.cond);
  } else {
   cond = branch.cond;
  }
  if (!branchType.equals(branch.value.getType())) {
   value = rexBuilder.makeAbstractCast(branchType, branch.value);
  } else {
   value = branch.value;
  }
  branches.add(new CaseBranch(cond, value));
 }
 result = simplifyBooleanCaseGeneric(rexBuilder, branches, branchType);
 return result;
}

内容来源于网络，如有侵权，请联系作者删除！

RexBuilder

相关文章

    查看更多
    微信公众号

    热门标签

    更多
    JavaquerypythonNode开发语言requestUtil数据库Table后端算法LoggerMessageElementParser

    最新文章

    更多

    RexBuilder类方法

    • 蜀ICP备13028337号-1 大数据知识库 https://www.saoniuhuo.com © All rights reserved
    • 本站内容来源互联网,如果侵犯您的权益请联系我们删除, 联系方式：448109455@qq.com