Fix storage UI bug

andrewor14 · andrewor14 · commit 408585aa9db6 · 2014-03-31T18:20:29.000-07:00
diff --git a/core/src/main/scala/org/apache/spark/ui/storage/BlockManagerUI.scala b/core/src/main/scala/org/apache/spark/ui/storage/BlockManagerUI.scala
@@ -84,7 +84,7 @@ private[ui] class BlockManagerListener(storageStatusListener: StorageStatusListe
 
   override def onStageSubmitted(stageSubmitted: SparkListenerStageSubmitted) = synchronized {
     val rddInfo = stageSubmitted.stageInfo.rddInfo
-    _rddInfoMap(rddInfo.id) = rddInfo
+    _rddInfoMap.getOrElseUpdate(rddInfo.id, rddInfo)
   }
 
   override def onStageCompleted(stageCompleted: SparkListenerStageCompleted) = synchronized {

Original file line number	Diff line number	Diff line change
`@@ -84,7 +84,7 @@ private[ui] class BlockManagerListener(storageStatusListener: StorageStatusListe`
`84`	`84`
`85`	`85`	`override def onStageSubmitted(stageSubmitted: SparkListenerStageSubmitted) = synchronized {`
`86`	`86`	`val rddInfo = stageSubmitted.stageInfo.rddInfo`
`87`		`- _rddInfoMap(rddInfo.id) = rddInfo`
	`87`	`+ _rddInfoMap.getOrElseUpdate(rddInfo.id, rddInfo)`
`88`	`88`	`}`
`89`	`89`
`90`	`90`	`override def onStageCompleted(stageCompleted: SparkListenerStageCompleted) = synchronized {`