spark AllStagesPage 源码

spark AllStagesPage 代码

文件路径：/core/src/main/scala/org/apache/spark/ui/jobs/AllStagesPage.scala

/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License.  You may obtain a copy of the License at
 *
 *    http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.apache.spark.ui.jobs

import javax.servlet.http.HttpServletRequest

import scala.xml.{Attribute, Elem, Node, NodeSeq, Null, Text}

import org.apache.spark.scheduler.Schedulable
import org.apache.spark.status.{AppSummary, PoolData}
import org.apache.spark.status.api.v1.{StageData, StageStatus}
import org.apache.spark.ui.{UIUtils, WebUIPage}

/** Page showing list of all ongoing and recently finished stages and pools */
private[ui] class AllStagesPage(parent: StagesTab) extends WebUIPage("") {
  private val sc = parent.sc
  private val subPath = "stages"

  def render(request: HttpServletRequest): Seq[Node] = {
    // For now, pool information is only accessible in live UIs
    val pools = sc.map(_.getAllPools).getOrElse(Seq.empty[Schedulable]).map { pool =>
      val uiPool = parent.store.asOption(parent.store.pool(pool.name)).getOrElse(
        new PoolData(pool.name, Set()))
      pool -> uiPool
    }.toMap
    val poolTable = new PoolTable(pools, parent)

    val allStatuses = Seq(StageStatus.ACTIVE, StageStatus.PENDING, StageStatus.COMPLETE,
      StageStatus.SKIPPED, StageStatus.FAILED)

    val allStages = parent.store.stageList(null)
    val appSummary = parent.store.appSummary()

    val (summaries, tables) = allStatuses.map(
      summaryAndTableForStatus(allStages, appSummary, _, request)).unzip

    val summary: NodeSeq =
      <div>
        <ul class="list-unstyled">
          {summaries.flatten}
        </ul>
      </div>

    val poolsDescription = if (parent.isFairScheduler) {
        <span class="collapse-aggregated-poolTable collapse-table"
            onClick="collapseTable('collapse-aggregated-poolTable','aggregated-poolTable')">
          <h4>
            <span class="collapse-table-arrow arrow-open"></span>
            <a>Fair Scheduler Pools ({pools.size})</a>
          </h4>
        </span> ++
        <div class="aggregated-poolTable collapsible-table">
          {poolTable.toNodeSeq(request)}
        </div>
      } else {
        Seq.empty[Node]
      }

    val content = summary ++ poolsDescription ++ tables.flatten.flatten

    UIUtils.headerSparkPage(request, "Stages for All Jobs", content, parent)
  }

  private def summaryAndTableForStatus(
      allStages: Seq[StageData],
      appSummary: AppSummary,
      status: StageStatus,
      request: HttpServletRequest): (Option[Elem], Option[NodeSeq]) = {
    val stages = if (status == StageStatus.FAILED) {
      allStages.filter(_.status == status).reverse
    } else {
      allStages.filter(_.status == status)
    }

    if (stages.isEmpty) {
      (None, None)
    } else {
      val killEnabled = status == StageStatus.ACTIVE && parent.killEnabled
      val isFailedStage = status == StageStatus.FAILED

      val stagesTable =
        new StageTableBase(parent.store, request, stages, statusName(status), stageTag(status),
          parent.basePath, subPath, parent.isFairScheduler, killEnabled, isFailedStage)
      val stagesSize = stages.size
      (Some(summary(appSummary, status, stagesSize)),
        Some(table(appSummary, status, stagesTable, stagesSize)))
    }
  }

  private def statusName(status: StageStatus): String = status match {
    case StageStatus.ACTIVE => "active"
    case StageStatus.COMPLETE => "completed"
    case StageStatus.FAILED => "failed"
    case StageStatus.PENDING => "pending"
    case StageStatus.SKIPPED => "skipped"
  }

  private def stageTag(status: StageStatus): String = s"${statusName(status)}Stage"

  private def headerDescription(status: StageStatus): String = statusName(status).capitalize

  private def summaryContent(appSummary: AppSummary, status: StageStatus, size: Int): String = {
    if (status == StageStatus.COMPLETE && appSummary.numCompletedStages != size) {
      s"${appSummary.numCompletedStages}, only showing $size"
    } else {
      s"$size"
    }
  }

  private def summary(appSummary: AppSummary, status: StageStatus, size: Int): Elem = {
    val summary =
      <li>
        <a href={s"#${statusName(status)}"}>
          <strong>{headerDescription(status)} Stages:</strong>
        </a>
        {summaryContent(appSummary, status, size)}
      </li>

    if (status == StageStatus.COMPLETE) {
      summary % Attribute(None, "id", Text("completed-summary"), Null)
    } else {
      summary
    }
  }

  private def table(
      appSummary: AppSummary,
      status: StageStatus,
      stagesTable: StageTableBase,
      size: Int): NodeSeq = {
    val classSuffix = s"${statusName(status).capitalize}Stages"
    <span id={statusName(status)}
          class={s"collapse-aggregated-all$classSuffix collapse-table"}
          onClick={s"collapseTable('collapse-aggregated-all$classSuffix'," +
            s" 'aggregated-all$classSuffix')"}>
      <h4>
        <span class="collapse-table-arrow arrow-open"></span>
        <a>{headerDescription(status)} Stages ({summaryContent(appSummary, status, size)})</a>
      </h4>
    </span> ++
      <div class={s"aggregated-all$classSuffix collapsible-table"}>
        {stagesTable.toNodeSeq}
      </div>
  }
}

相关信息

spark 源码目录

相关文章

spark AllJobsPage 源码

spark JobDataUtil 源码

spark JobPage 源码

spark JobsTab 源码

spark PoolPage 源码

spark PoolTable 源码

spark StagePage 源码

spark StageTable 源码

spark StagesTab 源码

spark TaskDetailsClassNames 源码