Examples with ColumnType org.apache.druid.segment.column.ColumnType used on opensource projects

Example 41 with ColumnType

use of org.apache.druid.segment.column.ColumnType in project druid by druid-io.

the class EarliestLatestAnySqlAggregator method toDruidAggregation.

@Nullable
@Override
public Aggregation toDruidAggregation(final PlannerContext plannerContext, final RowSignature rowSignature, final VirtualColumnRegistry virtualColumnRegistry, final RexBuilder rexBuilder, final String name, final AggregateCall aggregateCall, final Project project, final List<Aggregation> existingAggregations, final boolean finalizeAggregations) {
    final List<RexNode> rexNodes = aggregateCall.getArgList().stream().map(i -> Expressions.fromFieldAccess(rowSignature, project, i)).collect(Collectors.toList());
    final List<DruidExpression> args = Expressions.toDruidExpressions(plannerContext, rowSignature, rexNodes);
    if (args == null) {
        return null;
    }
    final String aggregatorName = finalizeAggregations ? Calcites.makePrefixedName(name, "a") : name;
    final ColumnType outputType = Calcites.getColumnTypeForRelDataType(aggregateCall.getType());
    if (outputType == null) {
        throw new ISE("Cannot translate output sqlTypeName[%s] to Druid type for aggregator[%s]", aggregateCall.getType().getSqlTypeName(), aggregateCall.getName());
    }
    final String fieldName = getColumnName(plannerContext, virtualColumnRegistry, args.get(0), rexNodes.get(0));
    final AggregatorFactory theAggFactory;
    switch(args.size()) {
        case 1:
            theAggFactory = aggregatorType.createAggregatorFactory(aggregatorName, fieldName, null, outputType, -1);
            break;
        case 2:
            theAggFactory = aggregatorType.createAggregatorFactory(aggregatorName, fieldName, null, outputType, RexLiteral.intValue(rexNodes.get(1)));
            break;
        default:
            throw new IAE("aggregation[%s], Invalid number of arguments[%,d] to [%s] operator", aggregatorName, args.size(), aggregatorType.name());
    }
    return Aggregation.create(Collections.singletonList(theAggFactory), finalizeAggregations ? new FinalizingFieldAccessPostAggregator(name, aggregatorName) : null);
}

Example 42 with ColumnType

use of org.apache.druid.segment.column.ColumnType in project druid by druid-io.

the class RowSignatures method fromRelDataType.

public static RowSignature fromRelDataType(final List<String> rowOrder, final RelDataType rowType) {
    if (rowOrder.size() != rowType.getFieldCount()) {
        throw new IAE("Field count %d != %d", rowOrder.size(), rowType.getFieldCount());
    }
    final RowSignature.Builder rowSignatureBuilder = RowSignature.builder();
    for (int i = 0; i < rowOrder.size(); i++) {
        final RelDataType dataType = rowType.getFieldList().get(i).getType();
        final ColumnType valueType = Calcites.getColumnTypeForRelDataType(dataType);
        rowSignatureBuilder.add(rowOrder.get(i), valueType);
    }
    return rowSignatureBuilder.build();
}

Example 43 with ColumnType

use of org.apache.druid.segment.column.ColumnType in project druid by druid-io.

the class RowSignatures method toRelDataType.

/**
 * Returns a Calcite RelDataType corresponding to a row signature.
 */
public static RelDataType toRelDataType(final RowSignature rowSignature, final RelDataTypeFactory typeFactory) {
    final RelDataTypeFactory.Builder builder = typeFactory.builder();
    final boolean nullNumeric = !NullHandling.replaceWithDefault();
    for (final String columnName : rowSignature.getColumnNames()) {
        final RelDataType type;
        if (ColumnHolder.TIME_COLUMN_NAME.equals(columnName)) {
            type = Calcites.createSqlType(typeFactory, SqlTypeName.TIMESTAMP);
        } else {
            final ColumnType columnType = rowSignature.getColumnType(columnName).orElseThrow(() -> new ISE("Encountered null type for column[%s]", columnName));
            switch(columnType.getType()) {
                case STRING:
                    // Note that there is no attempt here to handle multi-value in any special way. Maybe one day...
                    type = Calcites.createSqlTypeWithNullability(typeFactory, SqlTypeName.VARCHAR, true);
                    break;
                case LONG:
                    type = Calcites.createSqlTypeWithNullability(typeFactory, SqlTypeName.BIGINT, nullNumeric);
                    break;
                case FLOAT:
                    type = Calcites.createSqlTypeWithNullability(typeFactory, SqlTypeName.FLOAT, nullNumeric);
                    break;
                case DOUBLE:
                    type = Calcites.createSqlTypeWithNullability(typeFactory, SqlTypeName.DOUBLE, nullNumeric);
                    break;
                case ARRAY:
                    switch(columnType.getElementType().getType()) {
                        case STRING:
                            type = Calcites.createSqlArrayTypeWithNullability(typeFactory, SqlTypeName.VARCHAR, true);
                            break;
                        case LONG:
                            type = Calcites.createSqlArrayTypeWithNullability(typeFactory, SqlTypeName.BIGINT, nullNumeric);
                            break;
                        case DOUBLE:
                            type = Calcites.createSqlArrayTypeWithNullability(typeFactory, SqlTypeName.DOUBLE, nullNumeric);
                            break;
                        default:
                            throw new ISE("valueType[%s] not translatable", columnType);
                    }
                    break;
                case COMPLEX:
                    type = typeFactory.createTypeWithNullability(new ComplexSqlType(SqlTypeName.OTHER, columnType, true), true);
                    break;
                default:
                    throw new ISE("valueType[%s] not translatable", columnType);
            }
        }
        builder.add(columnName, type);
    }
    return builder.build();
}