HBASE-9885 Avoid some Result creation in protobuf conversions

git-svn-id: https://svn.apache.org/repos/asf/hbase/trunk@1539429 13f79535-47bb-0310-9956-ffa450edef68
This commit is contained in:
nkeywal 2013-11-06 18:58:37 +00:00
parent ff84a7ae16
commit 6ba3c2339b
2 changed files with 69 additions and 20 deletions

View File

@ -154,6 +154,35 @@ public final class ProtobufUtil {
private final static Map<String, Class<?>> private final static Map<String, Class<?>>
PRIMITIVES = new HashMap<String, Class<?>>(); PRIMITIVES = new HashMap<String, Class<?>>();
/**
* Many results are simple: no cell, exists true or false. To save on object creations,
* we reuse them across calls.
*/
private final static Cell[] EMPTY_CELL_ARRAY = new Cell[]{};
private final static Result EMPTY_RESULT = Result.create(EMPTY_CELL_ARRAY);
private final static Result EMPTY_RESULT_EXISTS_TRUE = Result.create(null, true);
private final static Result EMPTY_RESULT_EXISTS_FALSE = Result.create(null, false);
private final static ClientProtos.Result EMPTY_RESULT_PB;
private final static ClientProtos.Result EMPTY_RESULT_PB_EXISTS_TRUE;
private final static ClientProtos.Result EMPTY_RESULT_PB_EXISTS_FALSE;
static {
ClientProtos.Result.Builder builder = ClientProtos.Result.newBuilder();
builder.setExists(true);
EMPTY_RESULT_PB_EXISTS_TRUE = builder.build();
builder.clear();
builder.setExists(false);
EMPTY_RESULT_PB_EXISTS_FALSE = builder.build();
builder.clear();
builder.setAssociatedCellCount(0);
EMPTY_RESULT_PB = builder.build();
}
/** /**
* Dynamic class loader to load filter/comparators * Dynamic class loader to load filter/comparators
*/ */
@ -1083,16 +1112,20 @@ public final class ProtobufUtil {
* @return the converted protocol buffer Result * @return the converted protocol buffer Result
*/ */
public static ClientProtos.Result toResult(final Result result) { public static ClientProtos.Result toResult(final Result result) {
if (result.getExists() != null) {
return toResult(result.getExists());
}
Cell[] cells = result.rawCells();
if (cells == null || cells.length == 0) {
return EMPTY_RESULT_PB;
}
ClientProtos.Result.Builder builder = ClientProtos.Result.newBuilder(); ClientProtos.Result.Builder builder = ClientProtos.Result.newBuilder();
Cell [] cells = result.rawCells(); for (Cell c : cells) {
if (cells != null) { builder.addCell(toCell(c));
for (Cell c : cells) {
builder.addCell(toCell(c));
}
}
if (result.getExists() != null){
builder.setExists(result.getExists());
} }
return builder.build(); return builder.build();
} }
@ -1103,9 +1136,7 @@ public final class ProtobufUtil {
* @return the converted protocol buffer Result * @return the converted protocol buffer Result
*/ */
public static ClientProtos.Result toResult(final boolean existence) { public static ClientProtos.Result toResult(final boolean existence) {
ClientProtos.Result.Builder builder = ClientProtos.Result.newBuilder(); return existence ? EMPTY_RESULT_PB_EXISTS_TRUE : EMPTY_RESULT_PB_EXISTS_FALSE;
builder.setExists(existence);
return builder.build();
} }
/** /**
@ -1116,11 +1147,19 @@ public final class ProtobufUtil {
* @return the converted protocol buffer Result * @return the converted protocol buffer Result
*/ */
public static ClientProtos.Result toResultNoData(final Result result) { public static ClientProtos.Result toResultNoData(final Result result) {
ClientProtos.Result.Builder builder = ClientProtos.Result.newBuilder();
builder.setAssociatedCellCount(result.size());
if (result.getExists() != null){ if (result.getExists() != null){
builder.setExists(result.getExists()); return result.getExists() ? EMPTY_RESULT_PB_EXISTS_TRUE : EMPTY_RESULT_PB_EXISTS_FALSE;
} }
int size = result.size();
if (size == 0){
return EMPTY_RESULT_PB;
}
ClientProtos.Result.Builder builder = ClientProtos.Result.newBuilder();
builder.setAssociatedCellCount(size);
return builder.build(); return builder.build();
} }
@ -1132,10 +1171,14 @@ public final class ProtobufUtil {
*/ */
public static Result toResult(final ClientProtos.Result proto) { public static Result toResult(final ClientProtos.Result proto) {
if (proto.hasExists()) { if (proto.hasExists()) {
return Result.create(null, proto.getExists()); return proto.getExists() ? EMPTY_RESULT_EXISTS_TRUE : EMPTY_RESULT_EXISTS_FALSE;
} }
List<CellProtos.Cell> values = proto.getCellList(); List<CellProtos.Cell> values = proto.getCellList();
if (values.isEmpty()){
return EMPTY_RESULT;
}
List<Cell> cells = new ArrayList<Cell>(values.size()); List<Cell> cells = new ArrayList<Cell>(values.size());
for (CellProtos.Cell c : values) { for (CellProtos.Cell c : values) {
cells.add(toCell(c)); cells.add(toCell(c));
@ -1154,7 +1197,7 @@ public final class ProtobufUtil {
public static Result toResult(final ClientProtos.Result proto, final CellScanner scanner) public static Result toResult(final ClientProtos.Result proto, final CellScanner scanner)
throws IOException { throws IOException {
if (proto.hasExists()){ if (proto.hasExists()){
return Result.create(null, proto.getExists()); return proto.getExists() ? EMPTY_RESULT_EXISTS_TRUE : EMPTY_RESULT_EXISTS_FALSE;
} }
// TODO: Unit test that has some Cells in scanner and some in the proto. // TODO: Unit test that has some Cells in scanner and some in the proto.
@ -1168,13 +1211,20 @@ public final class ProtobufUtil {
} }
} }
List<CellProtos.Cell> values = proto.getCellList(); List<CellProtos.Cell> values = proto.getCellList();
if (cells == null) cells = new ArrayList<Cell>(values.size()); if (cells == null) {
for (CellProtos.Cell c: values) { if (values.isEmpty()) {
cells.add(toCell(c)); return EMPTY_RESULT;
} else {
cells = new ArrayList<Cell>(values.size());
for (CellProtos.Cell c : values) {
cells.add(toCell(c));
}
}
} }
return Result.create(cells, null); return Result.create(cells, null);
} }
/** /**
* Convert a ByteArrayComparable to a protocol buffer Comparator * Convert a ByteArrayComparable to a protocol buffer Comparator
* *

View File

@ -488,7 +488,6 @@ public final class RequestConverter {
throws IOException { throws IOException {
RegionAction.Builder builder = getRegionActionBuilderWithRegion(regionName); RegionAction.Builder builder = getRegionActionBuilderWithRegion(regionName);
ClientProtos.Action.Builder actionBuilder = ClientProtos.Action.newBuilder(); ClientProtos.Action.Builder actionBuilder = ClientProtos.Action.newBuilder();
MutationProto.Builder mutationBuilder = ClientProtos.MutationProto.newBuilder();
for (Action<R> action: actions) { for (Action<R> action: actions) {
Row row = action.getAction(); Row row = action.getAction();
actionBuilder.clear(); actionBuilder.clear();