Shuffle机制(一)

79 阅读1分钟

1. Shuffle机制

Map方法之后,Reduce方法之前的数据处理过程称之为Shuffle

image.png

2. Partition分区

  1. 问题引出
    要求将统计结果按照条件输出到不同文件中(分区)。比如:将统计结果按照手机 归属地不同省份输出到不同文件中(分区)

  2. 默认Partitioner分区

    public class HashPartitioner<K, V> extends Partitioner<K, V> {
       public int getPartition(K key, V value, int numReduceTasks) {
           return (key.hashCode() & Integer.MAX_VALUE) % numReduceTasks;
       }
    }
    

    默认分区是根据key的hashCode对ReduceTasks个数取模得到的。用户没法控制哪个 key存储到哪个分区。

  3. 自定义Partitioner步骤

  • 自定义类继承Partitioner,重写getPartition()方法
public class CustomPartitioner extends Partitioner<Text, FlowBean> {
    @Override
    public int getPartition(Text key, FlowBean value, int numPartitions) {
        // 控制分区代码逻辑
        … …
        return partition;
    }
}
  1. 在Job驱动中,设置自定义Partitioner
job.setPartitionerClass(CustomPartitioner.class);
  1. 自定义Partition后,要根据自定义Partitioner的逻辑设置相应数量的ReduceTask
job.setNumReduceTasks(5);
  1. 分区总结
  • 如果ReduceTask的数量> getPartition的结果数,则会多产生几个空的输出文件part-r-000xx;
  • 如果1<ReduceTask的数量<getPartition的结果数,则有一部分分区数据无处安放,会Exception;
  • 如果ReduceTask的数量=1,则不管MapTask端输出多少个分区文件,最终结果都交给这一个 ReduceTask,最终也就只会产生一个结果文件part-r-00000;
  1. 案例分析 例如:假设自定义分区数为5,则
  • job.setNumReduceTasks(1); 会正常运行,只不过会产生一个输出文件
  • job.setNumReduceTasks(2); 会报错
  • job.setNumReduceTasks(6); 大于5,程序会正常运行,会产生空文件

3. WritableComparable排序

排序是MapReduce框架中最重要的操作之一。

MapTask 和ReduceTask 均会对数据按照key进行排序。该操作属于Hadoop的默认行为。任何应用程序中的数据均会被排序,而不管逻辑上是否需要。

默认排序是按照字典顺序排序,且实现该排序的方法是快速排序

对于MapTask,它会将处理的结果暂时放到环形缓冲区中,当环形缓冲区使用率达到一定阈值后,再对缓冲区中的数据进行一次快速排序,并将这些有序数据溢写到磁盘上,而当数据处理完毕后,它会对磁盘上所有文件进行归并排序。

对于ReduceTask,它从每个MapTask上远程拷贝相应的数据文件,如果文件大小超过一定阈值,则溢写磁盘上,否则存储在内存中。如果磁盘上文件数目达到一定阈值,则进行一次归并排序以生成一个更大文件;如果内存中文件大小或者数目超过一定阈值,则进行一次合并后将数据溢写到磁盘上。当所有数据拷贝完 毕后,ReduceTask统一对内存和磁盘上的所有数据进行一次归并排序。

  1. 部分排序
    MapReduce根据输入记录的键对数据集排序。保证输出的每个文件内部有序。
  2. 全排序
    最终输出结果只有一个文件,且文件内部有序。实现方式是只设置一个ReduceTask。但该方法在 处理大型文件时效率极低,因为一台机器处理所有文件,完全丧失了MapReduce所提供的并行架构。
  3. 辅助排序:(GroupingComparator分组)
    在Reduce端对key进行分组。应用于:在接收的key为bean对象时,想让一个或几个字段相同(全部 字段比较不相同)的key进入到同一个reduce方法时,可以采用分组排序。
  4. 二次排序
    在自定义排序过程中,如果compareTo中的判断条件为两个即为二次排序

自定义排序WritableComparable 原理分析

bean 对象做为key 传输,需要实现WritableComparable 接口重写compareTo 方法,就可以实现排序。

@Override
public int compareTo(FlowBean bean) {
    int result;
    // 按照总流量大小,倒序排列
    if (this.sumFlow > bean.getSumFlow()) {
        result = -1;
    }else if (this.sumFlow < bean.getSumFlow()) {
        result = 1;
    }else {
        result = 0;
    }
    return result;
}