@Override
public void close() {
if (LOG.isTraceEnabled()) LOG.trace("close() -- ENTRY txID: " + ts.getTransactionId());
if (closed) {
if (LOG.isTraceEnabled())
LOG.trace("close() already closed -- EXIT txID: " + ts.getTransactionId());
return;
}
this.closed = true;
if (this.interrupted) {
if (LOG.isDebugEnabled())
LOG.debug("close() resetting connection, txID: " + ts.getTransactionId());
try {
ttable.resetConnection();
} catch (IOException e) {
if (LOG.isErrorEnabled())
LOG.error("close() unable to reset connection, txID: " + ts.getTransactionId());
return;
}
this.interrupted = false;
}
TrxRegionProtos.CloseScannerRequest.Builder requestBuilder = CloseScannerRequest.newBuilder();
requestBuilder.setTransactionId(ts.getTransactionId());
requestBuilder.setRegionName(ByteString.copyFromUtf8(currentRegion.getRegionNameAsString()));
requestBuilder.setScannerId(scannerID);
TrxRegionProtos.CloseScannerRequest closeRequest = requestBuilder.build();
try {
CoprocessorRpcChannel channel = ttable.coprocessorService(this.currentBeginKey);
TrxRegionService.BlockingInterface trxService = TrxRegionService.newBlockingStub(channel);
TrxRegionProtos.CloseScannerResponse response = trxService.closeScanner(null, closeRequest);
String exception = response.getException();
if (response.getHasException()) {
String errMsg =
"closeScanner encountered Exception txID: "
+ ts.getTransactionId()
+ " Exception: "
+ exception;
LOG.error(errMsg);
}
} catch (ServiceException se) {
this.interrupted = true;
this.closed = false;
} catch (Throwable e) {
String errMsg =
"CloseScanner error on coprocessor call, scannerID: " + this.scannerID + " " + e;
LOG.error(errMsg);
}
if (LOG.isTraceEnabled()) LOG.trace("close() -- EXIT txID: " + ts.getTransactionId());
}
@Override
public Result next() throws IOException {
// if (LOG.isTraceEnabled()) LOG.trace("next -- ENTRY txID: " + ts.getTransactionId() + " cache
// size: " + cache.size());
if (cache.size() == 0) {
if (LOG.isTraceEnabled())
LOG.trace("next -- cache.size() == 0 txID: " + ts.getTransactionId());
if (this.hasMore) {
if (LOG.isTraceEnabled())
LOG.trace("next() before coprocessor PerformScan call txID: " + ts.getTransactionId());
final long nextCallSeqInput = this.nextCallSeq;
TrxRegionProtos.PerformScanRequest.Builder requestBuilder = PerformScanRequest.newBuilder();
requestBuilder.setTransactionId(ts.getTransactionId());
requestBuilder.setRegionName(
ByteString.copyFromUtf8(currentRegion.getRegionNameAsString()));
requestBuilder.setScannerId(scannerID);
requestBuilder.setNumberOfRows(nbRows);
if (doNotCloseOnLast) requestBuilder.setCloseScanner(false);
else requestBuilder.setCloseScanner(true);
requestBuilder.setNextCallSeq(nextCallSeqInput);
TrxRegionProtos.PerformScanRequest perfScanRequest = requestBuilder.build();
TrxRegionProtos.PerformScanResponse response;
try {
CoprocessorRpcChannel channel = ttable.coprocessorService(this.currentBeginKey);
TrxRegionService.BlockingInterface trxService = TrxRegionService.newBlockingStub(channel);
response = trxService.performScan(null, perfScanRequest);
String exception = response.getException();
if (response.getHasException()) {
String errMsg =
"performScan encountered Exception txID: "
+ ts.getTransactionId()
+ " Exception: "
+ exception;
LOG.error(errMsg);
throw new IOException(errMsg);
}
} catch (ServiceException se) {
this.interrupted = true;
if (LOG.isDebugEnabled())
LOG.debug(
"PerformScan encountered Service Exception, scannerID: "
+ this.scannerID
+ " "
+ se);
return null;
} catch (Throwable e) {
String errMsg =
"PerformScan error on coprocessor call, scannerID: " + this.scannerID + " " + e;
if (LOG.isErrorEnabled()) LOG.error(errMsg);
throw new IOException(errMsg);
}
int count;
org.apache.hadoop.hbase.protobuf.generated.ClientProtos.Result result;
this.nextCallSeq = response.getNextCallSeq();
count = response.getResultCount();
if (LOG.isTraceEnabled())
LOG.trace("next() nextCallSeq: " + this.nextCallSeq + " count: " + count);
if (count == 0) {
this.hasMore = false;
} else {
for (int i = 0; i < count; i++) {
result = response.getResult(i);
if (result != null) {
cache.add(ProtobufUtil.toResult(result));
}
this.hasMore = response.getHasMore();
if (LOG.isTraceEnabled())
LOG.trace(
" PerformScan response count "
+ count
+ ", hasMore is "
+ hasMore
+ ", result "
+ result);
}
}
} else {
if (LOG.isTraceEnabled()) LOG.trace("hasMore is false");
if (nextScanner(false)) {
if (LOG.isTraceEnabled())
LOG.trace("next(), nextScanner == true nextCallSeq: " + this.nextCallSeq);
this.hasMore = true;
return next();
} else {
if (LOG.isTraceEnabled()) LOG.trace("next(), nextScanner == false");
this.moreScanners = false;
return null;
}
}
}
if (cache.size() > 0) {
// if(LOG.isTraceEnabled()) LOG.trace("next() returning cache.poll()");
return cache.poll();
} else if (this.moreScanners) {
if (LOG.isTraceEnabled()) LOG.trace("next() more scanners");
return next();
} else {
if (LOG.isTraceEnabled()) LOG.trace("next() returning null");
return null;
}
}
protected boolean nextScanner(final boolean done) throws IOException {
if (LOG.isTraceEnabled()) LOG.trace("nextScanner() -- ENTRY txID: " + ts.getTransactionId());
if (this.interrupted) {
if (LOG.isDebugEnabled())
LOG.debug("nextScanner() resetting connection, txID: " + ts.getTransactionId());
ttable.resetConnection();
this.interrupted = false;
}
if (this.currentBeginKey != null) {
if (LOG.isTraceEnabled())
LOG.trace("nextScanner() currentBeginKey != null txID: " + ts.getTransactionId());
if (doNotCloseOnLast) close();
if ((this.currentEndKey == HConstants.EMPTY_END_ROW)
|| Bytes.equals(this.currentEndKey, HConstants.EMPTY_BYTE_ARRAY)
|| checkScanStopRow(this.currentEndKey)
|| done) {
if (LOG.isTraceEnabled()) LOG.trace("endKey: " + Bytes.toString(this.currentEndKey));
if (LOG.isTraceEnabled())
LOG.trace("nextScanner() -- EXIT -- returning false txID: " + ts.getTransactionId());
this.moreScanners = false;
return false;
} else this.currentBeginKey = TransactionManager.binaryIncrementPos(this.currentEndKey, 1);
} else {
// First call to nextScanner
this.currentBeginKey = this.scan.getStartRow();
}
this.currentRegion = ttable.getRegionLocation(this.currentBeginKey).getRegionInfo();
this.currentEndKey = this.currentRegion.getEndKey();
if (LOG.isTraceEnabled()) LOG.trace("Region Info: " + currentRegion.getRegionNameAsString());
if (this.currentEndKey != HConstants.EMPTY_END_ROW)
this.currentEndKey = TransactionManager.binaryIncrementPos(currentRegion.getEndKey(), -1);
this.closed = false;
TrxRegionProtos.OpenScannerRequest.Builder requestBuilder = OpenScannerRequest.newBuilder();
requestBuilder.setTransactionId(ts.getTransactionId());
requestBuilder.setRegionName(ByteString.copyFromUtf8(currentRegion.getRegionNameAsString()));
requestBuilder.setScan(ProtobufUtil.toScan(scan));
TrxRegionProtos.OpenScannerRequest openRequest = requestBuilder.build();
try {
CoprocessorRpcChannel channel = ttable.coprocessorService(this.currentBeginKey);
TrxRegionService.BlockingInterface trxService = TrxRegionService.newBlockingStub(channel);
TrxRegionProtos.OpenScannerResponse response = trxService.openScanner(null, openRequest);
String exception = response.getException();
if (response.getHasException()) {
String errMsg =
"nextScanner encountered Exception txID: "
+ ts.getTransactionId()
+ " Exception: "
+ exception;
LOG.error(errMsg);
throw new IOException(errMsg);
}
this.scannerID = response.getScannerId();
} catch (ServiceException se) {
this.interrupted = true;
String errMsg =
"OpenScanner error encountered Service Exception, scannerID: "
+ this.scannerID
+ " "
+ se;
LOG.error(errMsg);
throw new IOException(errMsg);
} catch (Throwable e) {
String errMsg =
"OpenScanner error on coprocessor call, scannerID: " + this.scannerID + " " + e;
LOG.error(errMsg);
throw new IOException(errMsg);
}
this.nextCallSeq = 0;
if (LOG.isTraceEnabled())
LOG.trace("nextScanner() -- EXIT -- returning true txID: " + ts.getTransactionId());
return true;
}