put the filter inside

Author: XuTingjun

core/src/main/scala/org/apache/spark/ui/UIUtils.scala

@@ -362,7 +362,7 @@ private[spark] object UIUtils extends Logging {
               { g.incomingEdges.map { e => <div class="incoming-edge">{e.fromId},{e.toId}</div> } }
               { g.outgoingEdges.map { e => <div class="outgoing-edge">{e.fromId},{e.toId}</div> } }
               {
-                g.rootCluster.getAllNodes.filter(_.cached).map { n =>
+                g.rootCluster.getCachedNode.map { n =>
                   <div class="cached-rdd">{n.id}</div>
                 }
               }

core/src/main/scala/org/apache/spark/ui/scope/RDDOperationGraph.scala

@@ -70,6 +70,16 @@ private[ui] class RDDOperationCluster(val id: String, private var _name: String)
   def getAllNodes: Seq[RDDOperationNode] = {
     _childNodes ++ _childClusters.flatMap(_.childNodes)
   }
+
+  /** Return all the node which are cached. */
+  def getCachedNode: Seq[RDDOperationNode] = {
+    var cachedNodes = new ListBuffer[RDDOperationNode]
+    cachedNodes ++= (_childNodes.filter(_.cached))
+    for(cluster <- _childClusters) {
+      cachedNodes ++= (cluster._childNodes.filter(_.cached))
+    }
+    cachedNodes
+  }
 }
 
 private[ui] object RDDOperationGraph extends Logging {