Search in sources :

Example 1 with Chain

use of com.google.bigtable.v2.RowFilter.Chain in project simple-bigtable by spotify.

the class ColumnsReadImplTest method testStartQualifierClosed.

@Test
public void testStartQualifierClosed() throws Exception {
    final ColumnRange columnRange = ColumnRange.newBuilder().setStartQualifierClosed(ByteString.copyFromUtf8("qual")).build();
    final ColumnsWithinFamilyRead.ReadImpl read = (ColumnsWithinFamilyRead.ReadImpl) columnsRead.startQualifierClosed(columnRange.getStartQualifierClosed());
    final ReadRowsRequest.Builder readRequest = read.readRequest();
    verifyReadRequest(readRequest);
    final RowFilter.Chain chain = readRequest.getFilter().getChain();
    assertEquals(3, chain.getFiltersCount());
    assertEquals(columnRange, chain.getFilters(2).getColumnRangeFilter());
}
Also used : RowFilter(com.google.bigtable.v2.RowFilter) ColumnsWithinFamilyRead(com.spotify.bigtable.read.ReadColumns.ColumnsWithinFamilyRead) ReadRowsRequest(com.google.bigtable.v2.ReadRowsRequest) ColumnRange(com.google.bigtable.v2.ColumnRange) Test(org.junit.Test)

Example 2 with Chain

use of com.google.bigtable.v2.RowFilter.Chain in project simple-bigtable by spotify.

the class ColumnsReadImplTest method testEndQualifierOpen.

@Test
public void testEndQualifierOpen() throws Exception {
    final ColumnRange columnRange = ColumnRange.newBuilder().setEndQualifierOpen(ByteString.copyFromUtf8("qual")).build();
    final ColumnsWithinFamilyRead.ReadImpl read = (ColumnsWithinFamilyRead.ReadImpl) columnsRead.endQualifierOpen(columnRange.getEndQualifierOpen());
    final ReadRowsRequest.Builder readRequest = read.readRequest();
    verifyReadRequest(readRequest);
    final RowFilter.Chain chain = readRequest.getFilter().getChain();
    assertEquals(3, chain.getFiltersCount());
    assertEquals(columnRange, chain.getFilters(2).getColumnRangeFilter());
}
Also used : RowFilter(com.google.bigtable.v2.RowFilter) ColumnsWithinFamilyRead(com.spotify.bigtable.read.ReadColumns.ColumnsWithinFamilyRead) ReadRowsRequest(com.google.bigtable.v2.ReadRowsRequest) ColumnRange(com.google.bigtable.v2.ColumnRange) Test(org.junit.Test)

Example 3 with Chain

use of com.google.bigtable.v2.RowFilter.Chain in project simple-bigtable by spotify.

the class ColumnsReadImplTest method testEndQualifierClosed.

@Test
public void testEndQualifierClosed() throws Exception {
    final ColumnRange columnRange = ColumnRange.newBuilder().setEndQualifierClosed(ByteString.copyFromUtf8("qual")).build();
    final ColumnsWithinFamilyRead.ReadImpl read = (ColumnsWithinFamilyRead.ReadImpl) columnsRead.endQualifierClosed(columnRange.getEndQualifierClosed());
    final ReadRowsRequest.Builder readRequest = read.readRequest();
    verifyReadRequest(readRequest);
    final RowFilter.Chain chain = readRequest.getFilter().getChain();
    assertEquals(3, chain.getFiltersCount());
    assertEquals(columnRange, chain.getFilters(2).getColumnRangeFilter());
}
Also used : RowFilter(com.google.bigtable.v2.RowFilter) ColumnsWithinFamilyRead(com.spotify.bigtable.read.ReadColumns.ColumnsWithinFamilyRead) ReadRowsRequest(com.google.bigtable.v2.ReadRowsRequest) ColumnRange(com.google.bigtable.v2.ColumnRange) Test(org.junit.Test)

Example 4 with Chain

use of com.google.bigtable.v2.RowFilter.Chain in project simple-bigtable by spotify.

the class ColumnsReadImplTest method testMultipleFilters.

@Test
public void testMultipleFilters() throws Exception {
    final ColumnRange startQualRange = ColumnRange.newBuilder().setStartQualifierClosed(ByteString.copyFromUtf8("start")).build();
    final ColumnRange endQualRange = ColumnRange.newBuilder().setEndQualifierOpen(ByteString.copyFromUtf8("end")).build();
    final ColumnsWithinFamilyRead.ReadImpl read = (ColumnsWithinFamilyRead.ReadImpl) this.columnsRead.startQualifierClosed(startQualRange.getStartQualifierClosed()).endQualifierOpen(endQualRange.getEndQualifierOpen());
    final ReadRowsRequest.Builder readRequest = read.readRequest();
    verifyReadRequest(readRequest);
    final RowFilter.Chain chain = readRequest.getFilter().getChain();
    assertEquals(4, chain.getFiltersCount());
    assertEquals(startQualRange, chain.getFilters(2).getColumnRangeFilter());
    assertEquals(endQualRange, chain.getFilters(3).getColumnRangeFilter());
}
Also used : RowFilter(com.google.bigtable.v2.RowFilter) ColumnsWithinFamilyRead(com.spotify.bigtable.read.ReadColumns.ColumnsWithinFamilyRead) ReadRowsRequest(com.google.bigtable.v2.ReadRowsRequest) ColumnRange(com.google.bigtable.v2.ColumnRange) Test(org.junit.Test)

Example 5 with Chain

use of com.google.bigtable.v2.RowFilter.Chain in project simple-bigtable by spotify.

the class ColumnsReadImplTest method testStartQualifierOpen.

@Test
public void testStartQualifierOpen() throws Exception {
    final ColumnRange columnRange = ColumnRange.newBuilder().setStartQualifierOpen(ByteString.copyFromUtf8("qual")).build();
    final ColumnsWithinFamilyRead.ReadImpl read = (ColumnsWithinFamilyRead.ReadImpl) columnsRead.startQualifierOpen(columnRange.getStartQualifierOpen());
    final ReadRowsRequest.Builder readRequest = read.readRequest();
    verifyReadRequest(readRequest);
    final RowFilter.Chain chain = readRequest.getFilter().getChain();
    assertEquals(3, chain.getFiltersCount());
    assertEquals(columnRange, chain.getFilters(2).getColumnRangeFilter());
}
Also used : RowFilter(com.google.bigtable.v2.RowFilter) ColumnsWithinFamilyRead(com.spotify.bigtable.read.ReadColumns.ColumnsWithinFamilyRead) ReadRowsRequest(com.google.bigtable.v2.ReadRowsRequest) ColumnRange(com.google.bigtable.v2.ColumnRange) Test(org.junit.Test)

Aggregations

ReadRowsRequest (com.google.bigtable.v2.ReadRowsRequest)7 RowFilter (com.google.bigtable.v2.RowFilter)6 HeaderTracerUnaryCallable (com.google.cloud.bigtable.data.v2.stub.metrics.HeaderTracerUnaryCallable)6 Test (org.junit.Test)6 ColumnRange (com.google.bigtable.v2.ColumnRange)5 ImmutableMap (com.google.common.collect.ImmutableMap)5 ColumnsWithinFamilyRead (com.spotify.bigtable.read.ReadColumns.ColumnsWithinFamilyRead)5 Map (java.util.Map)5 StatsHeadersUnaryCallable (com.google.cloud.bigtable.data.v2.stub.metrics.StatsHeadersUnaryCallable)4 ByteString (com.google.protobuf.ByteString)4 SpanName (com.google.api.gax.tracing.SpanName)3 TracedUnaryCallable (com.google.api.gax.tracing.TracedUnaryCallable)2 CheckAndMutateRowRequest (com.google.bigtable.v2.CheckAndMutateRowRequest)2 CheckAndMutateRowResponse (com.google.bigtable.v2.CheckAndMutateRowResponse)2 MutateRowsRequest (com.google.bigtable.v2.MutateRowsRequest)2 Chain (com.google.bigtable.v2.RowFilter.Chain)2 Query (com.google.cloud.bigtable.data.v2.models.Query)2 TracedBatcherUnaryCallable (com.google.cloud.bigtable.data.v2.stub.metrics.TracedBatcherUnaryCallable)2 ReadRowsUserCallable (com.google.cloud.bigtable.data.v2.stub.readrows.ReadRowsUserCallable)2 ImmutableList (com.google.common.collect.ImmutableList)2