Added timeline view to StagePage

nemccarthy · May 1, 2015 · e694b8e · e694b8e
1 parent 8f6610c
commit e694b8e
Show file tree

Hide file tree

Showing 3 changed files with 304 additions and 193 deletions.
diff --git a/core/src/main/resources/org/apache/spark/ui/static/timeline-view.css b/core/src/main/resources/org/apache/spark/ui/static/timeline-view.css
@@ -34,10 +34,6 @@ div#application-timeline, div#job-timeline {
   height: 80px;
 }
 
-#task-assignment-timeline>div.control-panel {
-  float: left;
-}
-
 #task-assignment-timeline>.timeline-header:after {
   content: "";
   clear: both;
@@ -61,18 +57,61 @@ div#application-timeline, div#job-timeline {
   height: 19px;
 }
 
+rect.scheduler-delay-proportion {
+  fill: #F6D76B;
+}
+
+rect.deserialization-time-proportion {
+  fill: #FFBDD8;
+}
+
+rect.shuffle-read-time-proportion {
+  fill: #8AC7DE;
+}
+
+rect.executor-runtime-proportion {
+  fill: #D9EB52;
+}
+
+rect.shuffle-write-time-proportion {
+  fill: #87796F;
+}
+
+rect.serialization-time-proportion {
+  fill: #93DFB8;
+}
+
+rect.getting-result-time-proportion {
+  fill: #FF9036;
+}
+
 .vis.timeline .item.task.succeeded {
   background-color: #D5DDF6;
 }
 
+.legend-area rect.succeeded-task-legend {
+  fill: #D5DDF6;
+  stroke: #97B0F8;
+}
+
 .vis.timeline .item.task.failed {
   background-color: #FF5475;
 }
 
+.legend-area rect.failed-task-legend {
+  fill: #FF5475;
+  stroke: #97B0F8;
+}
+
 .vis.timeline .item.task.running {
   background-color: #FDFFCA;
 }
 
+.legend-area rect.running-task-legend {
+  fill: #FDFFCA;
+  stroke: #97B0F8;
+}
+
 #application-timeline div.legend-area {
   margin-top: 5px;
 }
@@ -215,11 +254,20 @@ tr.corresponding-item-hover>td, tr.corresponding-item-hover>th {
   display: none;
 }
 
+#task-assignment-timeline.collapsed {
+  display: none;
+}
+
 .control-panel {
   margin-bottom: 5px;
 }
 
-span.expand-application-timeline, span.expand-job-timeline {
+#task-assignment-timeline .control-panel {
+  float: left;
+}
+
+span.expand-application-timeline, span.expand-job-timeline,
+span.expand-task-assignment-timeline {
   cursor: pointer;
 }
 

diff --git a/core/src/main/resources/org/apache/spark/ui/static/timeline-view.js b/core/src/main/resources/org/apache/spark/ui/static/timeline-view.js
@@ -168,6 +168,41 @@ function drawTaskAssignmentTimeline(groupArray, eventObjArray, minLaunchTime, zo
   }
   taskTimeline.setWindow(minLaunchTime, curEnd);
   setupZoomable('#task-assignment-timeline-zoom-lock', taskTimeline);
+
+  function setupTaskEventAction() {
+    $(".item.range.task.task-assignment-timeline-object").each(function() {
+      var getTaskIdx = function(baseElem) {
+        var taskIdxText = $($(baseElem).find(".task-assignment-timeline-content")[0]).text();
+        var taskIdx = taskIdxText.match("Task (\\d+)\\(")[1];
+        return taskIdx;
+      };
+
+      $(this).hover(
+        function() {
+          var id = getTaskIdx(this);
+          $($(this).find("div.task-assignment-timeline-content")[0]).tooltip("show");
+        },
+        function() {
+          var id = getTaskIdx(this);
+          $($(this).find("div.task-assignment-timeline-content")[0])
+          .tooltip("hide");
+        }
+      );
+    });
+  }
+
+  setupTaskEventAction();
+  taskTimeline.on("rangechanged", function(properties) {
+    setupTaskEventAction();
+  });
+
+  $("span.expand-task-assignment-timeline").click(function() {
+    $("#task-assignment-timeline").toggleClass('collapsed');
+
+     // Switch the class of the arrow from open to closed.
+    $(this).find('.expand-task-assignment-timeline-arrow').toggleClass('arrow-open');
+    $(this).find('.expand-task-assignment-timeline-arrow').toggleClass('arrow-closed');
+  });
 }
 
 function setupExecutorEventAction() {