java - Hadoop - 如何从 mapred.JobConf 中提取 taskId？

Question

是否可以从 *mapred*.JobConf 创建一个有效的 *mapreduce*.TaskAttemptID？

背景

我需要为 ExistingFileInputFormat 编写一个 FileInputFormatAdapter。问题是适配器需要扩展 mapred.InputFormat 而现有格式扩展 mapreduce.InputFormat。

我需要构建一个 mapreduce.TaskAttemptContextImpl，以便我可以实例化 ExistingRecordReader。但是，我无法创建有效的 TaskId...taskId 为 null。

那么如何从 mapred.JobConf 获取 taskId、jobId 等。

特别是在适配器的 getRecordReader 中，我需要做类似的事情:

public org.apache.hadoop.mapred.RecordReader<NullWritable, MyWritable> getRecordReader(
        org.apache.hadoop.mapred.InputSplit split, JobConf job, Reporter reporter) throws IOException {

    SplitAdapter splitAdapter = (SplitAdapter) split;

    final Configuration conf = job;

    /*************************************************/
    //The problem is here, "mapred.task.id" is not in the conf
    /*************************************************/
    final TaskAttemptID taskId = TaskAttemptID.forName(conf.get("mapred.task.id"));

    final TaskAttemptContext context = new TaskAttemptContextImpl(conf, taskId);
    try {
        return new RecordReaderAdapter(new ExistingRecordReader(
                splitAdapter.getMapRedeuceSplit(),
                context));
    } catch (InterruptedException e) {
        throw new RuntimeException("Failed to create record-reader.", e);
    }
}

此代码抛出异常:

Caused by: java.lang.NullPointerException
    at org.apache.hadoop.mapreduce.task.TaskAttemptContextImpl.<init>(TaskAttemptContextImpl.java:44)
    at org.apache.hadoop.mapreduce.task.TaskAttemptContextImpl.<init>(TaskAttemptContextImpl.java:39)

'super(conf, taskId.getJobID());'正在抛出异常，很可能是因为 taskId 为 null。

Answer 1

我通过查看 HiveHbaseTableInputFormat 找到了答案。由于我的解决方案是针对配置单元的，因此效果很好。

 TaskAttemptContext tac = ShimLoader.getHadoopShims().newTaskAttemptContext(
        job.getConfiguration(), reporter);