Java RegionState.State方法代码示例

import org.apache.hadoop.hbase.master.RegionState; //导入方法依赖的package包/类
/**
 * Sets the location of <code>hbase:meta</code> in ZooKeeper to the
 * specified server address.
 * @param zookeeper
 * @param serverName
 * @param replicaId
 * @param state
 * @throws KeeperException
 */
public static void setMetaLocation(ZooKeeperWatcher zookeeper,
    ServerName serverName, int replicaId, RegionState.State state) throws KeeperException {
  LOG.info("Setting hbase:meta region location in ZooKeeper as " + serverName);
  // Make the MetaRegionServer pb and then get its bytes and save this as
  // the znode content.
  MetaRegionServer pbrsr = MetaRegionServer.newBuilder()
    .setServer(ProtobufUtil.toServerName(serverName))
    .setRpcVersion(HConstants.RPC_CURRENT_VERSION)
    .setState(state.convert()).build();
  byte[] data = ProtobufUtil.prependPBMagic(pbrsr.toByteArray());
  try {
    ZKUtil.setData(zookeeper, zookeeper.getZNodeForReplica(replicaId), data);
  } catch(KeeperException.NoNodeException nne) {
    if (replicaId == HRegionInfo.DEFAULT_REPLICA_ID) {
      LOG.debug("META region location doesn't exist, create it");
    } else {
      LOG.debug("META region location doesn't exist for replicaId " + replicaId +
          ", create it");
    }
    ZKUtil.createAndWatch(zookeeper, zookeeper.getZNodeForReplica(replicaId), data);
  }
}
 

import org.apache.hadoop.hbase.master.RegionState; //导入方法依赖的package包/类
/**
 * Sets the location of <code>hbase:meta</code> in ZooKeeper to the
 * specified server address.
 * @param zookeeper zookeeper reference
 * @param serverName The server hosting <code>hbase:meta</code>
 * @param state The region transition state
 * @throws KeeperException unexpected zookeeper exception
 */
public static void setMetaLocation(ZooKeeperWatcher zookeeper,
    ServerName serverName, RegionState.State state) throws KeeperException {
  LOG.info("Setting hbase:meta region location in ZooKeeper as " + serverName);
  // Make the MetaRegionServer pb and then get its bytes and save this as
  // the znode content.
  MetaRegionServer pbrsr = MetaRegionServer.newBuilder()
    .setServer(ProtobufUtil.toServerName(serverName))
    .setRpcVersion(HConstants.RPC_CURRENT_VERSION)
    .setState(state.convert()).build();
  byte[] data = ProtobufUtil.prependPBMagic(pbrsr.toByteArray());
  try {
    ZKUtil.setData(zookeeper, zookeeper.metaServerZNode, data);
  } catch(KeeperException.NoNodeException nne) {
    LOG.debug("META region location doesn't existed, create it");
    ZKUtil.createAndWatch(zookeeper, zookeeper.metaServerZNode, data);
  }
}
 

import org.apache.hadoop.hbase.master.RegionState; //导入方法依赖的package包/类
/**
 * Test normal operations
 */
@Test public void testMetaLookup()
        throws IOException, InterruptedException, ServiceException, KeeperException {
  final ClientProtos.ClientService.BlockingInterface client =
          Mockito.mock(ClientProtos.ClientService.BlockingInterface.class);

  Mockito.when(client.get((RpcController)Mockito.any(), (GetRequest)Mockito.any())).
          thenReturn(GetResponse.newBuilder().build());

  final MetaTableLocator mtl = new MetaTableLocator();
  assertNull(mtl.getMetaRegionLocation(this.watcher));
  for (RegionState.State state : RegionState.State.values()) {
    if (state.equals(RegionState.State.OPEN))
      continue;
    MetaTableLocator.setMetaLocation(this.watcher, SN, state);
    assertNull(mtl.getMetaRegionLocation(this.watcher));
    assertEquals(state, MetaTableLocator.getMetaRegionState(this.watcher).getState());
  }
  MetaTableLocator.setMetaLocation(this.watcher, SN, RegionState.State.OPEN);
  assertEquals(mtl.getMetaRegionLocation(this.watcher), SN);
  assertEquals(RegionState.State.OPEN,
    MetaTableLocator.getMetaRegionState(this.watcher).getState());

  mtl.deleteMetaLocation(this.watcher);
  assertNull(MetaTableLocator.getMetaRegionState(this.watcher).getServerName());
  assertEquals(MetaTableLocator.getMetaRegionState(this.watcher).getState(),
    RegionState.State.OFFLINE);
  assertNull(mtl.getMetaRegionLocation(this.watcher));
}
 

import org.apache.hadoop.hbase.master.RegionState; //导入方法依赖的package包/类
public Map<RegionState.State, List<RegionInfo>> getRegionByStateOfTable(TableName tableName) {
  final State[] states = State.values();
  final Map<RegionState.State, List<RegionInfo>> tableRegions =
      new HashMap<State, List<RegionInfo>>(states.length);
  for (int i = 0; i < states.length; ++i) {
    tableRegions.put(states[i], new ArrayList<RegionInfo>());
  }

  for (RegionStateNode node: regionsMap.values()) {
    if (node.getTable().equals(tableName)) {
      tableRegions.get(node.getState()).add(node.getRegionInfo());
    }
  }
  return tableRegions;
}
 

import org.apache.hadoop.hbase.master.RegionState; //导入方法依赖的package包/类
/**
 * Test normal operations
 */
@Test public void testMetaLookup()
        throws IOException, InterruptedException, ServiceException, KeeperException {
  final ClientProtos.ClientService.BlockingInterface client =
          Mockito.mock(ClientProtos.ClientService.BlockingInterface.class);

  Mockito.when(client.get((RpcController)Mockito.any(), (GetRequest)Mockito.any())).
          thenReturn(GetResponse.newBuilder().build());

  final MetaTableLocator mtl = new MetaTableLocator();
  assertNull(mtl.getMetaRegionLocation(this.watcher));
  for (RegionState.State state : RegionState.State.values()) {
    if (state.equals(RegionState.State.OPEN))
      continue;
    MetaTableLocator.setMetaLocation(this.watcher, SN, state);
    assertNull(mtl.getMetaRegionLocation(this.watcher));
    assertEquals(state, MetaTableLocator.getMetaRegionState(this.watcher).getState());
  }
  MetaTableLocator.setMetaLocation(this.watcher, SN, RegionState.State.OPEN);
  assertEquals(SN, mtl.getMetaRegionLocation(this.watcher));
  assertEquals(RegionState.State.OPEN,
    MetaTableLocator.getMetaRegionState(this.watcher).getState());

  mtl.deleteMetaLocation(this.watcher);
  assertNull(MetaTableLocator.getMetaRegionState(this.watcher).getServerName());
  assertEquals(RegionState.State.OFFLINE,
      MetaTableLocator.getMetaRegionState(this.watcher).getState());
  assertNull(mtl.getMetaRegionLocation(this.watcher));
}
 

import org.apache.hadoop.hbase.master.RegionState; //导入方法依赖的package包/类
static void addRegionStateToPut(Put put, RegionState.State  state) throws IOException {
  put.add(CellBuilderFactory.create(CellBuilderType.SHALLOW_COPY)
      .setRow(put.getRow())
      .setFamily(HConstants.CATALOG_FAMILY)
      .setQualifier(getRegionStateColumn())
      .setTimestamp(put.getTimeStamp())
      .setType(Cell.Type.Put)
      .setValue(Bytes.toBytes(state.name()))
      .build());
}
 

import org.apache.hadoop.hbase.master.RegionState; //导入方法依赖的package包/类
private Pair<RegionState.State, ServerName> getStateAndServerName(
    ZooKeeperProtos.MetaRegionServer proto) {
  RegionState.State state;
  if (proto.hasState()) {
    state = RegionState.State.convert(proto.getState());
  } else {
    state = RegionState.State.OPEN;
  }
  HBaseProtos.ServerName snProto = proto.getServer();
  return Pair.newPair(state,
    ServerName.valueOf(snProto.getHostName(), snProto.getPort(), snProto.getStartCode()));
}
 

import org.apache.hadoop.hbase.master.RegionState; //导入方法依赖的package包/类
/**
 * Sets the location of <code>hbase:meta</code> in ZooKeeper to the specified server address.
 * @param zookeeper reference to the {@link ZKWatcher} which also contains configuration and
 *                  operation
 * @param serverName the name of the server
 * @param replicaId the ID of the replica
 * @param state the state of the region
 * @throws KeeperException if a ZooKeeper operation fails
 */
public static void setMetaLocation(ZKWatcher zookeeper, ServerName serverName, int replicaId,
    RegionState.State state) throws KeeperException {
  if (serverName == null) {
    LOG.warn("Tried to set null ServerName in hbase:meta; skipping -- ServerName required");
    return;
  }
  LOG.info("Setting hbase:meta (replicaId=" + replicaId + ") location in ZooKeeper as " +
    serverName);
  // Make the MetaRegionServer pb and then get its bytes and save this as
  // the znode content.
  MetaRegionServer pbrsr = MetaRegionServer.newBuilder()
    .setServer(ProtobufUtil.toServerName(serverName))
    .setRpcVersion(HConstants.RPC_CURRENT_VERSION)
    .setState(state.convert()).build();
  byte[] data = ProtobufUtil.prependPBMagic(pbrsr.toByteArray());
  try {
    ZKUtil.setData(zookeeper,
        zookeeper.znodePaths.getZNodeForReplica(replicaId), data);
  } catch(KeeperException.NoNodeException nne) {
    if (replicaId == RegionInfo.DEFAULT_REPLICA_ID) {
      LOG.debug("META region location doesn't exist, create it");
    } else {
      LOG.debug("META region location doesn't exist for replicaId=" + replicaId +
          ", create it");
    }
    ZKUtil.createAndWatch(zookeeper, zookeeper.znodePaths.getZNodeForReplica(replicaId), data);
  }
}