Examples with DAG org.apache.tez.dag.app.dag.DAG used on opensource projects

Example 41 with DAG

use of org.apache.tez.dag.app.dag.DAG in project tez by apache.

the class DAGClientHandler method getDAG.

DAG getDAG(String dagIdStr) throws TezException {
    TezDAGID dagId;
    try {
        dagId = TezDAGID.fromString(dagIdStr);
    } catch (IllegalArgumentException e) {
        throw new TezException("Bad dagId: " + dagIdStr, e);
    }
    DAG currentDAG = getCurrentDAG();
    if (currentDAG == null) {
        throw new TezException("No running dag at present");
    }
    final String currentDAGIdStr = currentDAG.getID().toString();
    if (!currentDAGIdStr.equals(dagIdStr)) {
        if (getAllDagIDs().contains(dagIdStr)) {
            if (LOG.isDebugEnabled()) {
                LOG.debug("Looking for finished dagId " + dagIdStr + " current dag is " + currentDAGIdStr);
            }
            throw new DAGNotRunningException("DAG " + dagIdStr + " Not running, current dag is " + currentDAGIdStr);
        } else {
            LOG.warn("Current DAGID : " + currentDAGIdStr + ", Looking for string (not found): " + dagIdStr + ", dagIdObj: " + dagId);
            throw new TezException("Unknown dagId: " + dagIdStr);
        }
    }
    return currentDAG;
}

Example 42 with DAG

use of org.apache.tez.dag.app.dag.DAG in project tez by apache.

the class TaskCommunicatorContextImpl method getInputVertexNames.

@Override
public Iterable<String> getInputVertexNames(String vertexName) {
    Preconditions.checkNotNull(vertexName, "VertexName cannot be null: " + vertexName);
    DAG dag = getDag();
    Vertex vertex = dag.getVertex(vertexName);
    Set<Vertex> sources = vertex.getInputVertices().keySet();
    return Iterables.transform(sources, new Function<Vertex, String>() {

        @Override
        public String apply(Vertex input) {
            return input.getName();
        }
    });
}

Example 43 with DAG

use of org.apache.tez.dag.app.dag.DAG in project tez by apache.

the class TaskCommunicatorContextImpl method getVertexRunningTaskCount.

@Override
public int getVertexRunningTaskCount(String vertexName) {
    Preconditions.checkArgument(vertexName != null, "VertexName must be specified");
    DAG dag = getDag();
    Vertex vertex = dag.getVertex(vertexName);
    return vertex.getRunningTasks();
}

Example 44 with DAG

use of org.apache.tez.dag.app.dag.DAG in project tez by apache.

the class TaskCommunicatorContextImpl method getVertexTotalTaskCount.

@Override
public int getVertexTotalTaskCount(String vertexName) {
    Preconditions.checkArgument(vertexName != null, "VertexName must be specified");
    DAG dag = getDag();
    Vertex vertex = dag.getVertex(vertexName);
    return vertex.getTotalTasks();
}

Example 45 with DAG

use of org.apache.tez.dag.app.dag.DAG in project tez by apache.

the class TaskCommunicatorContextImpl method registerForVertexStateUpdates.

@Override
public void registerForVertexStateUpdates(String vertexName, @Nullable Set<VertexState> stateSet) {
    Preconditions.checkNotNull(vertexName, "VertexName cannot be null: " + vertexName);
    DAG dag = getDag();
    dag.getStateChangeNotifier().registerForVertexUpdates(vertexName, stateSet, this);
}