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

Class TestTableProvider

java.lang.Object

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

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

All Implemented Interfaces:

TableProvider

@AutoService(value=TableProvider.class)
public class TestTableProvider
extends InMemoryMetaTableProvider

Test in-memory table provider for use in tests.

Keeps global state and tracks class instances. Works only in DirectRunner.

Constructor Summary

Constructors

Constructor and Description
TestTableProvider()

Method Summary

Modifier and Type	Method and Description
void	addRows(java.lang.String tableName, Row... rows)
BeamSqlTable	buildBeamSqlTable(Table table) Build a BeamSqlTable using the given table meta info.
void	createTable(Table table) Creates a table.
void	dropTable(java.lang.String tableName) Drops a table.
java.util.Map<java.lang.String,Table>	getTables() Get all tables from this provider.
java.lang.String	getTableType() Gets the table type this provider handles.
java.util.List<Row>	tableRows(java.lang.String tableName)
java.util.Map<java.lang.String,org.apache.beam.sdk.extensions.sql.meta.provider.test.TestTableProvider.TableWithRows>	tables()

Methods inherited from class java.lang.Object

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

Constructor Detail

TestTableProvider

public TestTableProvider()

Method Detail

getTableType

public java.lang.String getTableType()

Description copied from interface: TableProvider

Gets the table type this provider handles.

tables

public java.util.Map<java.lang.String,org.apache.beam.sdk.extensions.sql.meta.provider.test.TestTableProvider.TableWithRows> tables()

createTable

public void createTable(Table table)

Description copied from interface: TableProvider

Creates a table.

Specified by:

createTable in interface TableProvider

Overrides:

createTable in class InMemoryMetaTableProvider

dropTable

public void dropTable(java.lang.String tableName)

Description copied from interface: TableProvider

Drops a table.

Specified by:

dropTable in interface TableProvider

Overrides:

dropTable in class InMemoryMetaTableProvider

getTables

public java.util.Map<java.lang.String,Table> getTables()

Description copied from interface: TableProvider

Get all tables from this provider.

Specified by:

getTables in interface TableProvider

Overrides:

getTables in class InMemoryMetaTableProvider

buildBeamSqlTable

public BeamSqlTable buildBeamSqlTable(Table table)

Description copied from interface: TableProvider

Build a BeamSqlTable using the given table meta info.

addRows

public void addRows(java.lang.String tableName,
                    Row... rows)

tableRows

public java.util.List<Row> tableRows(java.lang.String tableName)