Examples with Row org.apache.beam.sdk.values.Row used on opensource projects

Example 96 with Row

use of org.apache.beam.sdk.values.Row in project beam by apache.

the class ZetaSqlDialectSpecTest method testStructOfStructPassthrough.

@Test
public void testStructOfStructPassthrough() {
    String sql = "SELECT * FROM table_with_struct_of_struct";
    PCollection<Row> stream = execute(sql);
    PAssert.that(stream).containsInAnyOrder(Row.withSchema(TestInput.STRUCT_OF_STRUCT).attachValues(Row.withSchema(TestInput.STRUCT_SCHEMA).attachValues(1L, "1")), Row.withSchema(TestInput.STRUCT_OF_STRUCT).attachValues(Row.withSchema(TestInput.STRUCT_SCHEMA).attachValues(2L, "2")));
    pipeline.run().waitUntilFinish(Duration.standardMinutes(PIPELINE_EXECUTION_WAITTIME_MINUTES));
}

Example 97 with Row

use of org.apache.beam.sdk.values.Row in project beam by apache.

the class ZetaSqlDialectSpecTest method testZetaSQLBitOrNull.

@Test
public void testZetaSQLBitOrNull() {
    String sql = "SELECT bit_or(CAST(x as int64)) FROM " + "(SELECT NULL x UNION ALL SELECT 5 UNION ALL SELECT 6);";
    PCollection<Row> stream = execute(sql);
    final Schema schema = Schema.builder().addInt64Field("field1").build();
    PAssert.that(stream).containsInAnyOrder(Row.withSchema(schema).addValue(7L).build());
    pipeline.run().waitUntilFinish(Duration.standardMinutes(PIPELINE_EXECUTION_WAITTIME_MINUTES));
}

Example 98 with Row

use of org.apache.beam.sdk.values.Row in project beam by apache.

the class ZetaSqlDialectSpecTest method testZetaSQLSelectFromTableOrderByNoSelectLimit.

@Test
public void testZetaSQLSelectFromTableOrderByNoSelectLimit() {
    String sql = "SELECT Value FROM KeyValue ORDER BY Key DESC LIMIT 2;";
    PCollection<Row> stream = execute(sql);
    final Schema schema = Schema.builder().addStringField("field2").build();
    PAssert.that(stream).containsInAnyOrder(Row.withSchema(schema).addValues("KeyValue234").build(), Row.withSchema(schema).addValues("KeyValue235").build());
    pipeline.run().waitUntilFinish(Duration.standardMinutes(PIPELINE_EXECUTION_WAITTIME_MINUTES));
}

Example 99 with Row

use of org.apache.beam.sdk.values.Row in project beam by apache.

the class ZetaSqlDialectSpecTest method testReplace1.

@Test
public void testReplace1() {
    String sql = "SELECT REPLACE(@p0, @p1, @p2) AS ColA";
    ImmutableMap<String, Value> params = ImmutableMap.of("p0", Value.createStringValue(""), "p1", Value.createStringValue(""), "p2", Value.createStringValue("a"));
    PCollection<Row> stream = execute(sql, params);
    final Schema schema = Schema.builder().addStringField("field1").build();
    PAssert.that(stream).containsInAnyOrder(Row.withSchema(schema).addValues("").build());
    pipeline.run().waitUntilFinish(Duration.standardMinutes(PIPELINE_EXECUTION_WAITTIME_MINUTES));
}

Example 100 with Row

use of org.apache.beam.sdk.values.Row in project beam by apache.

the class ZetaSqlDialectSpecTest method testCaseWithValueNoElse.

@Test
public void testCaseWithValueNoElse() {
    String sql = "SELECT CASE 2 WHEN 1 THEN 'not possible' WHEN 2 THEN 'seems right' END";
    PCollection<Row> stream = execute(sql);
    PAssert.that(stream).containsInAnyOrder(Row.withSchema(Schema.builder().addStringField("str_field").build()).addValue("seems right").build());
    pipeline.run().waitUntilFinish(Duration.standardMinutes(PIPELINE_EXECUTION_WAITTIME_MINUTES));
}