HBASE-2448 Scanner threads are interrupted without acquiring lock properly

git-svn-id: https://svn.apache.org/repos/asf/hadoop/hbase/trunk@937651 13f79535-47bb-0310-9956-ffa450edef68

core/src/main/java/org/apache/hadoop/hbase/Chore.java

@@ -83,6 +83,15 @@ public abstract class Chore extends Thread {
     }
   }
   
+  /**
+   * If the thread is currently sleeping, trigger the core to happen immediately.
+   * If it's in the middle of its operation, will begin another operation
+   * immediately after finishing this one.
+   */
+  public void triggerNow() {
+    this.sleeper.skipSleepCycle();
+  }
+  
   /**
    * Override to run a task before we start looping.
    * @return true if initial chore was successful

core/src/main/java/org/apache/hadoop/hbase/master/BaseScanner.java

@@ -588,9 +588,9 @@ abstract class BaseScanner extends Chore implements HConstants {
   }
 
   /**
-   * Notify the thread to die at the end of its next run
+   * Interrupt thread regardless of what it's doing
    */
-  public void interruptIfAlive() {
+  public void interruptAndStop() {
     synchronized(scannerLock){
       if (isAlive()) {
         super.interrupt();

core/src/main/java/org/apache/hadoop/hbase/master/ModifyTableMeta.java

@@ -72,6 +72,6 @@ class ModifyTableMeta extends TableOperation {
       updateRegionInfo(server, m.getRegionName(), i);
     }
     // kick off a meta scan right away
-    master.getRegionManager().metaScannerThread.interrupt();
+    master.getRegionManager().metaScannerThread.triggerNow();
   }
 }

core/src/main/java/org/apache/hadoop/hbase/master/ProcessRegionOpen.java

@@ -100,7 +100,7 @@ class ProcessRegionOpen extends ProcessRegionStatusChange {
           master.getRegionManager().putMetaRegionOnline(m);
           // Interrupting the Meta Scanner sleep so that it can
           // process regions right away
-          master.getRegionManager().metaScannerThread.interrupt();
+          master.getRegionManager().metaScannerThread.triggerNow();
         }
       }
       // If updated successfully, remove from pending list if the state
@@ -120,4 +120,4 @@ class ProcessRegionOpen extends ProcessRegionStatusChange {
   protected int getPriority() {
     return 0; // highest priority
   }
-}
+}

core/src/main/java/org/apache/hadoop/hbase/master/RegionManager.java

@@ -581,11 +581,11 @@ public class RegionManager implements HConstants {
     if (LOG.isDebugEnabled()) {
       LOG.debug("telling root scanner to stop");
     }
-    rootScannerThread.interruptIfAlive();
+    rootScannerThread.interruptAndStop();
     if (LOG.isDebugEnabled()) {
       LOG.debug("telling meta scanner to stop");
     }
-    metaScannerThread.interruptIfAlive();
+    metaScannerThread.interruptAndStop();
     if (LOG.isDebugEnabled()) {
       LOG.debug("meta and root scanners notified");
     }

core/src/main/java/org/apache/hadoop/hbase/util/Sleeper.java

@@ -35,7 +35,10 @@ public class Sleeper {
   private final int period;
   private final AtomicBoolean stop;
   private static final long MINIMAL_DELTA_FOR_LOGGING = 10000;
-  
+ 
+  private final Object sleepLock = new Object();
+  private boolean triggerWake = false;
+
   /**
    * @param sleep sleep time in milliseconds
    * @param stop flag for when we stop
@@ -51,6 +54,17 @@ public class Sleeper {
   public void sleep() {
     sleep(System.currentTimeMillis());
   }
+
+  /**
+   * If currently asleep, stops sleeping; if not asleep, will skip the next
+   * sleep cycle.
+   */
+  public void skipSleepCycle() {
+    synchronized (sleepLock) {
+      triggerWake = true;
+      sleepLock.notify();
+    }
+  }
   
   /**
    * Sleep for period adjusted by passed <code>startTime<code>
@@ -72,7 +86,10 @@ public class Sleeper {
     while (waitTime > 0) {
       long woke = -1;
       try {
-        Thread.sleep(waitTime);
+        synchronized (sleepLock) {
+          if (triggerWake) break;
+          sleepLock.wait(waitTime);
+        }
         woke = System.currentTimeMillis();
         long slept = woke - now;
         if (slept - this.period > MINIMAL_DELTA_FOR_LOGGING) {
@@ -92,5 +109,6 @@ public class Sleeper {
       woke = (woke == -1)? System.currentTimeMillis(): woke;
       waitTime = this.period - (woke - startTime);
     }
+    triggerWake = false;
   }
 }