io.trino.execution.TaskState Maven / Gradle / Ivy
/*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package io.trino.execution;
import java.util.Set;
import java.util.stream.Stream;
import static com.google.common.collect.ImmutableSet.toImmutableSet;
public enum TaskState
{
/**
* Task is planned but has not been scheduled yet. A task will
* be in the planned state until, the dependencies of the task
* have begun producing output.
*/
PLANNED(false, false),
/**
* Task is running.
*/
RUNNING(false, false),
/**
* Task has finished executing and output is left to be consumed.
* In this state, there will be no new drivers, the existing drivers have finished
* and the output buffer of the task is at-least in a 'no-more-pages' state.
*/
FLUSHING(false, false),
/**
* Task has finished executing and all output has been consumed.
*/
FINISHED(true, false),
/**
* Task was canceled, but not all drivers have finishing exiting
*/
CANCELING(false, true),
/**
* Task was canceled by a user.
*/
CANCELED(true, false),
/**
* Task was told to abort, but not all drivers have finishing exiting
*/
ABORTING(false, true),
/**
* Task was aborted due to a failure in the query. The failure
* was not in this task.
*/
ABORTED(true, false),
/**
* Task has been marked as failed, but not all drivers have finishing exiting
*/
FAILING(false, true),
/**
* Task execution failed.
*/
FAILED(true, false);
public static final Set TERMINAL_TASK_STATES = Stream.of(TaskState.values()).filter(TaskState::isDone).collect(toImmutableSet());
private final boolean doneState;
private final boolean terminating;
TaskState(boolean doneState, boolean terminating)
{
this.doneState = doneState;
this.terminating = terminating;
}
/**
* Is this a terminal state.
*/
public boolean isDone()
{
return doneState;
}
public boolean isTerminating()
{
return terminating;
}
public boolean isTerminatingOrDone()
{
return terminating | doneState;
}
}