Java Code Examples for org.apache.hadoop.hbase.HBaseTestingUtility#waitUntilAllRegionsAssigned()

/**
 * Creates the table with the column families
 *
 * @param tableName
 * @param columnFamilies
 * @return
 */
public static void createTable(String tableName, String... columnFamilies) {
    HBaseTestingUtility testingUtility = getUtility();
    if (!status.get()) {
        throw new RuntimeException("The mini cluster hasn't started yet. " +
          " Call HBaseTestUtil#startMiniCluster() before creating a table");
    }
    final TableName name = TableName.valueOf(tableName);
    try (HBaseAdmin hBaseAdmin = testingUtility.getHBaseAdmin()) {
        final HTableDescriptor hTableDescriptor = new HTableDescriptor(name);
        for (String family : columnFamilies) {
            final HColumnDescriptor hColumnDescriptor = new HColumnDescriptor(Bytes.toBytes(family));
            hTableDescriptor.addFamily(hColumnDescriptor);
        }

        hBaseAdmin.createTable(hTableDescriptor);
        testingUtility.waitUntilAllRegionsAssigned(name);
    } catch (IOException e) {
        throw new RuntimeException(e);
    }
}
 

public static void createTable(HBaseTestingUtility testUtil, Admin admin, TableDescriptor htd,
    byte[][] splitKeys) throws Exception {
  // NOTE: We need a latch because admin is not sync,
  // so the postOp coprocessor method may be called after the admin operation returned.
  MasterSyncObserver observer = testUtil.getHBaseCluster().getMaster()
    .getMasterCoprocessorHost().findCoprocessor(MasterSyncObserver.class);
  observer.tableCreationLatch = new CountDownLatch(1);
  if (splitKeys != null) {
    admin.createTable(htd, splitKeys);
  } else {
    admin.createTable(htd);
  }
  observer.tableCreationLatch.await();
  observer.tableCreationLatch = null;
  testUtil.waitUntilAllRegionsAssigned(htd.getTableName());
}
 

public static void createTable(HBaseTestingUtility testUtil,
    TableDescriptorBuilder.ModifyableTableDescriptor tableDescriptor, byte [][] splitKeys)
    throws Exception {
  // NOTE: We need a latch because admin is not sync,
  // so the postOp coprocessor method may be called after the admin operation returned.
  MasterSyncObserver observer = testUtil.getHBaseCluster().getMaster()
    .getMasterCoprocessorHost().findCoprocessor(MasterSyncObserver.class);
  observer.tableCreationLatch = new CountDownLatch(1);
  Admin admin = testUtil.getAdmin();
  if (splitKeys != null) {
    admin.createTable(tableDescriptor, splitKeys);
  } else {
    admin.createTable(tableDescriptor);
  }
  observer.tableCreationLatch.await();
  observer.tableCreationLatch = null;
  testUtil.waitUntilAllRegionsAssigned(tableDescriptor.getTableName());
}
 

/**
 * Create a Mob table.
 *
 * @param util
 * @param tableName
 * @param families
 * @return An Table instance for the created table.
 * @throws IOException
 */
public static Table createMobTable(final HBaseTestingUtility util,
    final TableName tableName, final byte[]... families) throws IOException {
  TableDescriptorBuilder builder = TableDescriptorBuilder.newBuilder(tableName);
  for (byte[] family : families) {
    // Disable blooms (they are on by default as of 0.95) but we disable them
    // here because
    // tests have hard coded counts of what to expect in block cache, etc.,
    // and blooms being
    // on is interfering.
    builder.setColumnFamily(ColumnFamilyDescriptorBuilder.newBuilder(family)
            .setBloomFilterType(BloomType.NONE)
            .setMobEnabled(true)
            .setMobThreshold(0L)
            .build());
  }
  util.getAdmin().createTable(builder.build());
  // HBaseAdmin only waits for regions to appear in hbase:meta we should wait
  // until they are assigned
  util.waitUntilAllRegionsAssigned(tableName);
  return ConnectionFactory.createConnection(util.getConfiguration()).getTable(tableName);
}
 

@BeforeClass
public static void setupBeforeClass() throws Exception {
  TEST_UTIL = new HBaseTestingUtility();

  conf = TEST_UTIL.getConfiguration();

  // Try to frequently clean up compacted files
  conf.setInt("hbase.hfile.compaction.discharger.interval", 1000);
  conf.setInt("hbase.master.hfilecleaner.ttl", 1000);

  TEST_UTIL.startMiniCluster(1);
  TEST_UTIL.getHBaseCluster().waitForActiveAndReadyMaster();
  TEST_UTIL.waitUntilAllRegionsAssigned(TableName.META_TABLE_NAME);

  rootDir = CommonFSUtils.getRootDir(conf);
  fs = TEST_UTIL.getTestFileSystem();
}
 

public static void createTable(HBaseTestingUtility testUtil, TableDescriptor tableDescriptor,
    byte[][] splitKeys) throws Exception {
  // NOTE: We need a latch because admin is not sync,
  // so the postOp coprocessor method may be called after the admin operation returned.
  MasterSyncCoprocessor coproc = testUtil.getHBaseCluster().getMaster()
      .getMasterCoprocessorHost().findCoprocessor(MasterSyncCoprocessor.class);
  coproc.tableCreationLatch = new CountDownLatch(1);
  if (splitKeys != null) {
    admin.createTable(tableDescriptor, splitKeys);
  } else {
    admin.createTable(tableDescriptor);
  }
  coproc.tableCreationLatch.await();
  coproc.tableCreationLatch = null;
  testUtil.waitUntilAllRegionsAssigned(tableDescriptor.getTableName());
}