数据倾斜

为什么会数据倾斜

spark 中的数据倾斜并不是说原始数据存在倾斜,原始数据都是一个一个的 block,大小都一样,不存在数据倾斜;

而是指 shuffle 过程中产生的数据倾斜,由于不同的 key 对应的数据量不同导致不同 task 处理的数据量不同

 

注意:数据倾斜与数据过量不同,数据倾斜是某几个 task 处理的数据量很大,数据过量是所有 task 处理的数据量都很大

 

数据倾斜的表现

大部分 task 都快速执行完毕,少数 task 执行缓慢,甚至报错 OOM,即使最终运行完毕,也叫数据倾斜

 

数据倾斜的后果

1. 程序执行缓慢

2. 报错 OOM

 

定位数据倾斜问题

1. 经验:查看代码中的 shuffle 算子,如 reduceByKey、countByKey、groupByKey、join 等,根据代码逻辑判断是否会出现数据倾斜

2. spark log 文件:log 记录错误发生在代码的哪一行,从而再根据自己的理解定位哪个 shuffle 算子

3. spark web UI:

 

解决方案

聚合原数据

避免 shuffle 

1. 既然数据倾斜是 shuffle 产生的,那没有 shuffle 肯定就没有数据倾斜了;

2. 没有 shuffle 意味着在 原数据层面已经进行了 “shuffle”

这种方法适合 原数据方便实现大量数据合并操作 的数据源,确切的说就是 Hive;

比如在 Hive 中按照 key 进行分组,把每个 key 的 value 拼成一个 字符串(当然也可以是其他操作,根据需求定),那么这种数据进入 spark 后,无需 shuffle 了,直接 map 即可

 

缩小 key 粒度

比如把按 月 统计变成按 周 统计;

key 的粒度缩小,每个 key 对应的数据量降低,task 处理的数据量减少;

这样可以保证程序顺利执行,减少 OOM,但是可能引起更大的数据倾斜

 

增大 key 粒度

比如把按 周 统计变成按 月 统计;

key 的粒度增大,每个 key 对应的数据量增大,task 处理的数据量增大;

这样可以降低数据倾斜的可能,但是 可能引起 OOM

 

过滤导致倾斜的 key

极端方法,扔掉导致倾斜的 key,不建议

 

提高 shuffle 操作的 reduce 并行度

增加 reduce 并行度其实就是增加 reduce 端 task 的数量, 这样每个 task 处理的数据量减少,避免 oom

 

reduce 端并行度设置

1. 在很多 shuffle 算子中可直接指定并行度,如 reduceByKey(lambda x, y: x+y, 10)

  // 注意如果并行度大于 executor 数 x executor core 数,以小为准

2. 在 sparkSQL 中 groupBy、join 等 shuffle 操作,需要设定 spark 参数:spark.sql.shuffle.partitions

  // 该参数默认为 200

 

这样做是有缺陷的

map 端不断地写入数据,reduce task 不断地从指定位置读取数据,如果 task 很多,读取的速度增加,但是每个 key 对应的 reduce 处理的总量没变,

所以它并没有从根本上解决数据倾斜的问题,只是尽量去减少 reduce task 的数据量,适用于 较多 key 对应的数据量都很大的问题;

试想,如果只有 1 个 key 数据量较大,那么其他 key 高并行就是资源的浪费;

 

使用随机 key 进行双重聚合

双重聚合,就是聚合多次,大致思路为:

由于一个 key 对应的数据量太大,我先给这个 key 加个随机数,num_key,强行把一个 key 变成 多个 key,这样每个 key 的数据量减小,然后按  num_key 进行聚合,聚合之后,把 num_key 再转回 key,然后对 key 再次聚合;

把一次聚合变成多次聚合,如图

这种方法适合于 reduceByKey、groupByKey 等算子,不适合 join 算子

 

将 reduce join 变成 map join 

正常情况下,join 会产生 shuffle 过程,而且是 reduce join,即先将相同 key 对应的 value 汇聚到一个 reduce task 中,再进行 join,如下图

 

 

 

如果其中有一个 RDD 很小,就可以采用 广播小 RDD + map 大 RDD 实现 join 功能,

此时没有 shuffle 操作,自然不会有数据倾斜

 

注意:RDD 是不能广播的,只有将 RDD 通过 collect 拉取到 Driver 内存中才能进行广播

 

如果是两个 RDD 都很大,则不适合这种方法 

 

sample 采样对倾斜 key 单独进行 join

思考一个问题,如果 RDD 中只有一个 key,shuffle 如何做?

在 spark 中,如果只有一个 key,shuffle 过程会将 所有 value 打散,分配到不同的 reduce task 中;

 

那么如果某个 RDD 中因为单个 key 导致数据倾斜,我们可以把这个 key 拿出来形成一个新的 RDD1,其他 key 形成一个 RDD2,然后用 RDD1 再与目标 RDD 进行 join 操作,根据 spark 运行机制,他会把 RDD1 打散分配到多个 task 中进行 join 操作;

这样可以避免因 这个 key 对应的数据量太大导致数据倾斜,甚至 OOM

如果一个 RDD 中有多个 key 导致数据倾斜,则此法不适用

 

使用随机数以及扩容进行 join

如果有多个 key 导致数据倾斜,无法通过采样确定哪个 key 数据量大,只能把多个 key 都提出来,这样效率貌似不太高;

此时我们只有一种解决方案:对一个 RDD 扩容,对另一个 RDD 稀释,再进行 join

具体做法如图

注意:如果 RDD 过大,进行 随机数扩容后,可能产生 OOM

 

repartition

这个也是较常用的方法,它的本质就是减少 task 处理的数据量,一般发生在 shuffle 之前,当然它本身也是个 shuffle 操作

 

总结

数据倾斜 是 某些 task 处理了大量数据,所以数据倾斜很可能引起 OOM,数据倾斜和 OOM 的某些解决办法可以通用;

解决数据倾斜的大致思路为:

1. 避免 shuffle,可在数据输入 spark 之前进行 shuffle,或者 用 map 等替代 shuffle

  // 聚合原始数据、广播小 RDD

2. 如果避免不了 shuffle,就减少 reduce task 的数据量

  // 缩小 key 粒度、增加 reduce task 数量

  // 通过随机数多次聚合,减少每次聚合的数据量,针对 reduceByKey、groupByKey 等

3. 其他情况,单个 key 数据量大,多个 key 数据量大,针对 join