Fix dead lock

apache · Apr 13, 2015 · d6c94da · d6c94da
1 parent 8b73701
commit d6c94da
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/core/src/main/scala/org/apache/spark/util/collection/TieredDiskMerger.scala b/core/src/main/scala/org/apache/spark/util/collection/TieredDiskMerger.scala
@@ -93,8 +93,8 @@ private[spark] class TieredDiskMerger[K, C](
    * Notify the merger that no more on disk blocks will be registered.
    */
   def doneRegisteringOnDiskBlocks(): Unit = {
-    doneRegistering = true
     mergeReadyMonitor.synchronized {
+      doneRegistering = true
       mergeReadyMonitor.notify()
     }
   }
@@ -175,7 +175,9 @@ private[spark] class TieredDiskMerger[K, C](
       while (!doneRegistering || onDiskBlocks.size() > maxMergeFactor) {
         while (!shouldMergeNow()) {
           mergeReadyMonitor.synchronized {
-            mergeReadyMonitor.wait()
+            if (!shouldMergeNow()) {
+              mergeReadyMonitor.wait()
+            }
           }
         }