org.vertexium.cypher.executionPlan.ExecutionStepWithChildren Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of vertexium-cypher Show documentation

There is a newer version: 4.10.0

package org.vertexium.cypher.executionPlan;

import org.vertexium.cypher.VertexiumCypherQueryContext;
import org.vertexium.cypher.VertexiumCypherResult;
import org.vertexium.cypher.utils.StringUtils;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.stream.Stream;

import static com.google.common.base.Preconditions.checkNotNull;

public abstract class ExecutionStepWithChildren extends DefaultExecutionStep {
    private final List childSteps = new ArrayList<>();

    public ExecutionStepWithChildren(ExecutionStep... children) {
        addChildSteps(Arrays.stream(children));
    }

    public Stream getChildSteps() {
        return childSteps.stream();
    }

    void addChildStep(ExecutionStep p) {
        checkNotNull(p);
        childSteps.add(p);
    }

    void addChildSteps(Stream steps) {
        steps.forEach(this::addChildStep);
    }

    @Override
    public VertexiumCypherResult execute(VertexiumCypherQueryContext ctx, VertexiumCypherResult source) {
        for (ExecutionStep childStep : childSteps) {
            source = childStep.execute(ctx, source);
        }
        return source;
    }

    @Override
    public String toStringFull() {
        StringBuilder result = new StringBuilder();
        result.append(toString());
        getChildSteps().forEach(child -> {
            String childString = StringUtils.indent(2, child.toStringFull());
            result.append('\n').append(childString);
        });
        return result.toString();
    }
}