org.apache.beam.sdk.io.gcp.bigquery

Class BigQueryUtils

java.lang.Object

org.apache.beam.sdk.io.gcp.bigquery.BigQueryUtils

public class BigQueryUtils
extends java.lang.Object

Utility methods for BigQuery related operations.

Example: Writing to BigQuery

 PCollection<Row> rows = ...;

 rows.apply(BigQueryIO.<Row>write()
       .withSchema(BigQueryUtils.toTableSchema(rows))
       .withFormatFunction(BigQueryUtils.toTableRow())
       .to("my-project:my_dataset.my_table"));
 

Constructor Summary

Constructors

Constructor and Description
BigQueryUtils()

Method Summary

Modifier and Type	Method and Description
static Row	toBeamRow(GenericRecord record, Schema schema)
static SerializableFunction<SchemaAndRecord,Row>	toBeamRow(Schema schema) Convert SchemaAndRecord to a Beam Row.
static Row	toBeamRow(Schema rowSchema, TableSchema bqSchema, TableRow jsonBqRow) Tries to parse the JSON TableRow from BigQuery.
static SerializableFunction<Row,TableRow>	toTableRow() Convert a Beam Row to a BigQuery TableRow.
static TableRow	toTableRow(Row row) Convert a BigQuery TableRow to a Beam Row.
static TableSchema	toTableSchema(PCollection<Row> rows) Convert a Beam PCollection to a BigQuery TableSchema.
static TableSchema	toTableSchema(Schema schema) Convert a Beam Schema to a BigQuery TableSchema.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

BigQueryUtils

public BigQueryUtils()

Method Detail

toTableSchema

public static TableSchema toTableSchema(Schema schema)

Convert a Beam Schema to a BigQuery TableSchema.

toTableSchema

public static TableSchema toTableSchema(PCollection<Row> rows)

Convert a Beam PCollection to a BigQuery TableSchema.

toTableRow

public static SerializableFunction<Row,TableRow> toTableRow()

Convert a Beam Row to a BigQuery TableRow.

toBeamRow

public static SerializableFunction<SchemaAndRecord,Row> toBeamRow(Schema schema)

Convert SchemaAndRecord to a Beam Row.

toBeamRow

public static Row toBeamRow(GenericRecord record,
                            Schema schema)

toTableRow

public static TableRow toTableRow(Row row)

Convert a BigQuery TableRow to a Beam Row.

toBeamRow

public static Row toBeamRow(Schema rowSchema,
                            TableSchema bqSchema,
                            TableRow jsonBqRow)

Tries to parse the JSON TableRow from BigQuery.

Only supports basic types and arrays. Doesn't support date types.