Class SwapTablePlan

server/src/main/java/io/crate/planner/SwapTablePlan.java:45–84

import io.crate.types.DataTypes;

public class SwapTablePlan implements Plan {

    private final AnalyzedSwapTable swapTable;

    SwapTablePlan(AnalyzedSwapTable swapTable) {
        this.swapTable = swapTable;
    }

    @Override
    public StatementType type() {
        return StatementType.DDL;
    }

    @Override
    public void executeOrFail(DependencyCarrier dependencies,
                              PlannerContext plannerContext,
                              RowConsumer consumer,
                              Row params,
                              SubQueryResults subQueryResults) {
        boolean dropSource = Objects.requireNonNull(
            DataTypes.BOOLEAN.sanitizeValue(SymbolEvaluator.evaluate(
                plannerContext.transactionContext(),
                dependencies.nodeContext(),
                swapTable.dropSource(),
                params,
                subQueryResults
            )), SwapTableAnalyzer.DROP_SOURCE + " option must be true or false, not null");

        RelationName source = swapTable.source().ident();
        SwapRelationsRequest request = new SwapRelationsRequest(
            Collections.singletonList(new RelationNameSwap(source, swapTable.target().ident())),
            dropSource ? Collections.singletonList(source) : emptyList()
        );
        OneRowActionListener<AcknowledgedResponse> listener = new OneRowActionListener<>(
            consumer,
            r -> r.isAcknowledged() ? new Row1(1L) : new Row1(0L)
        );
        dependencies.client().execute(TransportSwapRelations.ACTION, request).whenComplete(listener);
    }
}