Examples with TableDefinition io.dingodb.common.table.TableDefinition used on opensource projects

Example 6 with TableDefinition

use of io.dingodb.common.table.TableDefinition in project dingo by dingodb.

the class DingoJobVisitor method visit.

@Override
public Collection<Output> visit(@Nonnull DingoPartition rel) {
    Collection<Output> inputs = dingo(rel.getInput()).accept(this);
    String tableName = getSimpleName(rel.getTable());
    List<Output> outputs = new LinkedList<>();
    final Map<String, Location> partLocations = Services.META.getPartLocations(tableName);
    final TableDefinition td = Services.META.getTableDefinition(tableName);
    final PartitionStrategy ps = new SimpleHashStrategy(partLocations.size());
    for (Output input : inputs) {
        Task task = input.getTask();
        PartitionOperator operator = new PartitionOperator(ps, td.getKeyMapping());
        operator.setId(idGenerator.get());
        operator.createOutputs(tableName, partLocations);
        task.putOperator(operator);
        input.setLink(operator.getInput(0));
        outputs.addAll(operator.getOutputs());
    }
    return outputs;
}

Example 7 with TableDefinition

use of io.dingodb.common.table.TableDefinition in project dingo by dingodb.

the class DingoJobVisitor method visit.

@Override
public Collection<Output> visit(@Nonnull DingoPartScan rel) {
    String tableName = getSimpleName(rel.getTable());
    TableDefinition td = Services.META.getTableDefinition(tableName);
    Map<String, Location> parts = Services.META.getPartLocations(tableName);
    List<Output> outputs = new ArrayList<>(parts.size());
    TableId tableId = new TableId(Services.META.getTableKey(tableName));
    String filterStr = null;
    if (rel.getFilter() != null) {
        filterStr = RexConverter.convert(rel.getFilter()).toString();
    }
    for (Map.Entry<String, Location> entry : parts.entrySet()) {
        final Object partId = entry.getKey();
        PartScanOperator operator = new PartScanOperator(tableId, entry.getKey(), td.getTupleSchema(), td.getKeyMapping(), filterStr, rel.getSelection());
        operator.setId(idGenerator.get());
        Task task = job.getOrCreate(entry.getValue());
        task.putOperator(operator);
        operator.getSoleOutput().setHint(OutputHint.of(tableName, partId));
        outputs.addAll(operator.getOutputs());
    }
    return outputs;
}

Example 8 with TableDefinition

use of io.dingodb.common.table.TableDefinition in project dingo by dingodb.

the class DingoGetByKeysRule method onMatch.

@Override
public void onMatch(@Nonnull RelOptRuleCall call) {
    final DingoTableScan rel = call.rel(0);
    RexNode rexNode = RexUtil.toDnf(rel.getCluster().getRexBuilder(), rel.getFilter());
    TableDefinition td = dingo(rel.getTable()).getTableDefinition();
    KeyTuplesRexVisitor visitor = new KeyTuplesRexVisitor(td);
    Set<Object[]> keyTuples = rexNode.accept(visitor);
    if (checkKeyTuples(keyTuples)) {
        call.transformTo(new DingoGetByKeys(rel.getCluster(), rel.getTraitSet().replace(DingoConventions.DISTRIBUTED), rel.getTable(), keyTuples, rel.getSelection()));
    }
}

Example 9 with TableDefinition

use of io.dingodb.common.table.TableDefinition in project dingo by dingodb.

the class DingoJobVisitor method visit.

@Override
public Collection<Output> visit(@Nonnull DingoGetByKeys rel) {
    String tableName = getSimpleName(rel.getTable());
    final Map<String, Location> partLocations = Services.META.getPartLocations(tableName);
    final TableDefinition td = Services.META.getTableDefinition(tableName);
    final PartitionStrategy ps = new SimpleHashStrategy(partLocations.size());
    final TableId tableId = new TableId(Services.META.getTableKey(tableName));
    Map<String, List<Object[]>> partMap = ps.partKeyTuples(rel.getKeyTuples());
    List<Output> outputs = new LinkedList<>();
    for (Map.Entry<String, List<Object[]>> entry : partMap.entrySet()) {
        final Object partId = entry.getKey();
        GetByKeysOperator operator = new GetByKeysOperator(tableId, partId, td.getTupleSchema(), td.getKeyMapping(), entry.getValue(), rel.getSelection());
        operator.setId(idGenerator.get());
        Task task = job.getOrCreate(partLocations.get(entry.getKey()));
        task.putOperator(operator);
        operator.getSoleOutput().setHint(OutputHint.of(tableName, partId));
        outputs.addAll(operator.getOutputs());
    }
    return outputs;
}

Example 10 with TableDefinition

use of io.dingodb.common.table.TableDefinition in project dingo by dingodb.

the class DingoPartModifyRule method checkUpdateInPart.

private static void checkUpdateInPart(@Nonnull DingoTableModify rel) {
    DingoTable table = rel.getTable().unwrap(DingoTable.class);
    assert table != null;
    TableDefinition td = table.getTableDefinition();
    List<String> updateList = rel.getUpdateColumnList();
    if (updateList != null && updateList.stream().anyMatch(c -> Objects.requireNonNull(td.getColumn(c)).isPrimary())) {
        throw new IllegalStateException("Update columns " + updateList + " contain primary columns and are not supported.");
    }
}