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

Class TestBoundedTable

java.lang.Object

org.apache.beam.sdk.extensions.sql.impl.schema.BaseBeamTable

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.impl.schema.BaseBeamTable

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	getRowCount(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.impl.schema.BaseBeamTable

getSchema

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

getRowCount

public BeamTableStatistics getRowCount(PipelineOptions options)

Description copied from interface: BeamSqlTable

Estimates the number of rows or the rate for unbounded Tables.

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