From d1b7f41671feb6e17e98383b1770757b4941cc3b Mon Sep 17 00:00:00 2001
From: Tathagata Das <tathagata.das1565@gmail.com>
Date: Fri, 27 Jul 2012 09:00:45 -0700
Subject: [PATCH] Fixed bug in BoundedMemoryCache.
---
core/src/main/scala/spark/BoundedMemoryCache.scala | 6 +++++-
1 file changed, 5 insertions(+), 1 deletion(-)
diff --git a/core/src/main/scala/spark/BoundedMemoryCache.scala b/core/src/main/scala/spark/BoundedMemoryCache.scala
index fa5dcee7bb..5ea5c303bd 100644
--- a/core/src/main/scala/spark/BoundedMemoryCache.scala
+++ b/core/src/main/scala/spark/BoundedMemoryCache.scala
@@ -91,7 +91,11 @@ class BoundedMemoryCache(maxBytes: Long) extends Cache with Logging {
protected def reportEntryDropped(datasetId: Any, partition: Int, entry: Entry) {
logInfo("Dropping key (%s, %d) of size %d to make space".format(datasetId, partition, entry.size))
// TODO: remove BoundedMemoryCache
- SparkEnv.get.cacheTracker.dropEntry(datasetId.asInstanceOf[(Int, Int)]._2, partition)
+ datasetId match {
+ case rddDatasetId: (Int, Int) =>
+ SparkEnv.get.cacheTracker.dropEntry(rddDatasetId._2, partition)
+ case _ =>
+ }
}
}
--
GitLab