HBASE-9241 Add cp hook before initialize variable set to true in master intialization

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

hbase-server/src/main/java/org/apache/hadoop/hbase/coprocessor/BaseMasterObserver.java

@@ -304,6 +304,11 @@ public class BaseMasterObserver implements MasterObserver {
       throws IOException {
   }
 
+  @Override
+  public void preMasterInitialization(
+      ObserverContext<MasterCoprocessorEnvironment> ctx) throws IOException {
+  }
+  
   @Override
   public void start(CoprocessorEnvironment ctx) throws IOException {
   }

hbase-server/src/main/java/org/apache/hadoop/hbase/coprocessor/MasterObserver.java

@@ -517,6 +517,13 @@ public interface MasterObserver extends Coprocessor {
   void postStartMaster(final ObserverContext<MasterCoprocessorEnvironment> ctx)
       throws IOException;
 
+  /**
+   * Call before the master initialization is set to true.
+   * {@link org.apache.hadoop.hbase.master.HMaster} process.
+   */
+  void preMasterInitialization(final ObserverContext<MasterCoprocessorEnvironment> ctx)
+      throws IOException;
+
   /**
    * Called before a new snapshot is taken.
    * Called as part of snapshot RPC call.

hbase-server/src/main/java/org/apache/hadoop/hbase/master/HMaster.java

@@ -918,6 +918,14 @@ MasterServices, Server {
       startNamespaceJanitorChore();
     }
     
+    if (this.cpHost != null) {
+      try {
+        this.cpHost.preMasterInitialization();
+      } catch (IOException e) {
+        LOG.error("Coprocessor preMasterInitialization() hook failed", e);
+      }
+    }
+
     status.markComplete("Initialization successful");
     LOG.info("Master has completed initialization");
     initialized = true;

hbase-server/src/main/java/org/apache/hadoop/hbase/master/MasterCoprocessorHost.java

@@ -1073,6 +1073,23 @@ public class MasterCoprocessorHost
     }
   }
 
+  public void preMasterInitialization() throws IOException {
+    ObserverContext<MasterCoprocessorEnvironment> ctx = null;
+    for (MasterEnvironment env : coprocessors) {
+      if (env.getInstance() instanceof MasterObserver) {
+        ctx = ObserverContext.createAndPrepare(env, ctx);
+        try {
+          ((MasterObserver) env.getInstance()).preMasterInitialization(ctx);
+        } catch (Throwable e) {
+          handleCoprocessorThrowable(env, e);
+        }
+        if (ctx.shouldComplete()) {
+          break;
+        }
+      }
+    }
+  }
+
   void postStartMaster() throws IOException {
     ObserverContext<MasterCoprocessorEnvironment> ctx = null;
     for (MasterEnvironment env: coprocessors) {

hbase-server/src/main/java/org/apache/hadoop/hbase/security/access/AccessController.java

@@ -779,6 +779,11 @@ public class AccessController extends BaseRegionObserver
     AccessControlLists.init(ctx.getEnvironment().getMasterServices());
   }
 
+  @Override
+  public void preMasterInitialization(
+      ObserverContext<MasterCoprocessorEnvironment> ctx) throws IOException {
+  }
+
   @Override
   public void preSnapshot(final ObserverContext<MasterCoprocessorEnvironment> ctx,
       final SnapshotDescription snapshot, final HTableDescriptor hTableDescriptor)

hbase-server/src/test/java/org/apache/hadoop/hbase/coprocessor/TestMasterObserver.java

@@ -103,6 +103,7 @@ public class TestMasterObserver {
     private boolean postBalanceSwitchCalled;
     private boolean preShutdownCalled;
     private boolean preStopMasterCalled;
+    private boolean preMasterInitializationCalled;
     private boolean postStartMasterCalled;
     private boolean startCalled;
     private boolean stopCalled;
@@ -610,6 +611,16 @@ public class TestMasterObserver {
       preStopMasterCalled = true;
     }
 
+    @Override
+    public void preMasterInitialization(
+        ObserverContext<MasterCoprocessorEnvironment> ctx) throws IOException {
+      preMasterInitializationCalled = true;
+    }
+    
+    public boolean wasMasterInitializationCalled(){
+      return preMasterInitializationCalled;
+    }
+    
     @Override
     public void postStartMaster(ObserverContext<MasterCoprocessorEnvironment> ctx)
         throws IOException {
@@ -965,6 +976,8 @@ public class TestMasterObserver {
 
     // check basic lifecycle
     assertTrue("MasterObserver should have been started", cp.wasStarted());
+    assertTrue("preMasterInitialization() hook should have been called",
+        cp.wasMasterInitializationCalled());
     assertTrue("postStartMaster() hook should have been called",
         cp.wasStartMasterCalled());
   }