MapReduce 性能调优：优化洗牌（shuffle）和排序阶段

优化洗牌（shuffle）和排序阶段

洗牌和排序阶段都很耗费资源。洗牌需要在map和reduce任务之间传输数据，会导致过大的网络消耗。排序和合并操作的消耗也是很显著的。这一节将介绍一系列的技术来缓解洗牌和排序阶段的消耗。

技术46 规避使用reduce

Reduce在用于连接数据集的时候将会产生大量的网络消耗。

问题

需要考虑在MapReduce规避reduce的使用。 

方案

通过将MapReduce参数setNumReduceTasks设置为0来创建一个只有map的作业。 

讨论

洗牌和排序阶段一般都是用来连接数据集。但连接操作并不一定需要洗牌和排序，正如第4章中所介绍的。满足一定条件的连接可以只在map端运行。那么就只需要只有map的作业了。设置只有map的作业的命令如下。

job.setNumReduceTasks(0);

小结

一个只有map的作业的OutputFormat是和普通作业中reduce的OutputFormat一样。如图6.39所示。

如果无法规避reduce，那么就要尽量减小它对你的作业执行时间的影响。

技术47   过滤和投影

Map到Reduce之间传输数据要通过网络，这个成本很高。 

问题

需要减少被洗牌的数据。 

方案

减少map输出的每条记录的大小，并尽可能地减少map输出的数据量。 

讨论

过滤和投影是关系运算中的概念，用以减少需要处理的数据。这些概念也可以用到MapReduce中减少map任务需要输出的数据。以下是过滤和投影的简明定义：

过滤是减少map输出的数据量。

投影是减少map输出的每条记录的大小。

以下是上述概念的演示代码：

 Text outputKey = new Text();  Text outputValue = new Text();    @Override  public void map(LongWritable key, Text value,                  OutputCollector<Text, Text> output,                  Reporter reporter) throws IOException {                        String v = value.toString();          if (!v.startsWith("10.")) {        String[] parts = StringUtils.split(v, ".", 3);         outputKey.set(parts[0]);         outputValue.set(parts[1]);         output.collect(outputKey, outputValue);     } }

小结

过滤和投影是在需要显著减少MapReduce作业运行时间时最容易的方法中的两种。

如果已经应用了这两种方法，但还需要进一步减少运行时间。那么就可以考虑combine。

技术48   使用combine

Combine可以在map阶段进行聚合操作来减少需要发送到reduce的数据。它是一个map端的优化工具，以map的输出作为输入。 

问题

需要在过滤和投影后进一步减少运行时间。 

方案

定义一个combine。在作业代码中使用setCombinerClass来调用它。 

讨论

在map输出数据到磁盘的过程中，有两个子过程：溢洒（spill）子过程，合并子过程。Combine在这两个子过程中都会被调用，如图6.40所示。为了让combine在分组数据中效率最大，可以在两个子过程调用combine之前进行初步（precursory）的排序。

与设置map类类似，作业使用setCombinClass来设置combine。

job.setCombinerClass(Combine.class);

Combine的实现必须严格遵从reduce的规格说明。这里将假定使用技术39种的map。将map的输出中的记录按照下述条件合并：第二个八进制数相同。代码如下。

  

public static class Combine implements Reducer<Text, Text, Text, Text> {            @Override      public void reduce(Text key, Iterator<Text> values,                          OutputCollector<Text,                          Text> output,                          Reporter reporter) throws IOException {                                    Text prev = null;         while (values.hasNext()) {             Text t = values.next();             if (!t.equals(prev)) {                 output.collect(key, t);             }             prev = ReflectionUtils.copy(job, t, prev);         }     } }

Combine函数必须是可分布的（distributive）。如图6.40（在前面）所示，combine要被调用多次处理多个具有相同输入键的记录。这些记录的顺序是不可预测的。可分布函数是指，不论输入数据的顺序如何，最终的结果都一样。 

小结

在MapReduce中combine非常有用，它能够减少map和reduce之间的网络传输数据和网络负载。下一个减少执行时间的有用工具就是二进制比较器。

技术49   用Comparator进行超快排序

MapReduce默认使用RawComparator对map的输出键进行比较排序。内置的Writable类（例如Text和IntWritable）是字节级实现。这样不用将字节形式的类解排列（unmarshal）成类对象。如果要通过WritableComparable实现自定义Writable，就有可能延长洗牌和排序阶段的时间，因为它需要进行解排列。 

问题

存在自定义的Writable。需要减少作业的排序时间。 

方案

实现字节级的Comparator来优化排序中的比较过程。

讨论

在MapReduce中很多阶段，排序是通过比较输出键来进行的。为了加快键排序，所有的map输出键必须实现WritableComparable接口。

 public interface WritableComparable<T> extends Writable, Comparable<T> {

 

 }

如果对4.2.1中的Person类进行改造，实现代码如下。

  

public class Person implements WritableComparable<Person> {      private String firstName;      private String lastName;            @Override      public int compareTo(Person other) {          int cmp = this.lastName.compareTo(other.lastName);          if (cmp != 0) {            return cmp;         }         return this.firstName.compareTo(other.firstName);     } ...

这个Comparator的问题在于，如果要进行比较，就需要将字节形式的map的中间结果数据解排列成Writable形式。解排列要重新创建对象，因此成本很高。

Hadoop中的自带的各种Writable类不但扩展了WritableComparable接口，也提供了基于WritableComparator类的自定义Comparator。代码如下。

  

public class WritableComparator implements RawComparator {        public int compare(byte[] b1, int s1, int l1, byte[] b2, int s2, int l2) {                try {              buffer.reset(b1, s1, l1);              key1.readFields(buffer);                buffer.reset(b2, s2, l2);             key2.readFields(buffer);         } catch (IOException e) {             throw new RuntimeException(e);         }         return compare(key1, key2);     }        /** Compare two WritableComparables.     *     * <p> The default implementation uses the natural ordering,     * calling {@link     * Comparable#compareTo(Object)}. */     @SuppressWarnings("unchecked")     public int compare(WritableComparable a, WritableComparable b) {         return a.compareTo(b);     }     ... }

要实现字节级的Comparator，需要重载compare方法。这里先学习一下IntWritable类如何实现这个方法。

 

 public class IntWritable implements WritableComparable {        public static class Comparator extends WritableComparator {               public Comparator() {              super(IntWritable.class);          }                   public int compare(byte[] b1, int s1, int l1, byte[] b2, int s2, int l2) {             int thisValue = readInt(b1, s1);             int thatValue = readInt(b2, s2);             return (thisValue<thatValue ? -1 :             (thisValue==thatValue ? 0 : 1));         }     }          static {         WritableComparator.define(IntWritable.class, new Comparator());     }

如果只使用内置的Writable，那就没有必要实现WritableComparator。它们都自带。如果需要使用自定义的Writable作为输出键，那么就需要自定义WritableComparator。这里基于前述Person类来说明如何实现。

在Person类中，有两个字符串类属性，firstName和lastName。使用writeUTF方法通过DataOutput输出它们。以下是实现代码。

 private String firstName; private String lastName;  @Override public void write(DataOutput out) throws IOException {     out.writeUTF(lastName);     out.writeUTF(firstName); }

首先需要理解Person对象是如何用字节形式表示的。writeUTF方法输出了字节长度（2个字节），字符内容（字符的长度，L1个字节）。如图6.41描述了字节是如何排列的。

假设需要对lastName和firstName进行字典式地比较（译注：就是看字典中的先后顺序）。显然不能直接用整个字节数组，因为其中还有字符长度。那么Comparator就需要足够聪明到能够跳过字符长度。以下是实现代码。

 

 @Override  public int compare(byte[] b1, int s1, int l1, byte[] b2, int s2, int l2) {        int lastNameResult = compare(b1, s1, b2, s2);      if (lastNameResult != 0) {          return lastNameResult;      }      int b1l1 = readUnsignedShort(b1, s1);      int b2l1 = readUnsignedShort(b2, s2);     return compare(b1, s1 + b1l1 + 2, b2, s2 + b2l1 + 2); }  public static int compare(byte[] b1, int s1, byte[] b2, int s2) {   int b1l1 = readUnsignedShort(b1, s1);     int b2l1 = readUnsignedShort(b2, s2);     return compareBytes(b1, s1 + 2, b1l1, b2, s2 + 2, b2l1); }  public static int readUnsignedShort(byte[] b, int offset) {     int ch1 = b[offset];     int ch2 = b[offset + 1];     return (ch1 << 8) + (ch2); }

小结

 writeUTF只支持小于65536字符的字符串类。对于人名来说，是足够了。大点的，可能就不行。这个时候就需要使用Hadoop的Text类来支持更大的字符串。Text类中的Comparator类的二进制字符串比较器的实现机制和刚才介绍的大致相当。（这个修饰真长。）那么针对Text类的lastName和firstName的Comparator的实现方式也会累死。