getDAGPlan

GetDAGPlan.java

  private DAGProtos.DAGPlan getDAGPlan() throws IOException {
    String stagingDir = conf.get(TezConfiguration.TEZ_AM_STAGING_DIR,
        TezConfiguration.TEZ_AM_STAGING_DIR_DEFAULT);
    Path binaryPlan =  new Path(
        stagingDir, TezConfiguration.TEZ_PB_PLAN_BINARY_NAME);
    FileInputStream dagPBBinaryStream = null;
    DAGProtos.DAGPlan dagPlan = null;
    try {
      // Read the protobuf DAG
      dagPBBinaryStream = new FileInputStream(binaryPlan.getName());
      dagPlan = DAGProtos.DAGPlan.parseFrom(dagPBBinaryStream);
    } finally {
      if (dagPBBinaryStream != null) {
        dagPBBinaryStream.close();
      }
    }
    return dagPlan;
  }