Fix the bug due to changed Listener that BlockHeartbeatReporter fails…

… to report to master on remove blocks
Alluxio · Jun 27, 2015 · dc62f88 · dc62f88
1 parent 24d7116
commit dc62f88
Showing 1 changed file with 9 additions and 9 deletions.
diff --git a/servers/src/main/java/tachyon/worker/block/BlockHeartbeatReporter.java b/servers/src/main/java/tachyon/worker/block/BlockHeartbeatReporter.java
@@ -105,14 +105,7 @@ public void preRemoveBlock(long userId, long blockId) {
 
   @Override
   public void postRemoveBlock(long userId, long blockId) {
-    synchronized (mLock) {
-      // Remove the block from list of added blocks, in case it was added in this heartbeat period.
-      removeBlockFromAddedBlocks(blockId);
-      // Add to the list of removed blocks in this heartbeat period.
-      if (!mRemovedBlocks.contains(blockId)) {
-        mRemovedBlocks.add(blockId);
-      }
-    }
+    // Do nothing
   }
 
   @Override
@@ -122,7 +115,14 @@ public void preEvictBlock(long userId, long blockId) {
 
   @Override
   public void postEvictBlock(long userId, long blockId) {
-    // Do nothing
+    synchronized (mLock) {
+      // Remove the block from list of added blocks, in case it was added in this heartbeat period.
+      removeBlockFromAddedBlocks(blockId);
+      // Add to the list of removed blocks in this heartbeat period.
+      if (!mRemovedBlocks.contains(blockId)) {
+        mRemovedBlocks.add(blockId);
+      }
+    }
   }
 
   @Override