Java SqlTypeFamily类代码示例

import org.apache.calcite.sql.type.SqlTypeFamily; //导入依赖的package包/类
public SqlDatePartOperator() {
  super(
      "DATE_PART",
      SqlKind.OTHER_FUNCTION,
      ReturnTypes.BIGINT_NULLABLE,
      null,
      OperandTypes.sequence(
          "<PERIOD LITERAL>, <DATE or TIMESTAMP or INTERVAL>",
          new EnumeratedListChecker(VALID_PERIODS.keySet()),
          OperandTypes.or(
              OperandTypes.family(SqlTypeFamily.DATE),
              OperandTypes.family(SqlTypeFamily.TIMESTAMP),
              OperandTypes.family(SqlTypeFamily.DATETIME),
              OperandTypes.family(SqlTypeFamily.DATETIME_INTERVAL),
              OperandTypes.family(SqlTypeFamily.INTERVAL_DAY_TIME),
              OperandTypes.family(SqlTypeFamily.INTERVAL_YEAR_MONTH))
          ),
          SqlFunctionCategory.SYSTEM);
}
 

import org.apache.calcite.sql.type.SqlTypeFamily; //导入依赖的package包/类
public static boolean isTemporal(final LayoutField field) {
  final Optional<SqlTypeFamily> familyOpt = getSqlTypeFamily(field);
  if (!familyOpt.isPresent()) {
    return false;
  }

  final SqlTypeFamily family = familyOpt.get();
  switch (family) {
    case DATETIME:
    case TIMESTAMP:
    case DATE:
    case TIME:
      return true;
    default:
      return false;
  }
}
 

import org.apache.calcite.sql.type.SqlTypeFamily; //导入依赖的package包/类
private Iterable<AggregateCall> createMeasuresFor(final RelNode view, final LayoutField field) {
  final Optional<SqlTypeFamily> family = TypeUtils.getSqlTypeFamily(field);
  if (!family.isPresent()) {
    return ImmutableList.of();
  }

  return FluentIterable
      .from(AccelerationUtils.selfOrEmptyCollection(calls.get(family.get())))
      .transform(new Function<SqlAggFunction, AggregateCall>() {
        private int index = 0;

        @Nullable
        @Override
        public AggregateCall apply(@Nullable final SqlAggFunction func) {
          // no distinct measures for now
          final int inputRef = getField(field.getName()).getIndex();
          return AggregateCall.create(func, false, ImmutableList.of(inputRef), -1, 1, view, null,
              String.format("agg-%s-%s", inputRef, index++));
        }
      });
}
 

import org.apache.calcite.sql.type.SqlTypeFamily; //导入依赖的package包/类
private Layout addAggLayout(Acceleration acceleration, int layoutNum) {
  List<LayoutDimensionField> dimFields = new ArrayList<>();
  dimFields.add(new LayoutDimensionField().setName("dim" + layoutNum).setTypeFamily(SqlTypeFamily.CHARACTER.toString()));
  List<LayoutField> measureFields = new ArrayList<>();
  measureFields.add(new LayoutField().setName("measure" + layoutNum).setTypeFamily(SqlTypeFamily.NUMERIC.toString()));

  Layout aggLayout = new Layout()
    .setLayoutType(LayoutType.AGGREGATION)
    .setDetails(new LayoutDetails().setDimensionFieldList(dimFields).setMeasureFieldList(measureFields))
    .setId(new LayoutId("aggLayout" + layoutNum))
    .setName("aggLayout")
    .setVersion(0);

  acceleration.getAggregationLayouts().getLayoutList().add(aggLayout);

  when(store.get(aggLayout.getId())).thenReturn(Optional.<MaterializedLayout>absent());

  return aggLayout;
}
 

import org.apache.calcite.sql.type.SqlTypeFamily; //导入依赖的package包/类
private Layout addRawLayout(Acceleration acceleration, int layoutNum) {
  List<LayoutField> measureFields = new ArrayList<>();
  measureFields.add(new LayoutField().setName("dim" + layoutNum).setTypeFamily(SqlTypeFamily.CHARACTER.toString()));
  measureFields.add(new LayoutField().setName("measure" + layoutNum).setTypeFamily(SqlTypeFamily.NUMERIC.toString()));

  Layout rawLayout = new Layout()
    .setLayoutType(LayoutType.RAW)
    .setDetails(new LayoutDetails().setDisplayFieldList(measureFields))
    .setId(new LayoutId("rawLayout" + layoutNum))
    .setName("rawLayout")
    .setVersion(0);

  acceleration.getRawLayouts().getLayoutList().add(rawLayout);

  when(store.get(rawLayout.getId())).thenReturn(Optional.<MaterializedLayout>absent());

  return rawLayout;
}
 

import org.apache.calcite.sql.type.SqlTypeFamily; //导入依赖的package包/类
private void mockContext(StageContext context) {
  Mockito.reset(context);

  RowType schema = new RowType();
  schema.setFieldList(ImmutableList.of(
    new ViewFieldType("dim0", SqlTypeName.CHAR.getName()).setTypeFamily(SqlTypeFamily.CHARACTER.toString()),
    new ViewFieldType("measure0", SqlTypeName.DOUBLE.getName()).setTypeFamily(SqlTypeFamily.NUMERIC.toString())
  ));

  // Set up an empty acceleration for the original.
  final Acceleration currentAcceleration = new Acceleration()
    .setContext(new AccelerationContext().setDataset(new DatasetConfig().setFullPathList(ImmutableList.of("a", "b"))).setDatasetSchema(schema))
    .setRawLayouts(new LayoutContainer().setLayoutList(new ArrayList<Layout>()))
    .setAggregationLayouts(new LayoutContainer().setLayoutList(new ArrayList<Layout>()));

  when(context.getCurrentAcceleration()).thenReturn(currentAcceleration);
}
 

import org.apache.calcite.sql.type.SqlTypeFamily; //导入依赖的package包/类
private Layout addAggLayout(Acceleration acceleration, int layoutNum) {
  List<LayoutDimensionField> dimFields = new ArrayList<>();
  dimFields.add(new LayoutDimensionField().setName("dim" + layoutNum).setTypeFamily(SqlTypeFamily.CHARACTER.toString()));
  List<LayoutField> measureFields = new ArrayList<>();
  measureFields.add(new LayoutField().setName("measure" + layoutNum).setTypeFamily(SqlTypeFamily.NUMERIC.toString()));

  Layout aggLayout = new Layout()
    .setLayoutType(LayoutType.AGGREGATION)
    .setDetails(new LayoutDetails().setDimensionFieldList(dimFields).setMeasureFieldList(measureFields))
    .setId(new LayoutId("aggLayout" + layoutNum))
    .setName("aggLayout")
    .setVersion(0);

  acceleration.getAggregationLayouts().getLayoutList().add(aggLayout);

  return aggLayout;
}
 

import org.apache.calcite.sql.type.SqlTypeFamily; //导入依赖的package包/类
private Layout addRawLayout(Acceleration acceleration, int layoutNum) {
  List<LayoutField> measureFields = new ArrayList<>();
  measureFields.add(new LayoutField().setName("dim" + layoutNum).setTypeFamily(SqlTypeFamily.CHARACTER.toString()));
  measureFields.add(new LayoutField().setName("measure" + layoutNum).setTypeFamily(SqlTypeFamily.NUMERIC.toString()));

  Layout rawLayout = new Layout()
    .setLayoutType(LayoutType.RAW)
    .setDetails(new LayoutDetails().setDisplayFieldList(measureFields))
    .setId(new LayoutId("rawLayout" + layoutNum))
    .setName("rawLayout")
    .setVersion(0);

  acceleration.getRawLayouts().getLayoutList().add(rawLayout);

  return rawLayout;
}
 

import org.apache.calcite.sql.type.SqlTypeFamily; //导入依赖的package包/类
/**
 * If one operand in a binary operator is a DateTime type, but the other isn't, we should not push down the predicate
 * @param call
 */
public static void checkForIncompatibleDateTimeOperands(RexCall call) {
  RelDataType op1 = call.getOperands().get(0).getType();
  RelDataType op2 = call.getOperands().get(1).getType();
  if (
      (SqlTypeFamily.DATETIME.contains(op1) && !SqlTypeFamily.DATETIME.contains(op2)) ||
      (SqlTypeFamily.DATETIME.contains(op2) && !SqlTypeFamily.DATETIME.contains(op1)) ||
      (SqlTypeFamily.DATE.contains(op1) && !SqlTypeFamily.DATE.contains(op2)) ||
      (SqlTypeFamily.DATE.contains(op2) && !SqlTypeFamily.DATE.contains(op1)) ||
      (SqlTypeFamily.TIMESTAMP.contains(op1) && !SqlTypeFamily.TIMESTAMP.contains(op2)) ||
      (SqlTypeFamily.TIMESTAMP.contains(op2) && !SqlTypeFamily.TIMESTAMP.contains(op1)) ||
      (SqlTypeFamily.TIME.contains(op1) && !SqlTypeFamily.TIME.contains(op2)) ||
      (SqlTypeFamily.TIME.contains(op2) && !SqlTypeFamily.TIME.contains(op1)))
  {
    throw new PredicateAnalyzerException("Cannot handle " + call.getKind() + " expression for _id field, " + call);
  }
}
 

import org.apache.calcite.sql.type.SqlTypeFamily; //导入依赖的package包/类
private SqlSingleOperandTypeChecker getChecker(RelDataType operandType) {
  switch (operandType.getSqlTypeName()) {
  case ARRAY:
    return OperandTypes.family(SqlTypeFamily.INTEGER);
  case MAP:
    return OperandTypes.family(
        operandType.getKeyType().getSqlTypeName().getFamily());
  case ANY:
  case DYNAMIC_STAR:
    return OperandTypes.or(
        OperandTypes.family(SqlTypeFamily.INTEGER),
        OperandTypes.family(SqlTypeFamily.CHARACTER));
  default:
    throw new AssertionError(operandType.getSqlTypeName());
  }
}
 

import org.apache.calcite.sql.type.SqlTypeFamily; //导入依赖的package包/类
public RelDataType getRowType(RelDataTypeFactory factory) {

    // if there are no fields defined, this is a dynamic view.
    if (isDynamic()) {
      return new RelDataTypeDrillImpl(new RelDataTypeHolder(), factory);
    }

    List<RelDataType> types = Lists.newArrayList();
    List<String> names = Lists.newArrayList();

    for (FieldType field : fields) {
      names.add(field.getName());
      RelDataType type;
      if (   SqlTypeFamily.INTERVAL_YEAR_MONTH == field.getType().getFamily()
          || SqlTypeFamily.INTERVAL_DAY_TIME   == field.getType().getFamily() ) {
       type = factory.createSqlIntervalType( field.getIntervalQualifier() );
      } else if (field.getPrecision() == null && field.getScale() == null) {
        type = factory.createSqlType(field.getType());
      } else if (field.getPrecision() != null && field.getScale() == null) {
        type = factory.createSqlType(field.getType(), field.getPrecision());
      } else {
        type = factory.createSqlType(field.getType(), field.getPrecision(), field.getScale());
      }

      if (field.getIsNullable()) {
        types.add(factory.createTypeWithNullability(type, true));
      } else {
        types.add(type);
      }
    }
    return factory.createStructType(types, names);
  }
 

import org.apache.calcite.sql.type.SqlTypeFamily; //导入依赖的package包/类
public RelDataType getRowType(RelDataTypeFactory factory) {

    List<RelDataType> types = Lists.newArrayList();
    List<String> names = Lists.newArrayList();

    for (FieldType field : fields) {
      names.add(field.getName());
      RelDataType type;
      if (   SqlTypeFamily.INTERVAL_YEAR_MONTH == field.getType().getFamily()
          || SqlTypeFamily.INTERVAL_DAY_TIME   == field.getType().getFamily() ) {
       type = factory.createSqlIntervalType( field.getIntervalQualifier() );
      } else if (field.getType().equals(SqlTypeName.ARRAY) || field.getType().equals(SqlTypeName.MAP)) {
        type = factory.createSqlType(SqlTypeName.ANY);
      } else if (field.getPrecision() == null && field.getScale() == null) {
        type = factory.createSqlType(field.getType());
      } else if (field.getPrecision() != null && field.getScale() == null) {
        type = factory.createSqlType(field.getType(), field.getPrecision());
      } else {
        type = factory.createSqlType(field.getType(), field.getPrecision(), field.getScale());
      }

      if (field.getIsNullable()) {
        types.add(factory.createTypeWithNullability(type, true));
      } else {
        types.add(type);
      }
    }
    return factory.createStructType(types, names);
  }
 

import org.apache.calcite.sql.type.SqlTypeFamily; //导入依赖的package包/类
@Override
public boolean checkSingleOperandType(SqlCallBinding callBinding, SqlNode node,
    int iFormalOperand, boolean throwOnFailure) {

  // check that the input is a literal.
  if(!super.checkSingleOperandType(callBinding, node, iFormalOperand, throwOnFailure)) {
    return false;
  }

  final RelDataType type = callBinding.getValidator().deriveType(callBinding.getScope(), node);
  final SqlTypeName typeName = type.getSqlTypeName();

  // Pass type checking for operators if it's of type 'ANY'.
  if (typeName.getFamily() == SqlTypeFamily.ANY) {
    return true;
  }

  if(!(typeName == SqlTypeName.CHAR || typeName == SqlTypeName.VARCHAR)) {
    if(throwOnFailure) {
      throw callBinding.newValidationSignatureError();
    }
    return false;
  }

  final SqlLiteral literal = (SqlLiteral) node;
  final String value = ((NlsString)literal.getValue()).getValue();
  if(validStrings.contains(value.toLowerCase())) {
    return true;
  }

  if(throwOnFailure) {
    throw callBinding.newValidationSignatureError();
    //throw new SqlValidatorException(String.format("DATE_PART function only accepts the following values for a date type: %s.", Joiner.on(", ").join(validStrings)), null);
  }

  return false;
}
 

import org.apache.calcite.sql.type.SqlTypeFamily; //导入依赖的package包/类
public static boolean isText(final LayoutField field) {
  final Optional<SqlTypeFamily> familyOpt = getSqlTypeFamily(field);
  if (!familyOpt.isPresent()) {
    return false;
  }

  final SqlTypeFamily family = familyOpt.get();
  return family == SqlTypeFamily.CHARACTER;
}
 

import org.apache.calcite.sql.type.SqlTypeFamily; //导入依赖的package包/类
public static boolean isNumeric(final LayoutField field) {
  final Optional<SqlTypeFamily> familyOpt = getSqlTypeFamily(field);
  if (!familyOpt.isPresent()) {
    return false;
  }

  final SqlTypeFamily family = familyOpt.get();
  return family == SqlTypeFamily.NUMERIC;
}