get JobProgressListener directly from UI

apache · Apr 1, 2015 · db61211 · db61211
1 parent fdfc181
commit db61211
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/core/src/main/scala/org/apache/spark/status/api/v1/AllStagesResource.scala b/core/src/main/scala/org/apache/spark/status/api/v1/AllStagesResource.scala
@@ -35,7 +35,7 @@ private[v1] class AllStagesResource(uiRoot: UIRoot) {
     @QueryParam("status") statuses: java.util.List[StageStatus]
   ): Seq[StageData] = {
     uiRoot.withSparkUI(appId) { ui =>
-      val listener = ui.stagesTab.listener
+      val listener = ui.jobProgressListener
       val stageAndStatus = AllStagesResource.stagesAndStatus(ui)
       val adjStatuses = {
         if (statuses.isEmpty()) {
@@ -118,7 +118,7 @@ private[v1] object AllStagesResource {
   }
 
   def stagesAndStatus(ui: SparkUI): Seq[(StageStatus, Seq[StageInfo])] = {
-    val listener = ui.stagesTab.listener
+    val listener = ui.jobProgressListener
     listener.synchronized {
       Seq(
         StageStatus.Active -> listener.activeStages.values.toSeq,