org.apache.beam.sdk.extensions.sql.meta.provider.test

Class TestBoundedTable

java.lang.Object

org.apache.beam.sdk.extensions.sql.meta.BaseBeamTable

org.apache.beam.sdk.extensions.sql.meta.SchemaBaseBeamTable

org.apache.beam.sdk.extensions.sql.meta.provider.test.TestTable

org.apache.beam.sdk.extensions.sql.meta.provider.test.TestBoundedTable

All Implemented Interfaces:

java.io.Serializable, BeamSqlTable

@Experimental
public class TestBoundedTable
extends TestTable

Mocked table for bounded data sources.

See Also:

Serialized Form

Field Summary

Fields inherited from class org.apache.beam.sdk.extensions.sql.meta.provider.test.TestTable

COUNTER

Fields inherited from class org.apache.beam.sdk.extensions.sql.meta.SchemaBaseBeamTable

schema

Constructor Summary

Constructors

Constructor and Description
TestBoundedTable(Schema beamSchema)

Method Summary

Modifier and Type	Method and Description
TestBoundedTable	addRows(java.lang.Object... args) Add rows to the builder.
PCollection<Row>	buildIOReader(PBegin begin) create a PCollection<Row> from source.
POutput	buildIOWriter(PCollection<Row> input) create a IO.write() instance to write to target.
BeamTableStatistics	getTableStatistics(PipelineOptions options) Estimates the number of rows or the rate for unbounded Tables.
PCollection.IsBounded	isBounded() Whether this table is bounded (known to be finite) or unbounded (may or may not be finite).
static TestBoundedTable	of(java.lang.Object... args) Convenient way to build a mocked bounded table.
static TestBoundedTable	of(Schema type) Build a mocked bounded table with the specified type.

Methods inherited from class org.apache.beam.sdk.extensions.sql.meta.SchemaBaseBeamTable

getSchema

Methods inherited from class org.apache.beam.sdk.extensions.sql.meta.BaseBeamTable

buildIOReader, constructFilter, supportsProjects

Methods inherited from class java.lang.Object

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

Constructor Detail

TestBoundedTable

public TestBoundedTable(Schema beamSchema)

Method Detail

getTableStatistics

public BeamTableStatistics getTableStatistics(PipelineOptions options)

Description copied from interface: BeamSqlTable

Estimates the number of rows or the rate for unbounded Tables. If it is not possible to estimate the row count or rate it will return BeamTableStatistics.BOUNDED_UNKNOWN.

Specified by:

getTableStatistics in interface BeamSqlTable

Overrides:

getTableStatistics in class BaseBeamTable

isBounded

public PCollection.IsBounded isBounded()

Description copied from interface: BeamSqlTable

Whether this table is bounded (known to be finite) or unbounded (may or may not be finite).

of

public static TestBoundedTable of(java.lang.Object... args)

Convenient way to build a mocked bounded table.

e.g.

 TestUnboundedTable
   .of(Types.BIGINT, "order_id",
       Types.INTEGER, "site_id",
       Types.DOUBLE, "price",
       Types.TIMESTAMP, "order_time")
 

of

public static TestBoundedTable of(Schema type)

Build a mocked bounded table with the specified type.

addRows

public TestBoundedTable addRows(java.lang.Object... args)

Add rows to the builder.

Sample usage:

 addRows(
   1, 3, "james", -- first row
   2, 5, "bond"   -- second row
   ...
 )
 

buildIOReader

public PCollection<Row> buildIOReader(PBegin begin)

Description copied from interface: BeamSqlTable

create a PCollection<Row> from source.

buildIOWriter

public POutput buildIOWriter(PCollection<Row> input)

Description copied from interface: BeamSqlTable

create a IO.write() instance to write to target.

Specified by:

buildIOWriter in interface BeamSqlTable

Overrides:

buildIOWriter in class TestTable