Java Code Examples for org.apache.calcite.rel.logical.LogicalTableScan#create()

The following examples show how to use org.apache.calcite.rel.logical.LogicalTableScan#create() . You can vote up the ones you like or vote down the ones you don't like, and go to the original project or source file by following the links above each example. You may check out the related API usage on the sidebar.

Example 1

Source File: MockCatalogReader.java From calcite with Apache License 2.0

6 votes

@Override public RelNode toRel(ToRelContext context) {
  RelNode rel = LogicalTableScan.create(context.getCluster(), fromTable,
      context.getTableHints());
  final RexBuilder rexBuilder = context.getCluster().getRexBuilder();
  rel = LogicalFilter.create(
      rel, getConstraint(rexBuilder, rel.getRowType()));
  final List<RelDataTypeField> fieldList =
      rel.getRowType().getFieldList();
  final List<Pair<RexNode, String>> projects =
      new AbstractList<Pair<RexNode, String>>() {
        @Override public Pair<RexNode, String> get(int index) {
          return RexInputRef.of2(mapping.get(index), fieldList);
        }

        @Override public int size() {
          return mapping.size();
        }
      };
  return LogicalProject.create(rel,
      ImmutableList.of(),
      Pair.left(projects),
      Pair.right(projects));
}

Example 2

Source File: StreamRules.java From calcite with Apache License 2.0

6 votes

@Override public void onMatch(RelOptRuleCall call) {
  final Delta delta = call.rel(0);
  final TableScan scan = call.rel(1);
  final RelOptCluster cluster = delta.getCluster();
  final RelOptTable relOptTable = scan.getTable();
  final StreamableTable streamableTable =
      relOptTable.unwrap(StreamableTable.class);
  if (streamableTable != null) {
    final Table table1 = streamableTable.stream();
    final RelOptTable relOptTable2 =
        RelOptTableImpl.create(relOptTable.getRelOptSchema(),
            relOptTable.getRowType(), table1,
            ImmutableList.<String>builder()
                .addAll(relOptTable.getQualifiedName())
                .add("(STREAM)").build());
    final LogicalTableScan newScan =
        LogicalTableScan.create(cluster, relOptTable2, scan.getHints());
    call.transformTo(newScan);
  }
}

Example 3

Source File: QueryOperationConverter.java From flink with Apache License 2.0

6 votes

private RelNode convertToDataStreamScan(DataStreamQueryOperation<?> operation) {
	DataStreamTable<?> dataStreamTable = new DataStreamTable<>(
			operation.getDataStream(),
			operation.isProducesUpdates(),
			operation.isAccRetract(),
			operation.getFieldIndices(),
			operation.getTableSchema().getFieldNames(),
			operation.getStatistic(),
			scala.Option.apply(operation.getFieldNullables()));

	List<String> names;
	if (operation.getQualifiedName() != null) {
		names = operation.getQualifiedName();
	} else {
		String refId = String.format("Unregistered_DataStream_%s", operation.getDataStream().getId());
		names = Collections.singletonList(refId);
	}

	FlinkRelOptTable table = FlinkRelOptTable.create(
			relBuilder.getRelOptSchema(),
			dataStreamTable.getRowType(relBuilder.getTypeFactory()),
			names,
			dataStreamTable);
	return LogicalTableScan.create(relBuilder.getCluster(), table);
}

Example 4

Source File: QueryOperationConverter.java From flink with Apache License 2.0

6 votes

private RelNode convertToDataStreamScan(DataStream<?> dataStream, int[] fieldIndices, TableSchema tableSchema) {
	DataStreamTable<?> dataStreamTable = new DataStreamTable<>(
		dataStream,
		false,
		false,
		fieldIndices,
		tableSchema.getFieldNames(),
		FlinkStatistic.UNKNOWN(),
		scala.Option.empty());

	String refId = String.format("Unregistered_DataStream_%s", dataStream.getId());
	FlinkRelOptTable table = FlinkRelOptTable.create(
		relBuilder.getRelOptSchema(),
		dataStreamTable.getRowType(relBuilder.getTypeFactory()),
		Collections.singletonList(refId),
		dataStreamTable);
	return LogicalTableScan.create(relBuilder.getCluster(), table);
}

Example 5

Source File: PushDownLogicTableRule.java From Mycat2 with GNU General Public License v3.0

6 votes

@NotNull
private RelNode global(RelOptCluster cluster,
                       Bindables.BindableTableScan bindableTableScan,
                       RelOptSchema relOptSchema,
                       MycatLogicTable logicTable) {
    final HashSet<String> context = new HashSet<>();
    RelNode logicalTableScan;
    MycatPhysicalTable mycatPhysicalTable = logicTable.getMycatGlobalPhysicalTable(context);
    RelOptTable dataNode = RelOptTableImpl.create(
            relOptSchema,
            logicTable.getRowType(cluster.getTypeFactory()),//这里使用logicTable,避免类型不一致
            mycatPhysicalTable,
            ImmutableList.of(mycatPhysicalTable.getBackendTableInfo().getUniqueName()));
    logicalTableScan = LogicalTableScan.create(cluster, dataNode, ImmutableList.of());
    return RelOptUtil.createProject(RelOptUtil.createFilter(logicalTableScan, bindableTableScan.filters), bindableTableScan.projects);
}

Example 6

Source File: QueryableRelBuilder.java From calcite with Apache License 2.0

6 votes

RelNode toRel(Queryable<T> queryable) {
  if (queryable instanceof QueryableDefaults.Replayable) {
    //noinspection unchecked
    ((QueryableDefaults.Replayable) queryable).replay(this);
    return rel;
  }
  if (queryable instanceof AbstractTableQueryable) {
    final AbstractTableQueryable tableQueryable =
        (AbstractTableQueryable) queryable;
    final QueryableTable table = tableQueryable.table;
    final CalciteSchema.TableEntry tableEntry =
        CalciteSchema.from(tableQueryable.schema)
            .add(tableQueryable.tableName, tableQueryable.table);
    final RelOptTableImpl relOptTable =
        RelOptTableImpl.create(null, table.getRowType(translator.typeFactory),
            tableEntry, null);
    if (table instanceof TranslatableTable) {
      return ((TranslatableTable) table).toRel(translator.toRelContext(),
          relOptTable);
    } else {
      return LogicalTableScan.create(translator.cluster, relOptTable, ImmutableList.of());
    }
  }
  return translator.translate(queryable.getExpression());
}

Example 7

Source File: StreamRules.java From Bats with Apache License 2.0

6 votes

@Override public void onMatch(RelOptRuleCall call) {
  final Delta delta = call.rel(0);
  final TableScan scan = call.rel(1);
  final RelOptCluster cluster = delta.getCluster();
  final RelOptTable relOptTable = scan.getTable();
  final StreamableTable streamableTable =
      relOptTable.unwrap(StreamableTable.class);
  if (streamableTable != null) {
    final Table table1 = streamableTable.stream();
    final RelOptTable relOptTable2 =
        RelOptTableImpl.create(relOptTable.getRelOptSchema(),
            relOptTable.getRowType(), table1,
            ImmutableList.<String>builder()
                .addAll(relOptTable.getQualifiedName())
                .add("(STREAM)").build());
    final LogicalTableScan newScan =
        LogicalTableScan.create(cluster, relOptTable2);
    call.transformTo(newScan);
  }
}

Example 8

Source File: CrelUniqifier.java From dremio-oss with Apache License 2.0

5 votes

@Override
public RelNode visit(RelNode other) {
  if(!data.add(other)) {
    if (other instanceof LogicalTableScan) {
      // LogicalTableScan does not have implementation of a deep copy. Create a new instance.
      other = LogicalTableScan.create(other.getCluster(), other.getTable());
    } else {
      other = other.copy(other.getTraitSet(), other.getInputs());
    }
  }

  return super.visit(other);
}

Example 9

Source File: DruidTable.java From calcite with Apache License 2.0

5 votes

public RelNode toRel(RelOptTable.ToRelContext context,
    RelOptTable relOptTable) {
  final RelOptCluster cluster = context.getCluster();
  final TableScan scan = LogicalTableScan.create(cluster, relOptTable, ImmutableList.of());
  return DruidQuery.create(cluster,
      cluster.traitSetOf(BindableConvention.INSTANCE), relOptTable, this,
      ImmutableList.of(scan));
}

Example 10

Source File: QueryOperationConverter.java From flink with Apache License 2.0

5 votes

private RelNode convertToDataStreamScan(
		DataStream<?> dataStream,
		int[] fieldIndices,
		TableSchema tableSchema,
		Optional<ObjectIdentifier> identifier) {
	List<String> names;
	if (identifier.isPresent()) {
		names = Arrays.asList(
			identifier.get().getCatalogName(),
			identifier.get().getDatabaseName(),
			identifier.get().getObjectName());
	} else {
		String refId = String.format("Unregistered_DataStream_%s", dataStream.getId());
		names = Collections.singletonList(refId);
	}
		final RelDataType rowType = DataStreamTable$.MODULE$
			.getRowType(relBuilder.getTypeFactory(),
				dataStream,
				tableSchema.getFieldNames(),
				fieldIndices,
				scala.Option.empty());
	DataStreamTable<?> dataStreamTable = new DataStreamTable<>(
		relBuilder.getRelOptSchema(),
		names,
		rowType,
		dataStream,
		fieldIndices,
		tableSchema.getFieldNames(),
		FlinkStatistic.UNKNOWN(),
		scala.Option.empty());
	return LogicalTableScan.create(relBuilder.getCluster(), dataStreamTable);
}

Example 11

Source File: QueryOperationConverter.java From flink with Apache License 2.0

5 votes

private RelNode convertToDataStreamScan(DataStreamQueryOperation<?> operation) {
	List<String> names;
	ObjectIdentifier identifier = operation.getIdentifier();
	if (identifier != null) {
		names = Arrays.asList(
			identifier.getCatalogName(),
			identifier.getDatabaseName(),
			identifier.getObjectName());
	} else {
		String refId = String.format("Unregistered_DataStream_%s", operation.getDataStream().getId());
		names = Collections.singletonList(refId);
	}

	final RelDataType rowType = DataStreamTable$.MODULE$
		.getRowType(relBuilder.getTypeFactory(),
			operation.getDataStream(),
			operation.getTableSchema().getFieldNames(),
			operation.getFieldIndices(),
			scala.Option.apply(operation.getFieldNullables()));
	DataStreamTable<?> dataStreamTable = new DataStreamTable<>(
		relBuilder.getRelOptSchema(),
		names,
		rowType,
		operation.getDataStream(),
		operation.getFieldIndices(),
		operation.getTableSchema().getFieldNames(),
		operation.getStatistic(),
		scala.Option.apply(operation.getFieldNullables()));
	return LogicalTableScan.create(relBuilder.getCluster(), dataStreamTable);
}

Example 12

Source File: PushDownLogicTableRule.java From Mycat2 with GNU General Public License v3.0

5 votes

@NotNull
private static RelNode getBindableTableScan(Bindables.BindableTableScan bindableTableScan, RelOptCluster cluster, RelOptSchema relOptSchema, DataNode backendTableInfo) {
    String uniqueName = backendTableInfo.getUniqueName();
    MycatLogicTable unwrap = bindableTableScan.getTable().unwrap(MycatLogicTable.class);
    MycatPhysicalTable mycatPhysicalTable = new MycatPhysicalTable(unwrap,backendTableInfo);
    RelOptTable dataNode = RelOptTableImpl.create(
            relOptSchema,
            mycatPhysicalTable.getRowType(cluster.getTypeFactory()),
            mycatPhysicalTable,
            ImmutableList.of(uniqueName));
    RelNode logicalTableScan = LogicalTableScan.create(cluster, dataNode, ImmutableList.of());
    return RelOptUtil.createProject(RelOptUtil.createFilter(logicalTableScan, bindableTableScan.filters), bindableTableScan.projects);
}

Example 13

Source File: QueryOperationConverter.java From flink with Apache License 2.0

5 votes

@Override
public <U> RelNode visit(TableSourceQueryOperation<U> tableSourceOperation) {
	TableSource<?> tableSource = tableSourceOperation.getTableSource();
	boolean isBatch;
	if (tableSource instanceof LookupableTableSource) {
		isBatch = tableSourceOperation.isBatch();
	} else if (tableSource instanceof StreamTableSource) {
		isBatch = ((StreamTableSource<?>) tableSource).isBounded();
	} else {
		throw new TableException(String.format("%s is not supported.", tableSource.getClass().getSimpleName()));
	}

	FlinkStatistic statistic;
	List<String> names;
	if (tableSourceOperation instanceof RichTableSourceQueryOperation &&
		((RichTableSourceQueryOperation<U>) tableSourceOperation).getQualifiedName() != null) {
		statistic = ((RichTableSourceQueryOperation<U>) tableSourceOperation).getStatistic();
		names = ((RichTableSourceQueryOperation<U>) tableSourceOperation).getQualifiedName();
	} else {
		statistic = FlinkStatistic.UNKNOWN();
		// TableSourceScan requires a unique name of a Table for computing a digest.
		// We are using the identity hash of the TableSource object.
		String refId = "Unregistered_TableSource_" + System.identityHashCode(tableSource);
		names = Collections.singletonList(refId);
	}

	TableSourceTable<?> tableSourceTable = new TableSourceTable<>(tableSource, !isBatch, statistic);
	FlinkRelOptTable table = FlinkRelOptTable.create(
		relBuilder.getRelOptSchema(),
		tableSourceTable.getRowType(relBuilder.getTypeFactory()),
		names,
		tableSourceTable);
	return LogicalTableScan.create(relBuilder.getCluster(), table);
}

Example 14

Source File: SqlToRelTestBase.java From calcite with Apache License 2.0

4 votes

public RelNode toRel(ToRelContext context) {
  return LogicalTableScan.create(context.getCluster(), this,
      context.getTableHints());
}

Example 15

Source File: MockCatalogReader.java From calcite with Apache License 2.0

4 votes

public RelNode toRel(ToRelContext context) {
  return LogicalTableScan.create(context.getCluster(), this, context.getTableHints());
}

Example 16

Source File: MycatLogicTable.java From Mycat2 with GNU General Public License v3.0

4 votes

@Override
public RelNode toRel(RelOptTable.ToRelContext context, RelOptTable relOptTable) {
    return LogicalTableScan.create(context.getCluster(), relOptTable,ImmutableList.of());
}

Example 17

Source File: MycatPhysicalTable.java From Mycat2 with GNU General Public License v3.0

4 votes

@Override
public RelNode toRel(RelOptTable.ToRelContext context, RelOptTable relOptTable) {
    return LogicalTableScan.create(context.getCluster(), relOptTable, ImmutableList.of());
}

Example 18

Source File: FlinkPreparingTableBase.java From flink with Apache License 2.0

4 votes

@Override
public RelNode toRel(RelOptTable.ToRelContext context) {
	return LogicalTableScan.create(context.getCluster(), this);
}

Example 19

Source File: RelOptAbstractTable.java From Bats with Apache License 2.0

4 votes

public RelNode toRel(ToRelContext context) {
  return LogicalTableScan.create(context.getCluster(), this);
}

Example 20

Source File: MockCatalogReader.java From calcite with Apache License 2.0

4 votes

@Override public RelNode toRel(RelOptTable.ToRelContext context, RelOptTable relOptTable) {
  return LogicalTableScan.create(context.getCluster(), relOptTable, context.getTableHints());
}