149、Spark核心编程进阶之Shuffle相关

shuffle操作原理

是spark中一些特殊的算子操作会触发的一种操作
shuffle操作,会导致大量的数据在不同的机器和节点之间进行传输,因此也是spark中最复杂、最消耗性能的一种操作

我们可以通过reduceByKey操作作为一个例子,来理解shuffle操作
先看图


shuffle操作原理.png

reduceByKey算子会将上一个RDD中的每个key对应的所有value都聚合成一个value,然后生成一个新的RDD
新的RDD的元素类型就是<key,value>对的格式,每个key对应一个聚合起来的value
这里最大的问题就在于,对于上一个RDD来说,并不是一个key对应的所有value都是在一个partition中的,也更不太可能说key的所有value都在一台机器上
所以对于这种情况来说,就必须在整个集群中,将各个节点上,同一个key对应的values,统一传输到一个节点上来聚合处理
这个过程中就会发生大量的网络数据的传输

在进行一个key对应的values的聚合时
首先,上一个stage的每个map task就必须保证将自己处理的当前分区中的数据,相同的key写入一个分区文件中,可能会写多个不同的分区文件
接着下一个stage的reduce task就必须从上一个stage所有task所在的机器上,将各个task写入的多个分区文件中,找到属于自己的那个分区文件
接着将属于自己的分区数据,拉取过来,这样就可以保证每个key对应的所有values都汇聚到一个节点上去处理和聚合
这个过程就称之为shuffle

shuffle是分为shuffle write和shuffle read两个部分的,是在两个不同的stage中进行的

shuffle操作过程中进行数据排序

默认情况下,shuffle操作是不会对每个分区中的数据进行排序的

如果想要对每个分区中的数据进行排序,那么可以使用以下三种方法:

  1. 使用mapPartitions算子处理每个partition,对每个partition中的数据进行排序
  2. 使用repartitionAndSortWithinPartitions,对RDD进行重分区,在重分区的过程中同时就进行分区内数据的排序
  3. 使用sortByKey对数据进行全局排序

上述三种方法中,相对来说,mapPartitions的代价比较小,因为不需要进行额外的shuffle操作
repartitionAndSortWithinPartitions和sortByKey可能会进行额外的shuffle操作的,因此性能并不是很高

val rdd2 = rdd1.reduceByKey(_ + _)
rdd2.mapPartitions(tuples.sort)
rdd2.repartitionAndSortWithinPartitions(),重分区,重分区的过程中,就进行分区内的key的排序,重分区的原理和repartition一样
rdd2.sortByKey,直接对rdd按照key进行全局性的排序

spark中会导致shuffle操作

spark中会导致shuffle操作的有以下几种算子

  1. repartition类的操作:比如repartition、repartitionAndSortWithinPartitions、coalesce等
  2. byKey类的操作:比如reduceByKey、groupByKey、sortByKey等
  3. join类的操作:比如join、cogroup等

重分区: 一般会shuffle,因为需要在整个集群中,对之前所有的分区的数据进行随机,均匀的打乱,然后把数据放入下游新的指定数量的分区内
byKey类的操作:因为你要对一个key,进行聚合操作,那么肯定要保证集群中,所有节点上的,相同的key,一定是到同一个节点上进行处理
join类的操作:两个rdd进行join,就必须将相同join key的数据,shuffle到同一个节点上,然后进行相同key的两个rdd数据的笛卡尔乘积

所以对于上述的操作
首先第一原则,就是,能不用shuffle操作,就尽量不用shuffle操作,尽量使用不shuffle的操作
第二原则,就是,如果使用了shuffle操作,那么肯定要进行shuffle的调优,甚至是解决碰到的数据倾斜的问题

shuffle操作对性能消耗的原理

shuffle操作是spark中唯一最最消耗性能的地方
因此也就成了最需要进行性能调优的地方,最需要解决线上报错的地方,也是唯一可能出现数据倾斜的地方
因为shuffle过程中,会产生大量的磁盘IO、数据序列化和反序列化、网络IO

为了实施shuffle操作
spark中才有了stage的概念,在发生shuffle操作的算子中,进行stage的拆分
shuffle操作的前半部分,是上一个stage来进行,也称之为map task,shuffle操作的后半部分,是下一个stage来进行,也称之为reduce task
其中map task负责数据的组织,也就是将同一个key对应的value都写入同一个下游task对应的分区文件中
其中reduce task负责数据的聚合,也就是将上一个stage的task所在节点上,将属于自己的各个分区文件,都拉取过来聚合
这种模型,是参考和模拟了MapReduce的shuffle过程来的

map task会将数据先保存在内存中,如果内存不够时,就溢写到磁盘文件中去
reduce task会读取各个节点上属于自己的分区磁盘文件,到自己节点的内存中,并进行聚合

shuffle操作会消耗大量的内存,因为无论是网络传输数据之前,还是之后,都会使用大量的内存中数据结构来实施聚合操作
比如reduceByKey和aggregateByKey操作,会在map side使用内存中的数据结构进行预先聚合
其他的byKey类的操作,都是在reduce side,使用内存数据结构进行聚合
在聚合过程中,如果内存不够,只能溢写到磁盘文件中去,此时就会发生大量的磁盘IO,降低性能

此外,shuffle过程中,还会产生大量的中间文件,也就是map side写入的大量分区文件
比如Spark 1.3版本,这些中间文件会一致保留着,直到RDD不再被使用,而且被垃圾回收掉了,才会去清理中间文件
这主要是为了,如果要重新计算shuffle后的RDD,那么map side不需要重新做一次磁盘写操作
但是这种情况下,如果我们的应用程序中,一直保持着对RDD的引用,导致很长时间以后才会进行RDD垃圾回收操作
保存中间文件的目录,由spark.local.dir属性指定

内存的消耗、磁盘IO、网络数据传输(IO)

shuffle操作所有相关参数详解以及性能调优

我们可以通过对一系列的参数进行调优,来优化shuffle的性能
spark 1.5.2版本

属性名称 默认值 属性说明
spark.reducer.maxSizeInFlight 48m reduce task的buffer缓冲,代表了每个reduce task每次能够拉取的map side数据最大大小,如果内存充足,可以考虑加大大小,从而减少网络传输次数,提升性能
spark.shuffle.blockTransferService netty shuffle过程中,传输数据的方式,两种选项,netty或nio,spark 1.2开始,默认就是netty,比较简单而且性能较高,spark 1.5开始nio就是过期的了,而且spark 1.6中会去除掉
spark.shuffle.compress true 是否对map side输出的文件进行压缩,默认是启用压缩的,压缩器是由spark.io.compression.codec属性指定的,默认是snappy压缩器,该压缩器强调的是压缩速度,而不是压缩率
spark.shuffle.consolidateFiles false 默认为false,如果设置为true,那么就会合并map side输出文件,对于reduce task数量特别的情况下,可以极大减少磁盘IO开销,提升性能
spark.shuffle.file.buffer 32k map side task的内存buffer大小,写数据到磁盘文件之前,会先保存在缓冲中,如果内存充足,可以适当加大大小,从而减少map side磁盘IO次数,提升性能
spark.shuffle.io.maxRetries 3 网络传输数据过程中,如果出现了网络IO异常,重试拉取数据的次数,默认是3次,对于耗时的shuffle操作,建议加大次数,以避免full gc或者网络不通常导致的数据拉取失败,进而导致task lost,增加shuffle操作的稳定性
spark.shuffle.io.retryWait 5s 每次重试拉取数据的等待间隔,默认是5s,建议加大时长,理由同上,保证shuffle操作的稳定性
spark.shuffle.io.numConnectionsPerPeer 1 机器之间的可以重用的网络连接,主要用于在大型集群中减小网络连接的建立开销,如果一个集群的机器并不多,可以考虑增加这个值
spark.shuffle.io.preferDirectBufs true 启用堆外内存,可以避免shuffle过程的频繁gc,如果堆外内存非常紧张,则可以考虑关闭这个选项
spark.shuffle.manager sort ShuffleManager,Spark 1.5以后,有三种可选的,hash、sort和tungsten-sort,sort-based ShuffleManager会更高效实用内存,并且避免产生大量的map side磁盘文件,从Spark 1.2开始就是默认的选项,tungsten-sort与sort类似,但是内存性能更高
spark.shuffle.memoryFraction 0.2 如果spark.shuffle.spill属性为true,那么该选项生效,代表了executor内存中,用于进行shuffle reduce side聚合的内存比例,默认是20%,如果内存充足,建议调高这个比例,给reduce聚合更多内存,避免内存不足频繁读写磁盘
spark.shuffle.service.enabled false 启用外部shuffle服务,这个服务会安全地保存shuffle过程中,executor写的磁盘文件,因此executor即使挂掉也不要紧,必须配合spark.dynamicAllocation.enabled属性设置为true,才能生效,而且外部shuffle服务必须进行安装和启动,才能启用这个属性
spark.shuffle.service.port 7337 外部shuffle服务的端口号,具体解释同上
spark.shuffle.sort.bypassMergeThreshold 200 对于sort-based ShuffleManager,如果没有进行map side聚合,而且reduce task数量少于这个值,那么就不会进行排序,如果你使用sort ShuffleManager,而且不需要排序,那么可以考虑将这个值加大,直到比你指定的所有task数量都打,以避免进行额外的sort,从而提升性能
spark.shuffle.spill true 当reduce side的聚合内存使用量超过了spark.shuffle.memoryFraction指定的比例时,就进行磁盘的溢写操作
spark.shuffle.spill.compress true 同上,进行磁盘溢写时,是否进行文件压缩,使用spark.io.compression.codec属性指定的压缩器,默认是snappy,速度优先
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 206,214评论 6 481
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 88,307评论 2 382
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 152,543评论 0 341
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 55,221评论 1 279
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 64,224评论 5 371
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 49,007评论 1 284
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,313评论 3 399
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,956评论 0 259
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 43,441评论 1 300
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,925评论 2 323
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,018评论 1 333
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,685评论 4 322
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,234评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,240评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,464评论 1 261
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,467评论 2 352
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,762评论 2 345

推荐阅读更多精彩内容

  • Apache Spark 是专为大规模数据处理而设计的快速通用的计算引擎。Spark是UC Berkeley AM...
    大佛爱读书阅读 2,812评论 0 20
  • 前言 继基础篇讲解了每个Spark开发人员都必须熟知的开发调优与资源调优之后,本文作为《Spark性能优化指南》的...
    Alukar阅读 868评论 0 2
  • 原文:https://tech.meituan.com/spark-tuning-pro.html Spark性能...
    code_solve阅读 1,181评论 0 34
  • 上午体测,全凭一口仙气突破了自己1000米的记录,结束之后能很快的调整呼吸节奏,很快回复状态,这幅躯体的强度比当初...
    极客之王阅读 183评论 0 0
  • 当有一次突然吐血了,健康与日俱下了?连工作也没办法继续了? 会不会想到,终日为这份小小的工作几多犯愁几多忧的,天天...
    叶子随笔阅读 211评论 0 0