Set currentMemory to 0 in clear().
Remove unnecessary entries.get() call.
This commit is contained in:
parent
293c758cc0
commit
d34672f668
|
@ -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")
|
||||
}
|
||||
|
|
Loading…
Reference in a new issue