Examples with DingoTableModify io.dingodb.calcite.rel.DingoTableModify used on opensource projects

Example 1 with DingoTableModify

use of io.dingodb.calcite.rel.DingoTableModify in project dingo by dingodb.

the class DingoDistributedValuesRule method onMatch.

@Override
public void onMatch(@Nonnull RelOptRuleCall call) {
    DingoTableModify modify = call.rel(0);
    Values values = call.rel(1);
    RelOptCluster cluster = modify.getCluster();
    RelOptTable table = modify.getTable();
    call.transformTo(new DingoPartModify(cluster, modify.getTraitSet().replace(DingoConventions.DISTRIBUTED), new DingoDistributedValues(cluster, values.getRowType(), values.getTuples(), values.getTraitSet().replace(DingoConventions.DISTRIBUTED), table), table, modify.getOperation(), modify.getUpdateColumnList(), modify.getSourceExpressionList()));
}

Example 2 with DingoTableModify

use of io.dingodb.calcite.rel.DingoTableModify 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.");
    }
}

Example 3 with DingoTableModify

use of io.dingodb.calcite.rel.DingoTableModify in project dingo by dingodb.

the class DingoPartModifyRule method onMatch.

@Override
public void onMatch(@Nonnull RelOptRuleCall call) {
    DingoTableModify rel = call.rel(0);
    RelNode input = rel.getInput();
    RelNode convertedInput = null;
    RelOptCluster cluster = rel.getCluster();
    switch(rel.getOperation()) {
        case INSERT:
            convertedInput = new DingoExchange(cluster, rel.getTraitSet().replace(DingoConventions.DISTRIBUTED), new DingoPartition(cluster, rel.getTraitSet().replace(DingoConventions.DISTRIBUTED), convert(input, DingoConventions.DISTRIBUTED), rel.getTable()));
            break;
        case UPDATE:
            // Only support update in part.
            checkUpdateInPart(rel);
            convertedInput = convert(input, DingoConventions.DISTRIBUTED);
            break;
        case DELETE:
            convertedInput = convert(input, DingoConventions.DISTRIBUTED);
            break;
        default:
            throw new IllegalStateException("Operation \"" + rel.getOperation() + "\" is not supported.");
    }
    call.transformTo(new DingoPartModify(cluster, rel.getTraitSet().replace(DingoConventions.DISTRIBUTED), convertedInput, rel.getTable(), rel.getOperation(), rel.getUpdateColumnList(), rel.getSourceExpressionList()));
}