HBASE-13312 SmallScannerCallable does not increment scan metrics
This commit is contained in:
parent
3d722dce55
commit
60b356a853
|
@ -207,6 +207,8 @@ public class ClientSmallScanner extends ClientScanner {
|
|||
} else {
|
||||
setHasMoreResultsContext(false);
|
||||
}
|
||||
// We need to update result metrics since we are overriding call()
|
||||
updateResultsMetrics(results);
|
||||
return results;
|
||||
} catch (ServiceException se) {
|
||||
throw ProtobufUtil.getRemoteException(se);
|
||||
|
|
|
@ -328,7 +328,7 @@ public class ScannerCallable extends RegionServerCallable<Result[]> {
|
|||
}
|
||||
}
|
||||
|
||||
private void updateResultsMetrics(Result[] rrs) {
|
||||
protected void updateResultsMetrics(Result[] rrs) {
|
||||
if (this.scanMetrics == null || rrs == null || rrs.length == 0) {
|
||||
return;
|
||||
}
|
||||
|
|
|
@ -5101,6 +5101,39 @@ public class TestFromClientSide {
|
|||
assertEquals("Did not access all the regions in the table", numOfRegions,
|
||||
scanMetrics.countOfRegions.get());
|
||||
|
||||
// check byte counters
|
||||
scan2 = new Scan();
|
||||
scan2.setScanMetricsEnabled(true);
|
||||
scan2.setCaching(1);
|
||||
scanner = ht.getScanner(scan2);
|
||||
int numBytes = 0;
|
||||
for (Result result : scanner.next(1)) {
|
||||
for (Cell cell: result.listCells()) {
|
||||
numBytes += CellUtil.estimatedSerializedSizeOf(cell);
|
||||
}
|
||||
}
|
||||
scanner.close();
|
||||
scanMetrics = scan2.getScanMetrics();
|
||||
assertEquals("Did not count the result bytes", numBytes,
|
||||
scanMetrics.countOfBytesInResults.get());
|
||||
|
||||
// check byte counters on a small scan
|
||||
scan2 = new Scan();
|
||||
scan2.setScanMetricsEnabled(true);
|
||||
scan2.setCaching(1);
|
||||
scan2.setSmall(true);
|
||||
scanner = ht.getScanner(scan2);
|
||||
numBytes = 0;
|
||||
for (Result result : scanner.next(1)) {
|
||||
for (Cell cell: result.listCells()) {
|
||||
numBytes += CellUtil.estimatedSerializedSizeOf(cell);
|
||||
}
|
||||
}
|
||||
scanner.close();
|
||||
scanMetrics = scan2.getScanMetrics();
|
||||
assertEquals("Did not count the result bytes", numBytes,
|
||||
scanMetrics.countOfBytesInResults.get());
|
||||
|
||||
// now, test that the metrics are still collected even if you don't call close, but do
|
||||
// run past the end of all the records
|
||||
/** There seems to be a timing issue here. Comment out for now. Fix when time.
|
||||
|
|
Loading…
Reference in New Issue