Examples with Row com.google.api.ads.admanager.axis.v202205.Row used on opensource projects

Example 6 with Row

use of com.google.api.ads.admanager.axis.v202205.Row in project gateway-service by hypertrace.

the class EdsEntityUpdater method update.

public UpdateEntityResponse.Builder update(UpdateEntityRequest updateRequest, UpdateExecutionContext updateExecutionContext) {
    UpdateEntityResponse.Builder responseBuilder = UpdateEntityResponse.newBuilder();
    EntityUpdateRequest eqsUpdateRequest = convertToEqsUpdateRequest(updateRequest);
    if (LOG.isDebugEnabled()) {
        LOG.debug("Sending update request to EDS  ======== \n {}", eqsUpdateRequest);
    }
    Iterator<ResultSetChunk> resultSetChunkIterator = eqsClient.update(eqsUpdateRequest, updateExecutionContext.getRequestHeaders());
    if (!resultSetChunkIterator.hasNext()) {
        return responseBuilder;
    }
    ResultSetChunk chunk = resultSetChunkIterator.next();
    if (LOG.isDebugEnabled()) {
        LOG.debug("Received chunk: " + chunk.toString());
    }
    if (chunk.getRowCount() == 0) {
        return responseBuilder;
    }
    if (chunk.getRowCount() > 1) {
        LOG.warn("Received more than 1 row back. Only the first out of {} rows will be returned in the response", chunk.getRowCount());
    }
    Map<String, AttributeMetadata> resultAttributeMetadata = AttributeMetadataUtil.remapAttributeMetadataByResultKey(updateRequest.getSelectionList(), updateExecutionContext.getAttributeMetadata());
    Row row = chunk.getRow(0);
    Entity.Builder entityBuilder = Entity.newBuilder().setEntityType(updateRequest.getEntityType());
    for (int i = 0; i < chunk.getResultSetMetadata().getColumnMetadataCount(); i++) {
        String resultName = chunk.getResultSetMetadata().getColumnMetadata(i).getColumnName();
        entityBuilder.putAttribute(resultName, EntityServiceAndGatewayServiceConverter.convertQueryValueToGatewayValue(row.getColumn(i), resultAttributeMetadata.get(resultName)));
    }
    responseBuilder.setEntity(entityBuilder);
    return responseBuilder;
}

Example 7 with Row

use of com.google.api.ads.admanager.axis.v202205.Row in project beam by apache.

the class BigtableIOTest method testReadWithReaderStartFailed.

@Test
public void testReadWithReaderStartFailed() throws IOException {
    FailureBigtableService failureService = new FailureBigtableService(FailureOptions.builder().setFailAtStart(true).build());
    BigtableConfig failureConfig = BigtableConfig.builder().setValidate(true).setBigtableService(failureService).build();
    final String table = "TEST-TABLE";
    final int numRows = 100;
    makeTableData(failureService, table, numRows);
    BigtableSource source = new BigtableSource(failureConfig.withTableId(StaticValueProvider.of(table)), BigtableReadOptions.builder().setKeyRanges(ALL_KEY_RANGE).build(), null);
    BoundedReader<Row> reader = source.createReader(TestPipeline.testingPipelineOptions());
    thrown.expect(IOException.class);
    thrown.expectMessage("Fake IOException at start()");
    reader.start();
}

Example 8 with Row

use of com.google.api.ads.admanager.axis.v202205.Row in project beam by apache.

the class BigtableIOTest method testGetSplitPointsConsumed.

@Test
public void testGetSplitPointsConsumed() throws Exception {
    final String table = "TEST-TABLE";
    final int numRows = 100;
    int splitPointsConsumed = 0;
    makeTableData(table, numRows);
    BigtableSource source = new BigtableSource(config.withTableId(StaticValueProvider.of(table)), BigtableReadOptions.builder().setKeyRanges(ALL_KEY_RANGE).build(), null);
    BoundedReader<Row> reader = source.createReader(TestPipeline.testingPipelineOptions());
    reader.start();
    // Started, 0 split points consumed
    assertEquals("splitPointsConsumed starting", splitPointsConsumed, reader.getSplitPointsConsumed());
    // Split points consumed increases for each row read
    while (reader.advance()) {
        assertEquals("splitPointsConsumed advancing", ++splitPointsConsumed, reader.getSplitPointsConsumed());
    }
    // Reader marked as done, 100 split points consumed
    assertEquals("splitPointsConsumed done", numRows, reader.getSplitPointsConsumed());
    reader.close();
}

Example 9 with Row

use of com.google.api.ads.admanager.axis.v202205.Row in project beam by apache.

the class BigtableIOTest method testReadWithReaderAdvanceFailed.

@Test
public void testReadWithReaderAdvanceFailed() throws IOException {
    FailureBigtableService failureService = new FailureBigtableService(FailureOptions.builder().setFailAtAdvance(true).build());
    BigtableConfig failureConfig = BigtableConfig.builder().setValidate(true).setBigtableService(failureService).build();
    final String table = "TEST-TABLE";
    final int numRows = 100;
    makeTableData(failureService, table, numRows);
    BigtableSource source = new BigtableSource(failureConfig.withTableId(StaticValueProvider.of(table)), BigtableReadOptions.builder().setKeyRanges(ALL_KEY_RANGE).build(), null);
    BoundedReader<Row> reader = source.createReader(TestPipeline.testingPipelineOptions());
    thrown.expect(IOException.class);
    thrown.expectMessage("Fake IOException at advance()");
    reader.start();
    reader.advance();
}

Example 10 with Row

use of com.google.api.ads.admanager.axis.v202205.Row in project beam by apache.

the class BigtableIOTest method makeTableData.

/**
 * Helper function to create a table and return the rows that it created with custom service.
 */
private static List<Row> makeTableData(FakeBigtableService fakeService, String tableId, int numRows) {
    fakeService.createTable(tableId);
    Map<ByteString, ByteString> testData = fakeService.getTable(tableId);
    List<Row> testRows = new ArrayList<>(numRows);
    for (int i = 0; i < numRows; ++i) {
        ByteString key = ByteString.copyFromUtf8(String.format("key%09d", i));
        ByteString value = ByteString.copyFromUtf8(String.format("value%09d", i));
        testData.put(key, value);
        testRows.add(makeRow(key, value));
    }
    return testRows;
}