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

Class BeamSqlEnv

java.lang.Object

org.apache.beam.sdk.extensions.sql.impl.BeamSqlEnv

@Internal
 @Experimental
public class BeamSqlEnv
extends java.lang.Object

Contains the metadata of tables/UDF functions, and exposes APIs to query/validate/optimize/translate SQL statements.

Method Summary

Modifier and Type	Method and Description
void	addSchema(java.lang.String name, TableProvider tableProvider)
void	executeDdl(java.lang.String sqlStatement)
java.lang.String	explain(java.lang.String sqlString)
org.apache.calcite.jdbc.CalcitePrepare.Context	getContext()
java.util.Map<java.lang.String,java.lang.String>	getPipelineOptions()
static BeamSqlEnv	inMemory(TableProvider... tableProviders)
boolean	isDdl(java.lang.String sqlStatement)
void	loadBeamBuiltinFunctions()
void	loadUdfUdafFromProvider() Load all UDF/UDAF from UdfUdafProvider.
BeamRelNode	parseQuery(java.lang.String query)
static BeamSqlEnv	readOnly(java.lang.String tableType, java.util.Map<java.lang.String,BeamSqlTable> tables)
void	registerUdaf(java.lang.String functionName, Combine.CombineFn combineFn) Register a UDAF function which can be used in GROUP-BY expression.
void	registerUdf(java.lang.String functionName, java.lang.Class<?> clazz, java.lang.String method) Register a UDF function which can be used in SQL expression.
void	registerUdf(java.lang.String functionName, java.lang.Class<? extends BeamSqlUdf> clazz) Register a UDF function which can be used in SQL expression.
void	registerUdf(java.lang.String functionName, SerializableFunction sfn) Register SerializableFunction as a UDF function which can be used in SQL expression.
void	setCurrentSchema(java.lang.String name)
static BeamSqlEnv	withTableProvider(TableProvider tableProvider)

Methods inherited from class java.lang.Object

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

Method Detail

readOnly

public static BeamSqlEnv readOnly(java.lang.String tableType,
                                  java.util.Map<java.lang.String,BeamSqlTable> tables)

withTableProvider

public static BeamSqlEnv withTableProvider(TableProvider tableProvider)

inMemory

public static BeamSqlEnv inMemory(TableProvider... tableProviders)

addSchema

public void addSchema(java.lang.String name,
                      TableProvider tableProvider)

setCurrentSchema

public void setCurrentSchema(java.lang.String name)

registerUdf

public void registerUdf(java.lang.String functionName,
                        java.lang.Class<?> clazz,
                        java.lang.String method)

Register a UDF function which can be used in SQL expression.

registerUdf

public void registerUdf(java.lang.String functionName,
                        java.lang.Class<? extends BeamSqlUdf> clazz)

Register a UDF function which can be used in SQL expression.

registerUdf

public void registerUdf(java.lang.String functionName,
                        SerializableFunction sfn)

Register SerializableFunction as a UDF function which can be used in SQL expression. Note, SerializableFunction must have a constructor without arguments.

registerUdaf

public void registerUdaf(java.lang.String functionName,
                         Combine.CombineFn combineFn)

Register a UDAF function which can be used in GROUP-BY expression. See Combine.CombineFn on how to implement a UDAF.

loadUdfUdafFromProvider

public void loadUdfUdafFromProvider()

Load all UDF/UDAF from UdfUdafProvider.

loadBeamBuiltinFunctions

public void loadBeamBuiltinFunctions()

parseQuery

public BeamRelNode parseQuery(java.lang.String query)
                       throws ParseException

Throws:

ParseException

isDdl

public boolean isDdl(java.lang.String sqlStatement)
              throws ParseException

Throws:

ParseException

executeDdl

public void executeDdl(java.lang.String sqlStatement)
                throws ParseException

Throws:

ParseException

getContext

public org.apache.calcite.jdbc.CalcitePrepare.Context getContext()

getPipelineOptions

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

explain

public java.lang.String explain(java.lang.String sqlString)
                         throws ParseException

Throws:

ParseException