Set currentMemory to 0 in clear().

Remove unnecessary entries.get() call.
This commit is contained in:
Mike 2013-09-11 18:01:19 -07:00
parent 293c758cc0
commit d34672f668

View file

@ -110,9 +110,8 @@ private class MemoryStore(blockManager: BlockManager, maxMemory: Long)
override def remove(blockId: String): Boolean = {
entries.synchronized {
val entry = entries.get(blockId)
val entry = entries.remove(blockId)
if (entry != null) {
entries.remove(blockId)
currentMemory -= entry.size
logInfo("Block %s of size %d dropped from memory (free %d)".format(
blockId, entry.size, freeMemory))
@ -126,6 +125,7 @@ private class MemoryStore(blockManager: BlockManager, maxMemory: Long)
override def clear() {
entries.synchronized {
entries.clear()
currentMemory = 0
}
logInfo("MemoryStore cleared")
}