RemoveCorrelationRexShuttle( RexBuilder rexBuilder, boolean projectPulledAboveLeftCorrelator, RexInputRef nullIndicator, Set<Integer> isCount) { this.projectPulledAboveLeftCorrelator = projectPulledAboveLeftCorrelator; this.nullIndicator = nullIndicator; // may be null this.isCount = ImmutableSet.copyOf(isCount); this.rexBuilder = rexBuilder; this.typeFactory = rexBuilder.getTypeFactory(); }
public RemoveCorrelationRexShuttle( RexBuilder rexBuilder, boolean projectPulledAboveLeftCorrelator, RexInputRef nullIndicator, Set<Integer> isCount) { this.projectPulledAboveLeftCorrelator = projectPulledAboveLeftCorrelator; this.nullIndicator = nullIndicator; // may be null this.isCount = ImmutableSet.copyOf(isCount); this.rexBuilder = rexBuilder; this.typeFactory = rexBuilder.getTypeFactory(); }
public StreamsPlanCreator(Map<String, ISqlStreamsDataSource> sources, RexBuilder rexBuilder) { this.sources = sources; this.rexCompiler = new RexNodeToJavaCodeCompiler(rexBuilder); this.typeFactory = (JavaTypeFactory) rexBuilder.getTypeFactory(); this.streamBuilder = new StreamBuilder(); this.dataContext = new StormDataContext(); }
/*********************** Convert Hive Types To Calcite Types ***********************/ public static RelDataType getType(RelOptCluster cluster, List<ColumnInfo> cInfoLst) throws CalciteSemanticException { RexBuilder rexBuilder = cluster.getRexBuilder(); RelDataTypeFactory dtFactory = rexBuilder.getTypeFactory(); List<RelDataType> fieldTypes = new LinkedList<RelDataType>(); List<String> fieldNames = new LinkedList<String>(); for (ColumnInfo ci : cInfoLst) { fieldTypes.add(convert(ci.getType(), dtFactory)); fieldNames.add(ci.getInternalName()); } return dtFactory.createStructType(fieldTypes, fieldNames); }
/*********************** Convert Hive Types To Calcite Types ***********************/ public static RelDataType getType(RelOptCluster cluster, List<ColumnInfo> cInfoLst) throws CalciteSemanticException { RexBuilder rexBuilder = cluster.getRexBuilder(); RelDataTypeFactory dtFactory = rexBuilder.getTypeFactory(); List<RelDataType> fieldTypes = new LinkedList<RelDataType>(); List<String> fieldNames = new LinkedList<String>(); for (ColumnInfo ci : cInfoLst) { fieldTypes.add(convert(ci.getType(), dtFactory)); fieldNames.add(ci.getInternalName()); } return dtFactory.createStructType(fieldTypes, fieldNames); }
public static RelDataType getType(RelOptCluster cluster, RowResolver rr, List<String> neededCols) throws CalciteSemanticException { RexBuilder rexBuilder = cluster.getRexBuilder(); RelDataTypeFactory dtFactory = rexBuilder.getTypeFactory(); RowSchema rs = rr.getRowSchema(); List<RelDataType> fieldTypes = new LinkedList<RelDataType>(); List<String> fieldNames = new LinkedList<String>(); for (ColumnInfo ci : rs.getSignature()) { if (neededCols == null || neededCols.contains(ci.getInternalName())) { fieldTypes.add(convert(ci.getType(), dtFactory)); fieldNames.add(ci.getInternalName()); } } return dtFactory.createStructType(fieldTypes, fieldNames); }
public static RelDataType getType(RelOptCluster cluster, RowResolver rr, List<String> neededCols) throws CalciteSemanticException { RexBuilder rexBuilder = cluster.getRexBuilder(); RelDataTypeFactory dtFactory = rexBuilder.getTypeFactory(); RowSchema rs = rr.getRowSchema(); List<RelDataType> fieldTypes = new LinkedList<RelDataType>(); List<String> fieldNames = new LinkedList<String>(); for (ColumnInfo ci : rs.getSignature()) { if (neededCols == null || neededCols.contains(ci.getInternalName())) { fieldTypes.add(convert(ci.getType(), dtFactory)); fieldNames.add(ci.getInternalName()); } } return dtFactory.createStructType(fieldTypes, fieldNames); }
RelDataTypeFactory dtFactory = rexBuilder.getTypeFactory();
RelDataTypeFactory dtFactory = cluster.getRexBuilder().getTypeFactory(); for (RelDataTypeField field : rowType.getFieldList()) { if (DruidTable.DEFAULT_TIMESTAMP_COLUMN.equals(field.getName())) {
RelDataTypeFactory dtFactory = rexBuilder.getTypeFactory(); RowSchema rs = selectRR.getRowSchema(); for (ColumnInfo ci : rs.getSignature()) {
final BlockBuilder builder = new BlockBuilder(); final JavaTypeFactoryImpl javaTypeFactory = new JavaTypeFactoryImpl(rexBuilder.getTypeFactory().getTypeSystem());
RelDataTypeFactory dtFactory = rexBuilder.getTypeFactory(); RowSchema rs = selectRR.getRowSchema(); for (ColumnInfo ci : rs.getSignature()) {
protected RexNode convert(ExprNodeConstantDesc literal) throws CalciteSemanticException { RexBuilder rexBuilder = cluster.getRexBuilder(); RelDataTypeFactory dtFactory = rexBuilder.getTypeFactory(); PrimitiveTypeInfo hiveType = (PrimitiveTypeInfo) literal.getTypeInfo(); RelDataType calciteDataType = TypeConverter.convert(hiveType, dtFactory);
protected RexNode convert(ExprNodeConstantDesc literal) throws CalciteSemanticException { final RexBuilder rexBuilder = cluster.getRexBuilder(); final RelDataTypeFactory dtFactory = rexBuilder.getTypeFactory(); final PrimitiveTypeInfo hiveType = (PrimitiveTypeInfo) literal.getTypeInfo(); final RelDataType calciteDataType = TypeConverter.convert(hiveType, dtFactory); rexBuilder.getTypeFactory().createSqlType( SqlTypeName.TIMESTAMP, rexBuilder.getTypeFactory().getTypeSystem().getDefaultPrecision(SqlTypeName.TIMESTAMP)), false); break; rexBuilder.getTypeFactory().getTypeSystem().getDefaultPrecision(SqlTypeName.TIMESTAMP_WITH_LOCAL_TIME_ZONE)); break; case INTERVAL_YEAR_MONTH:
+ rightKey + " (type " + rightKeyType + ")"); RelDataType targetKeyType = TypeConverter.convert(tgtType, rexBuilder.getTypeFactory());
Set<String> metrics = new HashSet<>(); RexBuilder rexBuilder = cluster.getRexBuilder(); RelDataTypeFactory dtFactory = rexBuilder.getTypeFactory(); List<RelDataType> druidColTypes = new ArrayList<>(); List<String> druidColNames = new ArrayList<>();
/** * Create a cast for partition column. Partition column is output as "VARCHAR" in native parquet reader. Cast it * appropriate type according the partition type in HiveScan. */ private RexNode createPartitionColumnCast(final DrillScanRel hiveScanRel, final DrillScanRel nativeScanRel, final String outputColName, final String dirColName, final RexBuilder rb) { final RelDataType outputType = hiveScanRel.getRowType().getField(outputColName, false, false).getType(); final RelDataTypeField inputField = nativeScanRel.getRowType().getField(dirColName, false, false); final RexInputRef inputRef = rb.makeInputRef(rb.getTypeFactory().createSqlType(SqlTypeName.VARCHAR), inputField.getIndex()); if (outputType.getSqlTypeName() == SqlTypeName.CHAR) { return rb.makeCall(RTRIM, inputRef); } return rb.makeCast(outputType, inputRef); } }
getFieldType(oldAggRel.getInput(), iAvgInput), true); final RelDataType sumReturnType = getSumReturnType( rexBuilder.getTypeFactory(), sum0InputType, oldCall.getType()); final AggregateCall sumCall = AggregateCall.create(
final int argRefOrdinal = lookupOrAdd(inputExprs, argRef); final RelDataType sumReturnType = getSumReturnType( rexBuilder.getTypeFactory(), argRef.getType(), oldCall.getType()); final int argSquaredOrdinal = lookupOrAdd(inputExprs, argSquared); final RelDataType sumSquaredReturnType = getSumReturnType( rexBuilder.getTypeFactory(), argSquared.getType(), oldCall.getType());
getFieldType(oldAggRel.getInput(), iAvgInput), true); final RelDataType sumReturnType = getSumReturnType( rexBuilder.getTypeFactory(), avgInputType, oldCall.getType()); final AggregateCall sumCall = AggregateCall.create(