From 6222697f41a69ac12717f3b421aa303b26c4642c Mon Sep 17 00:00:00 2001 From: Tathagata Das Date: Fri, 4 Apr 2014 10:10:59 -0700 Subject: [PATCH] Fixed bug and adding unit test for removeBroadcast in BlockManagerSuite. --- .../scala/org/apache/spark/storage/ShuffleBlockManager.scala | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/core/src/main/scala/org/apache/spark/storage/ShuffleBlockManager.scala b/core/src/main/scala/org/apache/spark/storage/ShuffleBlockManager.scala index 1f9732565709d..4eeeb9aa9c7ab 100644 --- a/core/src/main/scala/org/apache/spark/storage/ShuffleBlockManager.scala +++ b/core/src/main/scala/org/apache/spark/storage/ShuffleBlockManager.scala @@ -170,9 +170,9 @@ class ShuffleBlockManager(blockManager: BlockManager) extends Logging { } /** Remove all the blocks / files and metadata related to a particular shuffle. */ - def removeShuffle(shuffleId: ShuffleId) { - removeShuffleBlocks(shuffleId) + def removeShuffle(shuffleId: ShuffleId): Boolean = { shuffleStates.remove(shuffleId) + removeShuffleBlocks(shuffleId) } /** Remove all the blocks / files related to a particular shuffle. */