HBASE-9885 Avoid some Result creation in protobuf conversions
git-svn-id: https://svn.apache.org/repos/asf/hbase/trunk@1539692 13f79535-47bb-0310-9956-ffa450edef68
This commit is contained in:
parent
bf2144fe60
commit
ed1705d366
|
@ -10,7 +10,7 @@
|
||||||
* http://www.apache.org/licenses/LICENSE-2.0
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||||
*
|
*
|
||||||
* Unless required by applicable law or agreed to in writing, software
|
* Unless required by applicable law or agreed to in writing, software
|
||||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
* distributed under the License is distr=ibuted on an "AS IS" BASIS,
|
||||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
* See the License for the specific language governing permissions and
|
* See the License for the specific language governing permissions and
|
||||||
* limitations under the License.
|
* limitations under the License.
|
||||||
|
@ -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();
|
|
||||||
if (cells != null) {
|
|
||||||
for (Cell c : cells) {
|
for (Cell c : cells) {
|
||||||
builder.addCell(toCell(c));
|
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));
|
||||||
|
@ -1153,28 +1196,38 @@ 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()){
|
List<CellProtos.Cell> values = proto.getCellList();
|
||||||
return Result.create(null, proto.getExists());
|
|
||||||
|
if (proto.hasExists()) {
|
||||||
|
if ((values != null && !values.isEmpty()) ||
|
||||||
|
(proto.hasAssociatedCellCount() && proto.getAssociatedCellCount() > 0)) {
|
||||||
|
throw new IllegalArgumentException("bad proto: exists with cells is no allowed " + proto);
|
||||||
|
}
|
||||||
|
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.
|
||||||
List<Cell> cells = null;
|
List<Cell> cells = null;
|
||||||
if (proto.hasAssociatedCellCount()) {
|
if (proto.hasAssociatedCellCount()) {
|
||||||
int count = proto.getAssociatedCellCount();
|
int count = proto.getAssociatedCellCount();
|
||||||
cells = new ArrayList<Cell>(count);
|
cells = new ArrayList<Cell>(count + values.size());
|
||||||
for (int i = 0; i < count; i++) {
|
for (int i = 0; i < count; i++) {
|
||||||
if (!scanner.advance()) throw new IOException("Failed get " + i + " of " + count);
|
if (!scanner.advance()) throw new IOException("Failed get " + i + " of " + count);
|
||||||
cells.add(scanner.current());
|
cells.add(scanner.current());
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
List<CellProtos.Cell> values = proto.getCellList();
|
|
||||||
|
if (!values.isEmpty()){
|
||||||
if (cells == null) cells = new ArrayList<Cell>(values.size());
|
if (cells == null) cells = new ArrayList<Cell>(values.size());
|
||||||
for (CellProtos.Cell c: values) {
|
for (CellProtos.Cell c: values) {
|
||||||
cells.add(toCell(c));
|
cells.add(toCell(c));
|
||||||
}
|
}
|
||||||
return Result.create(cells, null);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
return (cells == null || cells.isEmpty()) ? EMPTY_RESULT : Result.create(cells, null);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Convert a ByteArrayComparable to a protocol buffer Comparator
|
* Convert a ByteArrayComparable to a protocol buffer Comparator
|
||||||
*
|
*
|
||||||
|
|
|
@ -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();
|
||||||
|
|
Loading…
Reference in New Issue