/** Adds a key to looked up */ public Query rowKey(ByteString key) { Preconditions.checkNotNull(key, "Key can't be null."); builder.getRowsBuilder().addRowKeys(key); return this; }
RpcExpectation expectRequest(String... keys) { for (String key : keys) { requestBuilder.getRowsBuilder().addRowKeys(ByteString.copyFromUtf8(key)); } return this; }
@Test public void rowKeysTest() { Query query = Query.create(TABLE_ID) .rowKey("simple-string") .rowKey(ByteString.copyFromUtf8("byte-string")); ReadRowsRequest.Builder expectedProto = expectedProtoBuilder(); expectedProto .getRowsBuilder() .addRowKeys(ByteString.copyFromUtf8("simple-string")) .addRowKeys(ByteString.copyFromUtf8("byte-string")); ReadRowsRequest actualProto = query.toProto(requestContext); assertThat(actualProto).isEqualTo(expectedProto.build()); }
builder.getRowsBuilder().addRowRanges(rangeBuilder.build());
/** * Adds a range to be looked up. * * @param start The beginning of the range (inclusive). Can be null to represent negative * infinity. * @param end The end of the range (exclusive). Can be null to represent positive infinity. */ public Query range(ByteString start, ByteString end) { RowRange.Builder rangeBuilder = RowRange.newBuilder(); if (start != null) { rangeBuilder.setStartKeyClosed(start); } if (end != null) { rangeBuilder.setEndKeyOpen(end); } builder.getRowsBuilder().addRowRanges(rangeBuilder.build()); return this; }
/** Adds a key to looked up */ public Query rowKey(ByteString key) { Preconditions.checkNotNull(key, "Key can't be null."); builder.getRowsBuilder().addRowKeys(key); return this; }
builder.getRowsBuilder().addRowRanges(rangeBuilder.build());
RpcExpectation expectRequest(Range<String> range) { RowRange.Builder rowRange = requestBuilder.getRowsBuilder().addRowRangesBuilder();
@Test public void rowRangeTest() { Query query = Query.create(TABLE_ID) .range("simple-begin", "simple-end") .range(ByteString.copyFromUtf8("byte-begin"), ByteString.copyFromUtf8("byte-end")) .range(ByteStringRange.create("range-begin", "range-end")); Builder expectedProto = expectedProtoBuilder(); expectedProto .getRowsBuilder() .addRowRanges( RowRange.newBuilder() .setStartKeyClosed(ByteString.copyFromUtf8("simple-begin")) .setEndKeyOpen(ByteString.copyFromUtf8("simple-end"))) .addRowRanges( RowRange.newBuilder() .setStartKeyClosed(ByteString.copyFromUtf8("byte-begin")) .setEndKeyOpen(ByteString.copyFromUtf8("byte-end"))) .addRowRanges( RowRange.newBuilder() .setStartKeyClosed(ByteString.copyFromUtf8("range-begin")) .setEndKeyOpen(ByteString.copyFromUtf8("range-end"))); ReadRowsRequest actualProto = query.toProto(requestContext); assertThat(actualProto).isEqualTo(expectedProto.build()); }
/** * Adds a range to be looked up. * * @param start The beginning of the range (inclusive). Can be null to represent negative * infinity. * @param end The end of the range (exclusive). Can be null to represent positive infinity. */ public Query range(ByteString start, ByteString end) { RowRange.Builder rangeBuilder = RowRange.newBuilder(); if (start != null) { rangeBuilder.setStartKeyClosed(start); } if (end != null) { rangeBuilder.setEndKeyOpen(end); } builder.getRowsBuilder().addRowRanges(rangeBuilder.build()); return this; }