Examples with RegionLocations org.apache.hadoop.hbase.RegionLocations used on opensource projects

Example 6 with RegionLocations

use of org.apache.hadoop.hbase.RegionLocations in project hbase by apache.

the class HBaseAdmin method getRegion.

/**
   * @param regionName Name of a region.
   * @return a pair of HRegionInfo and ServerName if <code>regionName</code> is
   *  a verified region name (we call {@link
   *  MetaTableAccessor#getRegionLocation(Connection, byte[])}
   *  else null.
   * Throw IllegalArgumentException if <code>regionName</code> is null.
   * @throws IOException
   */
Pair<HRegionInfo, ServerName> getRegion(final byte[] regionName) throws IOException {
    if (regionName == null) {
        throw new IllegalArgumentException("Pass a table name or region name");
    }
    Pair<HRegionInfo, ServerName> pair = MetaTableAccessor.getRegion(connection, regionName);
    if (pair == null) {
        final AtomicReference<Pair<HRegionInfo, ServerName>> result = new AtomicReference<>(null);
        final String encodedName = Bytes.toString(regionName);
        MetaTableAccessor.Visitor visitor = new MetaTableAccessor.Visitor() {

            @Override
            public boolean visit(Result data) throws IOException {
                HRegionInfo info = MetaTableAccessor.getHRegionInfo(data);
                if (info == null) {
                    LOG.warn("No serialized HRegionInfo in " + data);
                    return true;
                }
                RegionLocations rl = MetaTableAccessor.getRegionLocations(data);
                boolean matched = false;
                ServerName sn = null;
                if (rl != null) {
                    for (HRegionLocation h : rl.getRegionLocations()) {
                        if (h != null && encodedName.equals(h.getRegionInfo().getEncodedName())) {
                            sn = h.getServerName();
                            info = h.getRegionInfo();
                            matched = true;
                        }
                    }
                }
                if (!matched)
                    return true;
                result.set(new Pair<>(info, sn));
                // found the region, stop
                return false;
            }
        };
        MetaTableAccessor.fullScanRegions(connection, visitor);
        pair = result.get();
    }
    return pair;
}

Example 7 with RegionLocations

use of org.apache.hadoop.hbase.RegionLocations in project hbase by apache.

the class HRegionLocator method listRegionLocations.

@VisibleForTesting
List<RegionLocations> listRegionLocations() throws IOException {
    final List<RegionLocations> regions = new ArrayList<>();
    MetaTableAccessor.Visitor visitor = new MetaTableAccessor.TableVisitorBase(tableName) {

        @Override
        public boolean visitInternal(Result result) throws IOException {
            RegionLocations locations = MetaTableAccessor.getRegionLocations(result);
            if (locations == null)
                return true;
            regions.add(locations);
            return true;
        }
    };
    MetaTableAccessor.scanMetaForTableRegions(connection, visitor, tableName);
    return regions;
}

Example 8 with RegionLocations

use of org.apache.hadoop.hbase.RegionLocations in project hbase by apache.

the class HRegionLocator method getAllRegionLocations.

@Override
public List<HRegionLocation> getAllRegionLocations() throws IOException {
    TableName tableName = getName();
    List<Pair<HRegionInfo, ServerName>> locations = MetaTableAccessor.getTableRegionsAndLocations(this.connection, tableName);
    ArrayList<HRegionLocation> regions = new ArrayList<>(locations.size());
    for (Pair<HRegionInfo, ServerName> entry : locations) {
        regions.add(new HRegionLocation(entry.getFirst(), entry.getSecond()));
    }
    if (regions.size() > 0) {
        connection.cacheLocation(tableName, new RegionLocations(regions));
    }
    return regions;
}

Example 9 with RegionLocations

use of org.apache.hadoop.hbase.RegionLocations in project hbase by apache.

the class MetaCache method clearCache.

/**
   * Deletes the cached location of the region if necessary, based on some error from source.
   * @param hri The region in question.
   */
public void clearCache(HRegionInfo hri) {
    ConcurrentMap<byte[], RegionLocations> tableLocations = getTableLocations(hri.getTable());
    RegionLocations regionLocations = tableLocations.get(hri.getStartKey());
    if (regionLocations != null) {
        HRegionLocation oldLocation = regionLocations.getRegionLocation(hri.getReplicaId());
        if (oldLocation == null)
            return;
        RegionLocations updatedLocations = regionLocations.remove(oldLocation);
        boolean removed;
        if (updatedLocations != regionLocations) {
            if (updatedLocations.isEmpty()) {
                removed = tableLocations.remove(hri.getStartKey(), regionLocations);
            } else {
                removed = tableLocations.replace(hri.getStartKey(), regionLocations, updatedLocations);
            }
            if (removed) {
                if (metrics != null) {
                    metrics.incrMetaCacheNumClearRegion();
                }
                if (LOG.isTraceEnabled()) {
                    LOG.trace("Removed " + oldLocation + " from cache");
                }
            }
        }
    }
}

Example 10 with RegionLocations

use of org.apache.hadoop.hbase.RegionLocations in project hbase by apache.

the class MetaCache method clearCache.

/**
   * Delete a cached location for a table, row and server
   */
public void clearCache(final TableName tableName, final byte[] row, ServerName serverName) {
    ConcurrentMap<byte[], RegionLocations> tableLocations = getTableLocations(tableName);
    RegionLocations regionLocations = getCachedLocation(tableName, row);
    if (regionLocations != null) {
        RegionLocations updatedLocations = regionLocations.removeByServer(serverName);
        if (updatedLocations != regionLocations) {
            byte[] startKey = regionLocations.getRegionLocation().getRegionInfo().getStartKey();
            boolean removed = false;
            if (updatedLocations.isEmpty()) {
                removed = tableLocations.remove(startKey, regionLocations);
            } else {
                removed = tableLocations.replace(startKey, regionLocations, updatedLocations);
            }
            if (removed) {
                if (metrics != null) {
                    metrics.incrMetaCacheNumClearRegion();
                }
                if (LOG.isTraceEnabled()) {
                    LOG.trace("Removed locations of table: " + tableName + " ,row: " + Bytes.toString(row) + " mapping to server: " + serverName + " from cache");
                }
            }
        }
    }
}