Search in sources :

Example 1 with Task

use of io.crate.execution.jobs.Task in project crate by crate.

the class JobLauncher method getHandlerBucketReceivers.

private List<PageBucketReceiver> getHandlerBucketReceivers(RootTask rootTask, List<Tuple<ExecutionPhase, RowConsumer>> handlerPhases) {
    final List<PageBucketReceiver> pageBucketReceivers = new ArrayList<>(handlerPhases.size());
    for (Tuple<ExecutionPhase, ?> handlerPhase : handlerPhases) {
        Task ctx = rootTask.getTaskOrNull(handlerPhase.v1().phaseId());
        if (ctx instanceof DownstreamRXTask) {
            PageBucketReceiver pageBucketReceiver = ((DownstreamRXTask) ctx).getBucketReceiver((byte) 0);
            pageBucketReceivers.add(pageBucketReceiver);
        }
    }
    return pageBucketReceivers;
}
Also used : Task(io.crate.execution.jobs.Task) RootTask(io.crate.execution.jobs.RootTask) DownstreamRXTask(io.crate.execution.jobs.DownstreamRXTask) PageBucketReceiver(io.crate.execution.jobs.PageBucketReceiver) ArrayList(java.util.ArrayList) ExecutionPhase(io.crate.execution.dsl.phases.ExecutionPhase) DownstreamRXTask(io.crate.execution.jobs.DownstreamRXTask)

Aggregations

ExecutionPhase (io.crate.execution.dsl.phases.ExecutionPhase)1 DownstreamRXTask (io.crate.execution.jobs.DownstreamRXTask)1 PageBucketReceiver (io.crate.execution.jobs.PageBucketReceiver)1 RootTask (io.crate.execution.jobs.RootTask)1 Task (io.crate.execution.jobs.Task)1 ArrayList (java.util.ArrayList)1