Hadoop源码分析（九）

上周我们对输入分片InputSplit进行了分析，对其分片大小进行了进一步的探究。随后展开了对与mapreduce中的核心类mapper的分析，在了解其作用的基础上，又对其几个重要的子类做了源码分析。本此我们将继续分析，首先从org.apache.hadoop.mapreduce.MarkableIterator开始。

首先附上文件的源代码：

package org.apache.hadoop.mapreduce;

import java.io.IOException;
import java.util.Iterator;

import org.apache.hadoop.classification.InterfaceAudience;
import org.apache.hadoop.classification.InterfaceStability;


@InterfaceAudience.Public
@InterfaceStability.Evolving
public class MarkableIterator 
  implements MarkableIteratorInterface {

  MarkableIteratorInterface baseIterator;

  
  public MarkableIterator(Iterator itr)  {
    if (!(itr instanceof MarkableIteratorInterface)) {
      throw new IllegalArgumentException("Input Iterator not markable");
    }
    baseIterator = (MarkableIteratorInterface) itr;
  }

  @Override
  public void mark() throws IOException {
    baseIterator.mark();
  }

  @Override
  public void reset() throws IOException {
    baseIterator.reset();
  }

  @Override
  public void clearMark() throws IOException {
    baseIterator.clearMark();
  }

  @Override
  public boolean hasNext() { 
    return baseIterator.hasNext();
  }

  @Override
  public VALUE next() {
    return baseIterator.next();
  }

  @Override
  public void remove() {
    throw new UnsupportedOperationException("Remove Not Implemented");
  }
}

可以看出，MarkableIterator是一个包装迭代器类，它实现了MarkableIteratorInterface接口。

其中：
mark()方法能够标记当前记录，对 reset 的后续调用会将迭代器倒回到此记录。
reset()方法可以在调用上一个标记之前将迭代器重置为最后一条记录。
clearMark()可以清除任何先前设置的标记。

接下来分析OutputCommitter。

Map-Reduce 框架依赖于OutputCommitter作业：
1.在初始化期间设置作业。例如，在作业初始化期间为作业创建临时输出目录。
2.作业完成后清理作业。例如，在作业完成后删除临时输出目录。
3.设置任务临时输出。
4.检查任务是否需要提交。如果任务不需要提交，这是为了避免提交过程。
5.提交任务输出。
6.放弃任务提交。

新版本 MapReduce API中，OutputCommitter 由OutputFormat 通过getOutputCommitter() 方法确定。默认为FileOutputCommitter，适用于有文件输出的MapReduce任务。若是需要，也可以实现一个新的OutputCommitter类，以对作业的完成或任务做自定义设置或清理。

OutputCommiter 部分源码如下：

public abstract class OutputCommitter extends org.apache.hadoop.mapreduce.OutputCommitter {
     public OutputCommitter() {
     }
 
     public abstract void setupJob(JobContext var1) throws IOException;
 
     
     @Deprecated
     public void cleanupJob(JobContext jobContext) throws IOException {
     }
 
     public void commitJob(JobContext jobContext) throws IOException {
         this.cleanupJob(jobContext);
     }
 
     public void abortJob(JobContext jobContext, int status) throws IOException {
         this.cleanupJob(jobContext);
     }
 
     public abstract void setupTask(TaskAttemptContext var1) throws IOException;
 
     public abstract boolean needsTaskCommit(TaskAttemptContext var1) throws IOException;
 
     public abstract void commitTask(TaskAttemptContext var1) throws IOException;
 
     public abstract void abortTask(TaskAttemptContext var1) throws IOException;

其中 setupJob在作业运行前被调用，用于初始化操作。当OutputCommitter 被设置为 FileOutputCommitter时，它会创建最终的输出目录${mapreduce.output.fileoutputformat.outputdir}，并为任务的输出创建一个临时文件夹 _temporary，作为最终输出目录的子目录。

FileOutputCommitter 中setupJob() 方法源码如下：

public void setupJob(JobContext context) throws IOException {
     if (this.hasOutputPath()) {
         Path jobAttemptPath = this.getJobAttemptPath(context);
         FileSystem fs = jobAttemptPath.getFileSystem(context.getConfiguration());
         if (!fs.mkdirs(jobAttemptPath)) {
             LOG.error("Mkdirs failed to create " + jobAttemptPath);
         }
     } else {
         LOG.warn("Output Path is null in setupJob()");
     }
 
 }

其中 jobAttemptPath 由 getJobAttemptPath(context) 获取，一层层往下查看此方法调用，最终可以看到FileOutputCommitter 创建的临时目录为：目标输出目录下的_temporary 子目录：

private static Path getPendingJobAttemptsPath(Path out) {
     return new Path(out, "_temporary");
 }

如果作业成功，则调用 commitJob() 方法。此方法会做临时文件的清理（cleanupJob()），并在最终输出目录中创建名为_SUCCESS的文件，表示Job成功执行完成。若是Job 执行失败，则被状态对象调用abortJob()，默认会调用 cleanupJob() 的方法，对临时文件进行清理。

以上提到的是Job 级别的Committer。在 Task级别，同样也有上述几种方法：

public abstract void setupTask(TaskAttemptContext var1) throws IOException;
 
 public abstract boolean needsTaskCommit(TaskAttemptContext var1) throws IOException;
 
 public abstract void commitTask(TaskAttemptContext var1) throws IOException;
 
 public abstract void abortTask(TaskAttemptContext var1) throws IOException;

其中，在 task 执行之前会调用 setupTask()，但是默认并不做任何工作。因为创建临时任务的输出路径的工作已经在setupJob() 阶段完成。方法needsTaskCommit返回是否需要task 执行提交阶段。提交阶段的工作为：将临时目录下的输出（若有）移动到最终目录。若设置为 false，则执行框架不会为任务运行分布式提交协议，也就不会执行commitTask() 或 abortTask()。当此task没有写任何输出时，FileOutputCommitter会跳过 commit （提交）阶段。

如果task成功执行，并且有输出，则会调用commitTask() 方法，（默认的实现为）将临时目录下的输出文件移动到最终目录（mapreduce.output.fileoutputformat.outputdir）。若是执行失败，则调用abortTask()，删除任务输出的临时目录及文件。

执行框架会保证一个task在有多次尝试的情况下，仅有一个task会被提交。

FileOutputCommitter 有两个方法，commitTask 和 commitJob。Apache Spark 2.0 以及更高版本使用的是 Apache Hadoop 2。
Apache Hadoop 2 使用 mapreduce.fileoutputcommitter.algorithm.version 控制 commitTask 和 commitJob 如何工作。
在 Hadoop 2 中，默认的值是 1。在这种情况下，commitTask 会将 task 的输出文件从 task 的临时目录移动到 job 的临时目录下。在所有 task 任务完成后，commitJob 将生成的数据从 job 的临时目录移动到最终的 job 目录下。这个工作在 spark 中由 driver 完成。

若是使用的是云存储（如 s3），则这个操作会消耗较长时间。会看到所有 task 已结束，但是任务仍未结束。

在设置 mapreduce.fileoutputcommitter.algorithm.version 的值为 2 后，commitTask 会将 task 生成的输出文件从 task 临时目录直接移动到 job 的最终目录。

此时，commitJob 基本无操作。

package org.apache.hadoop.mapreduce;

import java.io.IOException;

import org.apache.hadoop.classification.InterfaceAudience;
import org.apache.hadoop.classification.InterfaceStability;
import org.apache.hadoop.fs.FileSystem;


@InterfaceAudience.Public
@InterfaceStability.Stable
public abstract class OutputFormat {

  public abstract RecordWriter 
    getRecordWriter(TaskAttemptContext context
                    ) throws IOException, InterruptedException;

 
  public abstract void checkOutputSpecs(JobContext context
                                        ) throws IOException, 
                                                 InterruptedException;


  public abstract 
  OutputCommitter getOutputCommitter(TaskAttemptContext context
                                     ) throws IOException, InterruptedException;
}

public abstract class OutputFormat { 
    // 获取具体的数据写出对象
    public abstract RecordWriter getRecordWriter(TaskAttemptContext context)
     throws IOException, InterruptedException; 
    // 检查输出配置信息是否正确
    public abstract void checkOutputSpecs(JobContext context)
     throws IOException, InterruptedException; 
    // 获取输出job的提交者对象
    public abstract OutputCommitter getOutputCommitter(TaskAttemptContext context)
     throws IOException, InterruptedException; 
}

public abstract class RecordWriter {  
    // 具体的写数据的方法
    public abstract void write(K key, V value)
     throws IOException, InterruptedException;  
     // 关闭资源
    public abstract void close(TaskAttemptContext context)
     throws IOException, InterruptedException; 
}