org.apache.calcite.sql2rel.InitializerContext Java Examples

@SuppressWarnings("deprecation")
public boolean columnHasDefaultValue(RelDataType rowType, int ordinal,
    InitializerContext initializerContext) {
  // This method is no longer used
  final Table table = this.unwrap(Table.class);
  if (table != null && table instanceof Wrapper) {
    final InitializerExpressionFactory initializerExpressionFactory =
        ((Wrapper) table).unwrap(InitializerExpressionFactory.class);
    if (initializerExpressionFactory != null) {
      return initializerExpressionFactory
          .newColumnDefaultValue(this, ordinal, initializerContext)
          .getType().getSqlTypeName() != SqlTypeName.NULL;
    }
  }
  if (ordinal >= rowType.getFieldList().size()) {
    return true;
  }
  return !rowType.getFieldList().get(ordinal).getType().isNullable();
}
 

@Override public RexNode newColumnDefaultValue(RelOptTable table,
    int iColumn, InitializerContext context) {
  final RexBuilder rexBuilder = context.getRexBuilder();
  final RelDataTypeFactory typeFactory = rexBuilder.getTypeFactory();
  switch (iColumn) {
  case 0:
    return rexBuilder.makeExactLiteral(new BigDecimal(123),
        typeFactory.createSqlType(SqlTypeName.INTEGER));
  case 1:
    return rexBuilder.makeLiteral("Bob");
  case 5:
    return rexBuilder.makeExactLiteral(new BigDecimal(555),
        typeFactory.createSqlType(SqlTypeName.INTEGER));
  default:
    return super.newColumnDefaultValue(table, iColumn, context);
  }
}
 

@Override public RexNode newColumnDefaultValue(RelOptTable table,
    int iColumn, InitializerContext context) {
  final ModifiableViewTable viewTable = table.unwrap(ModifiableViewTable.class);
  assert iColumn < viewTable.columnMapping.size();
  final RexBuilder rexBuilder = context.getRexBuilder();
  final RelDataTypeFactory typeFactory = rexBuilder.getTypeFactory();
  final RelDataType viewType = viewTable.getRowType(typeFactory);
  final RelDataType iType = viewType.getFieldList().get(iColumn).getType();

  // Use the view constraint to generate the default value if the column is constrained.
  final int mappedOrdinal = viewTable.columnMapping.get(iColumn);
  final RexNode viewConstraint = projectMap.get(mappedOrdinal);
  if (viewConstraint != null) {
    return rexBuilder.ensureType(iType, viewConstraint, true);
  }

  // Otherwise use the default value of the underlying table.
  final Table schemaTable = viewTable.getTable();
  if (schemaTable instanceof Wrapper) {
    final InitializerExpressionFactory initializerExpressionFactory =
        ((Wrapper) schemaTable).unwrap(InitializerExpressionFactory.class);
    if (initializerExpressionFactory != null) {
      final RexNode tableConstraint =
          initializerExpressionFactory.newColumnDefaultValue(table, iColumn,
              context);
      return rexBuilder.ensureType(iType, tableConstraint, true);
    }
  }

  // Otherwise Sql type of NULL.
  return super.newColumnDefaultValue(table, iColumn, context);
}
 

@Override public RexNode newColumnDefaultValue(RelOptTable table,
    int iColumn, InitializerContext context) {
  final ModifiableViewTable viewTable = table.unwrap(ModifiableViewTable.class);
  assert iColumn < viewTable.columnMapping.size();
  final RexBuilder rexBuilder = context.getRexBuilder();
  final RelDataTypeFactory typeFactory = rexBuilder.getTypeFactory();
  final RelDataType viewType = viewTable.getRowType(typeFactory);
  final RelDataType iType = viewType.getFieldList().get(iColumn).getType();

  // Use the view constraint to generate the default value if the column is constrained.
  final int mappedOrdinal = viewTable.columnMapping.get(iColumn);
  final RexNode viewConstraint = projectMap.get(mappedOrdinal);
  if (viewConstraint != null) {
    return rexBuilder.ensureType(iType, viewConstraint, true);
  }

  // Otherwise use the default value of the underlying table.
  final Table schemaTable = viewTable.getTable();
  if (schemaTable instanceof Wrapper) {
    final InitializerExpressionFactory initializerExpressionFactory =
        ((Wrapper) schemaTable).unwrap(InitializerExpressionFactory.class);
    if (initializerExpressionFactory != null) {
      final RexNode tableConstraint =
          initializerExpressionFactory.newColumnDefaultValue(table, iColumn,
              context);
      return rexBuilder.ensureType(iType, tableConstraint, true);
    }
  }

  // Otherwise Sql type of NULL.
  return super.newColumnDefaultValue(table, iColumn, context);
}
 

@Override public RexNode newColumnDefaultValue(RelOptTable table,
    int iColumn, InitializerContext context) {
  THREAD_CALL_COUNT.get().incrementAndGet();
  final RelDataTypeField field =
      table.getRowType().getFieldList().get(iColumn);
  if (defaultColumns.contains(field.getName())) {
    final RexBuilder rexBuilder = context.getRexBuilder();
    return rexBuilder.makeExactLiteral(BigDecimal.ONE);
  }
  return super.newColumnDefaultValue(table, iColumn, context);
}
 

@Override public RexNode newAttributeInitializer(RelDataType type,
    SqlFunction constructor, int iAttribute,
    List<RexNode> constructorArgs, InitializerContext context) {
  THREAD_CALL_COUNT.get().incrementAndGet();
  return super.newAttributeInitializer(type, constructor, iAttribute,
     constructorArgs, context);
}
 

@Override public RexNode newColumnDefaultValue(
    RelOptTable table, int iColumn, InitializerContext context) {
  if (iColumn == 4) {
    final SqlNode node = context.parseExpression(SqlParser.Config.DEFAULT, "A + 1");
    // Actually we should validate the node with physical schema,
    // here full table schema(includes the virtual columns) also works
    // because the expression "A + 1" does not reference any virtual column.
    final SqlNode validated = context.validateExpression(table.getRowType(), node);
    return context.convertExpression(validated);
  } else {
    return super.newColumnDefaultValue(table, iColumn, context);
  }
}
 

@Override public RexNode newAttributeInitializer(RelDataType type,
    SqlFunction constructor, int iAttribute, List<RexNode> constructorArgs,
    InitializerContext context) {
  throw new UnsupportedOperationException("Not implemented - unknown requirements");
}
 

@Override
public boolean columnHasDefaultValue(RelDataType rowType, int ordinal, InitializerContext initializerContext) {
  return false;
}
 

@Override
public boolean columnHasDefaultValue(RelDataType rowType, int ordinal, InitializerContext initializerContext) {
  return false;
}
 

/**
 * Returns whether the ordinal column has a default value.
 */
@Deprecated // to be removed before 2.0
boolean columnHasDefaultValue(RelDataType rowType, int ordinal,
    InitializerContext initializerContext);
 

@Override public RexNode newAttributeInitializer(RelDataType type,
    SqlFunction constructor, int iAttribute, List<RexNode> constructorArgs,
    InitializerContext context) {
  throw new UnsupportedOperationException("Not implemented - unknown requirements");
}
 

/**
 * Obtains whether the ordinal column has a default value, which is not supported now.
 *
 * @param rowType            Row type of field
 * @param ordinal            Index of the given column
 * @param initializerContext Context for
 *                           {@link org.apache.calcite.sql2rel.InitializerExpressionFactory}
 * @return true if the column has a default value
 */
public boolean columnHasDefaultValue(RelDataType rowType, int ordinal, InitializerContext initializerContext) {
	return false;
}