前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >SparkSql中多个Stage的并发执行

SparkSql中多个Stage的并发执行

作者头像
数据仓库践行者
发布2022-11-25 19:36:37
1.4K0
发布2022-11-25 19:36:37
举报
文章被收录于专栏:数据仓库践行者

写一篇水水的技术文,总结一下sparksql中不同stage的并行执行相关,也是来自于一位群友的提问:

我们群里有很多技术很棒并且很热心的大佬,哈哈~

Hive中Job并发执行

hive中,同一sql里,如果涉及到多个job,默认情况下,每个job是顺序执行的。

但如果某些job没有前后依赖关系的话,是阔以并行执行的,这样可能使得整个job的执行时间缩短。

可以通过设置参数 set hive.exec.parallel=true,实现job并发执行,该参数默认可以并发执行的job数为8,相关参数如下:

代码语言:javascript
复制
set hive.exec.parallel=true;              //打开任务并行执行
set hive.exec.parallel.thread.number=16;  //同一个sql允许最大并行度,默认为8。

Spark中多个Stage的并发执行

先给结论:

  • 没有相互依赖关系的Stage是可以并行执行的,比如union all 两侧的sql
  • 存在依赖的Stage必须在依赖的Stage执行完成后才能执行下一个Stage
  • Stage的并行度取决于资源数(配制的参数以及队列资源的核数、内存等),相关参数如下:
代码语言:javascript
复制
set spark.dynamicAllocation.enabled=true; --动态资源开关
set spark.executor.cores=4; --每个Container分配的core数量
set spark.dynamicAllocation.maxExecutors=1200;  --executor最大申请个数
set spark.dynamicAllocation.initialExecutors=5;  -- 初始申请executor个数,默认等于minExecutors
set spark.dynamicAllocation.minExecutors =5; --executor最小申请个数

大概就是这几个参数,代表能并行4*1200 ,也就是4800个task。
把maxExecutors调大点,就能并行的更多
代码语言:javascript
复制

源码角度的解释

如果一个Stage有多个依赖,会依次递归(按stage id从小到大排列,也就是stage是从后往前提交的)提交父stages,直到到了根节点,如果有多个根节点,都会通过submitMissingTasks 提交上去运行。

提交stage:

代码语言:javascript
复制
/** Submits stage, but first recursively submits any missing parents. */
  private def submitStage(stage: Stage): Unit = {
   //获取stage所属的active的JobId
    val jobId = activeJobForStage(stage)
    if (jobId.isDefined) {
      logDebug(s"submitStage($stage (name=${stage.name};" +
        s"jobs=${stage.jobIds.toSeq.sorted.mkString(",")}))")
      //非waiting、running、failed的stage
      if (!waitingStages(stage) && !runningStages(stage) && !failedStages(stage)) {
        //获取该stage未提交的父stages,并按stage id从小到大排序,也就是stage是从后往前提交的
        val missing = getMissingParentStages(stage).sortBy(_.id)
        logDebug("missing: " + missing)
        if (missing.isEmpty) {
          logInfo("Submitting " + stage + " (" + stage.rdd + "), which has no missing parents")
          //若无未提交的父stage, 则提交该stage对应的tasks
          submitMissingTasks(stage, jobId.get)
        } else {
          //若存在未提交的父stage, 依次提交所有父stage (若父stage也存在未提交的父stage, 则提交, 依次类推)
          for (parent <- missing) {
            submitStage(parent)
          }
          //并把该stage添加到等待stage队列中
          waitingStages += stage
        }
      }
    } else {
      abortStage(stage, "No active job for stage " + stage.id, None)
    }
  }

获取未提交的父stages:

代码语言:javascript
复制
//以参数stage为起点,向前遍历所有stage,判断stage是否为未提交,若使则加入missing中
private def getMissingParentStages(stage: Stage): List[Stage] = {
  val missing = new HashSet[Stage] //未提交的stage
  val visited = new HashSet[RDD[_]]  //存储已经被访问到得RDD
  // We are manually maintaining a stack here to prevent StackOverflowError
  // caused by recursively visiting
  val waitingForVisit = new ListBuffer[RDD[_]]
  waitingForVisit += stage.rdd
  def visit(rdd: RDD[_]): Unit = {
    if (!visited(rdd)) {
      visited += rdd
      val rddHasUncachedPartitions = getCacheLocs(rdd).contains(Nil)
      if (rddHasUncachedPartitions) {
        for (dep <- rdd.dependencies) {
          dep match {
             //若为宽依赖,生成新的stage
            case shufDep: ShuffleDependency[_, _, _] =>
            //根据shufDep.shuffleId获取对应的ShuffleMapStage
              val mapStage = getOrCreateShuffleMapStage(shufDep, stage.firstJobId)
              if (!mapStage.isAvailable || !mapStage.shuffleDep.shuffleMergeFinalized) {
                  //若stage得状态为available或者 基于push based shuffle合并没有完成,则为未提交stage
                missing += mapStage
              } else {
                // Forward the nextAttemptId if skipped and get visited for the first time.
                // Otherwise, once it gets retried,
                // 1) the stuffs in stage info become distorting, e.g. task num, input byte, e.t.c
                // 2) the first attempt starts from 0-idx, it will not be marked as a retry
                mapStage.increaseAttemptIdOnFirstSkip()
              }
            case narrowDep: NarrowDependency[_] =>
             //若为窄依赖,那就属于同一个stage。并将依赖的RDD放入waitingForVisit中,以能够在下面的while中继续向上visit,直至遍历了整个DAG图
              waitingForVisit.prepend(narrowDep.rdd)
          }
        }
      }
    }
  }
  while (waitingForVisit.nonEmpty) {
    visit(waitingForVisit.remove(0))
  }
  missing.toList
}

以上!

本文参与 腾讯云自媒体同步曝光计划,分享自微信公众号。
原始发表:2022-07-08,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 数据仓库践行者 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档