com.google.cloud.bigtable.hbase.BigtableRegionLocator Maven / Gradle / Ivy
Show all versions of bigtable-hbase Show documentation
/*
* Copyright 2015 Google LLC
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.google.cloud.bigtable.hbase;
import com.google.api.core.InternalApi;
import com.google.cloud.bigtable.hbase.util.Logger;
import com.google.cloud.bigtable.hbase.wrappers.BigtableHBaseSettings;
import com.google.cloud.bigtable.hbase.wrappers.DataClientWrapper;
import java.io.IOException;
import java.util.List;
import java.util.concurrent.ExecutionException;
import org.apache.hadoop.hbase.HRegionInfo;
import org.apache.hadoop.hbase.HRegionLocation;
import org.apache.hadoop.hbase.TableName;
import org.apache.hadoop.hbase.client.RegionLocator;
import org.apache.hadoop.hbase.util.Bytes;
import org.apache.hadoop.hbase.util.Pair;
/**
* BigtableRegionLocator class.
*
* For internal use only - public for technical reasons.
*/
@InternalApi("For internal usage only")
public abstract class BigtableRegionLocator extends AbstractBigtableRegionLocator
implements RegionLocator {
/** Constant LOG */
protected static final Logger LOG = new Logger(BigtableRegionLocator.class);
/**
* Constructor for BigtableRegionLocator.
*
* @param tableName a {@link TableName} object.
* @param settings a {@link BigtableHBaseSettings} object.
* @param client a {@link DataClientWrapper} object.
*/
public BigtableRegionLocator(
TableName tableName, BigtableHBaseSettings settings, DataClientWrapper client) {
super(tableName, settings, client);
}
/** {@inheritDoc} */
@Override
public HRegionLocation getRegionLocation(byte[] row) throws IOException {
return getRegionLocation(row, false);
}
/** {@inheritDoc} */
@Override
public HRegionLocation getRegionLocation(byte[] row, boolean reload) throws IOException {
List regions = getRegions(reload);
return findRegion(regions, row);
}
private HRegionLocation findRegion(List regions, byte[] row) throws IOException {
int low = 0;
int high = regions.size() - 1;
while (low <= high) {
int mid = (low + high) >>> 1;
HRegionLocation regionLocation = regions.get(mid);
HRegionInfo regionInfo = regionLocation.getRegionInfo();
// This isn't the last region (endKey != "") and row key is greater than the current bound
if (regionInfo.getEndKey().length > 0 && Bytes.compareTo(row, regionInfo.getEndKey()) >= 0) {
low = mid + 1;
} else if (Bytes.compareTo(row, regionInfo.getStartKey()) < 0) {
high = mid - 1;
} else {
return regionLocation;
}
}
// Should never happen because the regions are contiguous
throw new IOException("Region not found for row: " + Bytes.toStringBinary(row));
}
private List getRegions(boolean reload) throws IOException {
try {
return getRegionsAsync(reload).get();
} catch (InterruptedException e) {
Thread.interrupted();
throw new IOException("getRegionLocation was interrupted");
} catch (ExecutionException e) {
if (e.getCause() instanceof IOException) {
throw (IOException) e.getCause();
} else {
throw new IOException("getRegionLocation ExecutionException", e);
}
}
}
/** {@inheritDoc} */
@Override
public List getAllRegionLocations() throws IOException {
return getRegions(false);
}
/** {@inheritDoc} */
@Override
public byte[][] getStartKeys() throws IOException {
return getStartEndKeys().getFirst();
}
/** {@inheritDoc} */
@Override
public byte[][] getEndKeys() throws IOException {
return getStartEndKeys().getSecond();
}
/** {@inheritDoc} */
@Override
public Pair getStartEndKeys() throws IOException {
List regions = getAllRegionLocations();
byte[][] startKeys = new byte[regions.size()][];
byte[][] endKeys = new byte[regions.size()][];
int i = 0;
for (HRegionLocation region : regions) {
startKeys[i] = region.getRegionInfo().getStartKey();
endKeys[i] = region.getRegionInfo().getEndKey();
i++;
}
return Pair.newPair(startKeys, endKeys);
}
/** {@inheritDoc} */
@Override
public TableName getName() {
return tableName;
}
/** {@inheritDoc} */
@Override
public void close() throws IOException {}
}