Examples with TaskGraphWalkerContext org.apache.hadoop.hive.ql.lib.TaskGraphWalker.TaskGraphWalkerContext used on opensource projects

Example 1 with TaskGraphWalkerContext

use of org.apache.hadoop.hive.ql.lib.TaskGraphWalker.TaskGraphWalkerContext in project hive by apache.

the class AbstractJoinTaskDispatcher method dispatch.

@Override
public Object dispatch(Node nd, Stack<Node> stack, Object... nodeOutputs) throws SemanticException {
    if (nodeOutputs == null || nodeOutputs.length == 0) {
        throw new SemanticException("No Dispatch Context");
    }
    TaskGraphWalkerContext walkerCtx = (TaskGraphWalkerContext) nodeOutputs[0];
    Task<? extends Serializable> currTask = (Task<? extends Serializable>) nd;
    // not map reduce task or not conditional task, just skip
    if (currTask.isMapRedTask()) {
        if (currTask instanceof ConditionalTask) {
            // get the list of task
            List<Task<? extends Serializable>> taskList = ((ConditionalTask) currTask).getListTasks();
            for (Task<? extends Serializable> tsk : taskList) {
                if (tsk.isMapRedTask()) {
                    Task<? extends Serializable> newTask = this.processCurrentTask((MapRedTask) tsk, ((ConditionalTask) currTask), physicalContext.getContext());
                    walkerCtx.addToDispatchList(newTask);
                }
            }
        } else {
            Task<? extends Serializable> newTask = this.processCurrentTask((MapRedTask) currTask, null, physicalContext.getContext());
            walkerCtx.addToDispatchList(newTask);
        }
    }
    return null;
}