use of com.google.cloud.bigtable.data.v2.models.ReadModifyWriteRow in project java-bigtable-hbase by googleapis.
the class TestIncrementAdapter method testBasicRowKeyIncrement.
@Test
public void testBasicRowKeyIncrement() {
byte[] rowKey = dataHelper.randomData("rk1-");
Increment incr = new Increment(rowKey);
ReadModifyWriteRow readModifyWriteRow = ReadModifyWriteRow.create(TABLE_ID, ByteString.copyFrom(rowKey));
incrementAdapter.adapt(incr, readModifyWriteRow);
ReadModifyWriteRowRequest requestBuilder = readModifyWriteRow.toProto(requestContext);
ByteString adaptedRowKey = requestBuilder.getRowKey();
Assert.assertArrayEquals(rowKey, adaptedRowKey.toByteArray());
}
use of com.google.cloud.bigtable.data.v2.models.ReadModifyWriteRow in project java-bigtable-hbase by googleapis.
the class TestIncrementAdapter method testMultipleIncrementWithDuplicateQualifier.
@Test
public void testMultipleIncrementWithDuplicateQualifier() {
byte[] rowKey = dataHelper.randomData("rk1-");
byte[] family1 = Bytes.toBytes("family1");
byte[] qualifier1 = Bytes.toBytes("qualifier1");
long amount1 = 1234;
byte[] family2 = Bytes.toBytes("family2");
byte[] qualifier2 = Bytes.toBytes("qualifier2");
long amount2 = 4321;
long amount3 = 5000;
Increment incr = new Increment(rowKey);
incr.addColumn(family1, qualifier1, amount1);
incr.addColumn(family2, qualifier2, amount2);
incr.addColumn(family2, qualifier2, amount3);
ReadModifyWriteRow readModifyWriteRow = ReadModifyWriteRow.create(TABLE_ID, ByteString.copyFrom(rowKey));
incrementAdapter.adapt(incr, readModifyWriteRow);
ReadModifyWriteRowRequest requestBuilder = readModifyWriteRow.toProto(requestContext);
Assert.assertEquals(2, requestBuilder.getRulesCount());
ReadModifyWriteRule rule = requestBuilder.getRules(0);
Assert.assertEquals("family1", rule.getFamilyName());
Assert.assertEquals("qualifier1", rule.getColumnQualifier().toStringUtf8());
Assert.assertEquals(amount1, rule.getIncrementAmount());
rule = requestBuilder.getRules(1);
Assert.assertEquals("family2", rule.getFamilyName());
Assert.assertEquals("qualifier2", rule.getColumnQualifier().toStringUtf8());
// amount3 since it was added after amount2:
Assert.assertEquals(amount3, rule.getIncrementAmount());
}
use of com.google.cloud.bigtable.data.v2.models.ReadModifyWriteRow in project java-bigtable-hbase by googleapis.
the class TestIncrementAdapter method testIncrementTimeRange.
@Test
public void testIncrementTimeRange() throws IOException {
byte[] rowKey = dataHelper.randomData("rk1-");
Increment incr = new Increment(rowKey);
incr.setTimeRange(0, 10);
expectedException.expect(UnsupportedOperationException.class);
expectedException.expectMessage("Setting the time range in an Increment is not implemented");
ReadModifyWriteRow readModifyWriteRow = ReadModifyWriteRow.create(TABLE_ID, ByteString.copyFrom(rowKey));
incrementAdapter.adapt(incr, readModifyWriteRow);
}
use of com.google.cloud.bigtable.data.v2.models.ReadModifyWriteRow in project java-bigtable-hbase by googleapis.
the class TestAppendAdapter method testBasicRowKeyAppend.
@Test
public void testBasicRowKeyAppend() {
byte[] rowKey = dataHelper.randomData("rk1-");
Append append = new Append(rowKey);
ReadModifyWriteRow readModifyWriteRow = ReadModifyWriteRow.create(TABLE_ID, ByteString.copyFrom(rowKey));
appendAdapter.adapt(append, readModifyWriteRow);
ReadModifyWriteRowRequest request = readModifyWriteRow.toProto(requestContext);
ByteString adaptedRowKey = request.getRowKey();
Assert.assertArrayEquals(rowKey, adaptedRowKey.toByteArray());
}
use of com.google.cloud.bigtable.data.v2.models.ReadModifyWriteRow in project java-bigtable-hbase by googleapis.
the class TestAppendAdapter method testMultipleAppends.
@Test
public void testMultipleAppends() {
byte[] rowKey = dataHelper.randomData("rk1-");
byte[] family1 = Bytes.toBytes("family1");
byte[] qualifier1 = Bytes.toBytes("qualifier1");
byte[] value1 = Bytes.toBytes("value1");
byte[] family2 = Bytes.toBytes("family2");
byte[] qualifier2 = Bytes.toBytes("qualifier2");
byte[] value2 = Bytes.toBytes("value2");
Append append = new Append(rowKey);
append.add(family1, qualifier1, value1);
append.add(family2, qualifier2, value2);
ReadModifyWriteRow readModifyWriteRow = ReadModifyWriteRow.create(TABLE_ID, ByteString.copyFrom(rowKey));
appendAdapter.adapt(append, readModifyWriteRow);
ReadModifyWriteRowRequest request = readModifyWriteRow.toProto(requestContext);
List<ReadModifyWriteRule> rules = request.getRulesList();
Assert.assertEquals(2, rules.size());
Assert.assertEquals("family1", rules.get(0).getFamilyName());
Assert.assertEquals("qualifier1", rules.get(0).getColumnQualifier().toStringUtf8());
Assert.assertEquals("value1", rules.get(0).getAppendValue().toStringUtf8());
Assert.assertEquals("family2", rules.get(1).getFamilyName());
Assert.assertEquals("qualifier2", rules.get(1).getColumnQualifier().toStringUtf8());
Assert.assertEquals("value2", rules.get(1).getAppendValue().toStringUtf8());
}
Aggregations