Java EventHandlerListener类代码示例

import org.apache.hadoop.hbase.executor.EventHandler.EventHandlerListener; //导入依赖的package包/类
@Test (timeout=300000) public void testCloseRegion()
throws Exception {
  LOG.info("Running testCloseRegion");
  MiniHBaseCluster cluster = TEST_UTIL.getHBaseCluster();
  LOG.info("Number of region servers = " + cluster.getLiveRegionServerThreads().size());

  int rsIdx = 0;
  HRegionServer regionServer = TEST_UTIL.getHBaseCluster().getRegionServer(rsIdx);
  HRegionInfo hri = getNonMetaRegion(regionServer.getOnlineRegions());
  LOG.debug("Asking RS to close region " + hri.getRegionNameAsString());

  AtomicBoolean closeEventProcessed = new AtomicBoolean(false);
  EventHandlerListener listener =
    new CloseRegionEventListener(hri.getRegionNameAsString(),
        closeEventProcessed);
  cluster.getMaster().executorService.registerListener(EventType.RS_ZK_REGION_CLOSED, listener);

  cluster.getMaster().assignmentManager.unassign(hri);

  while (!closeEventProcessed.get()) {
    Threads.sleep(100);
  }
  LOG.info("Done with testCloseRegion");
}
 

import org.apache.hadoop.hbase.executor.EventHandler.EventHandlerListener; //导入依赖的package包/类
@Test (timeout=300000) public void testCloseRegion()
throws Exception {
  LOG.info("Running testCloseRegion");
  MiniHBaseCluster cluster = TEST_UTIL.getHBaseCluster();
  LOG.info("Number of region servers = " + cluster.getLiveRegionServerThreads().size());

  int rsIdx = 0;
  HRegionServer regionServer = TEST_UTIL.getHBaseCluster().getRegionServer(rsIdx);
  HRegionInfo hri = getNonMetaRegion(ProtobufUtil.getOnlineRegions(regionServer));
  LOG.debug("Asking RS to close region " + hri.getRegionNameAsString());

  AtomicBoolean closeEventProcessed = new AtomicBoolean(false);
  EventHandlerListener listener =
    new CloseRegionEventListener(hri.getRegionNameAsString(),
        closeEventProcessed);
  cluster.getMaster().executorService.registerListener(EventType.RS_ZK_REGION_CLOSED, listener);

  cluster.getMaster().assignmentManager.unassign(hri);

  while (!closeEventProcessed.get()) {
    Threads.sleep(100);
  }
  LOG.info("Done with testCloseRegion");
}
 

import org.apache.hadoop.hbase.executor.EventHandler.EventHandlerListener; //导入依赖的package包/类
protected Executor(String name, int maxThreads,
    final Map<EventType, EventHandlerListener> eventHandlerListeners) {
  this.id = seqids.incrementAndGet();
  this.name = name;
  this.eventHandlerListeners = eventHandlerListeners;
  // create the thread pool executor
  this.threadPoolExecutor = new TrackingThreadPoolExecutor(
      maxThreads, maxThreads,
      keepAliveTimeInMillis, TimeUnit.MILLISECONDS, q);
  // name the threads for this threadpool
  ThreadFactoryBuilder tfb = new ThreadFactoryBuilder();
  tfb.setNameFormat(this.name + "-%d");
  this.threadPoolExecutor.setThreadFactory(tfb.build());
}
 

import org.apache.hadoop.hbase.executor.EventHandler.EventHandlerListener; //导入依赖的package包/类
/**
 * Submit the event to the queue for handling.
 * @param event
 */
void submit(final EventHandler event) {
  // If there is a listener for this type, make sure we call the before
  // and after process methods.
  EventHandlerListener listener =
    this.eventHandlerListeners.get(event.getEventType());
  if (listener != null) {
    event.setListener(listener);
  }
  this.threadPoolExecutor.execute(event);
}
 

import org.apache.hadoop.hbase.executor.EventHandler.EventHandlerListener; //导入依赖的package包/类
protected Executor(String name, int maxThreads,
    final Map<EventHandler.EventType, EventHandlerListener> eventHandlerListeners) {
  this.id = seqids.incrementAndGet();
  this.name = name;
  this.eventHandlerListeners = eventHandlerListeners;
  // create the thread pool executor
  this.threadPoolExecutor = new TrackingThreadPoolExecutor(
      maxThreads, maxThreads,
      keepAliveTimeInMillis, TimeUnit.MILLISECONDS, q);
  // name the threads for this threadpool
  ThreadFactoryBuilder tfb = new ThreadFactoryBuilder();
  tfb.setNameFormat(this.name + "-%d");
  this.threadPoolExecutor.setThreadFactory(tfb.build());
}
 

import org.apache.hadoop.hbase.executor.EventHandler.EventHandlerListener; //导入依赖的package包/类
/**
 * Test we reopen a region once closed.
 * @throws Exception
 */
@Test (timeout=300000) public void testReOpenRegion()
throws Exception {
  MiniHBaseCluster cluster = TEST_UTIL.getHBaseCluster();
  LOG.info("Number of region servers = " +
    cluster.getLiveRegionServerThreads().size());

  int rsIdx = 0;
  HRegionServer regionServer =
    TEST_UTIL.getHBaseCluster().getRegionServer(rsIdx);
  HRegionInfo hri = getNonMetaRegion(regionServer.getOnlineRegions());
  LOG.debug("Asking RS to close region " + hri.getRegionNameAsString());

  AtomicBoolean closeEventProcessed = new AtomicBoolean(false);
  AtomicBoolean reopenEventProcessed = new AtomicBoolean(false);

  EventHandlerListener closeListener =
    new ReopenEventListener(hri.getRegionNameAsString(),
        closeEventProcessed, EventType.RS_ZK_REGION_CLOSED);
  cluster.getMaster().executorService.
    registerListener(EventType.RS_ZK_REGION_CLOSED, closeListener);

  EventHandlerListener openListener =
    new ReopenEventListener(hri.getRegionNameAsString(),
        reopenEventProcessed, EventType.RS_ZK_REGION_OPENED);
  cluster.getMaster().executorService.
    registerListener(EventType.RS_ZK_REGION_OPENED, openListener);

  LOG.info("Unassign " + hri.getRegionNameAsString());
  cluster.getMaster().assignmentManager.unassign(hri);

  while (!closeEventProcessed.get()) {
    Threads.sleep(100);
  }

  while (!reopenEventProcessed.get()) {
    Threads.sleep(100);
  }

  LOG.info("Done with testReOpenRegion");
}
 

import org.apache.hadoop.hbase.executor.EventHandler.EventHandlerListener; //导入依赖的package包/类
/**
 * This test shows how a region won't be able to be assigned to a RS
 * if it's already "processing" it.
 * @throws Exception
 */
@Test
public void testRSAlreadyProcessingRegion() throws Exception {
  LOG.info("starting testRSAlreadyProcessingRegion");
  MiniHBaseCluster cluster = TEST_UTIL.getHBaseCluster();

  HRegionServer hr0 =
      cluster.getLiveRegionServerThreads().get(0).getRegionServer();
  HRegionServer hr1 =
      cluster.getLiveRegionServerThreads().get(1).getRegionServer();
  HRegionInfo hri = getNonMetaRegion(hr0.getOnlineRegions());

  // Fake that hr1 is processing the region. At top of this test we made a
  // regionserver that gave access addRegionsInTransition. Need to cast as
  // TestZKBasedOpenCloseRegionRegionServer.
  ((TestZKBasedOpenCloseRegionRegionServer) hr1).addRegionsInTransition(hri, "OPEN");

  AtomicBoolean reopenEventProcessed = new AtomicBoolean(false);
  EventHandlerListener openListener =
    new ReopenEventListener(hri.getRegionNameAsString(),
        reopenEventProcessed, EventType.RS_ZK_REGION_OPENED);
  cluster.getMaster().executorService.
    registerListener(EventType.RS_ZK_REGION_OPENED, openListener);

  // now ask the master to move the region to hr1, will fail
  TEST_UTIL.getHBaseAdmin().move(hri.getEncodedNameAsBytes(),
      Bytes.toBytes(hr1.getServerName().toString()));

  // make sure the region came back
  assertEquals(hr1.getOnlineRegion(hri.getEncodedNameAsBytes()), null);

  // remove the block and reset the boolean
  hr1.removeFromRegionsInTransition(hri);
  reopenEventProcessed.set(false);
  
  // now try moving a region when there is no region in transition.
  hri = getNonMetaRegion(hr1.getOnlineRegions());

  openListener =
    new ReopenEventListener(hri.getRegionNameAsString(),
        reopenEventProcessed, EventType.RS_ZK_REGION_OPENED);

  cluster.getMaster().executorService.
    registerListener(EventType.RS_ZK_REGION_OPENED, openListener);
  
  TEST_UTIL.getHBaseAdmin().move(hri.getEncodedNameAsBytes(),
      Bytes.toBytes(hr0.getServerName().toString()));

  while (!reopenEventProcessed.get()) {
    Threads.sleep(100);
  }

  // make sure the region has moved from the original RS
  assertTrue(hr1.getOnlineRegion(hri.getEncodedNameAsBytes()) == null);

}