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

Class CalciteConnectionWrapper

java.lang.Object

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

All Implemented Interfaces:

AutoCloseable, Connection, Wrapper, org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.jdbc.CalciteConnection, org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.linq4j.QueryProvider

Direct Known Subclasses:

JdbcConnection

public abstract class CalciteConnectionWrapper
extends Object
implements org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.jdbc.CalciteConnection

Abstract wrapper for CalciteConnection to simplify extension.

Calcite JDBC model lacks convenient Connection class to extend that would also be supported by its factories without significant copy-pasting.

The purpose of this class is to hide the delegation logic from the children classes (JdbcConnection) to make them cleaner and easier to read. It has no functional significance.

This class only delegates to the underlying CalciteConnection, all added or modified functionality should go into into subclasses.

Ultimately a patch to Calcite can be made to simplify this logic.

Field Summary

Fields inherited from interface java.sql.Connection

TRANSACTION_NONE, TRANSACTION_READ_COMMITTED, TRANSACTION_READ_UNCOMMITTED, TRANSACTION_REPEATABLE_READ, TRANSACTION_SERIALIZABLE

Constructor Summary

Constructors

Modifier	Constructor	Description
protected	CalciteConnectionWrapper(org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.jdbc.CalciteConnection connection)

Method Summary

Modifier and Type	Method	Description
void	abort(Executor executor)
void	clearWarnings()
void	close()
void	commit()
org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.config.CalciteConnectionConfig	config()
protected org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.jdbc.CalciteConnection	connection()
Array	createArrayOf(String typeName, Object[] elements)
Blob	createBlob()
Clob	createClob()
NClob	createNClob()
org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.jdbc.CalcitePrepare.Context	createPrepareContext()
<T> org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.linq4j.Queryable<T>	createQuery(org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.linq4j.tree.Expression expression, Class<T> rowType)
<T> org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.linq4j.Queryable<T>	createQuery(org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.linq4j.tree.Expression expression, Type rowType)
SQLXML	createSQLXML()
Statement	createStatement()
Statement	createStatement(int resultSetType, int resultSetConcurrency)
Statement	createStatement(int resultSetType, int resultSetConcurrency, int resultSetHoldability)
Struct	createStruct(String typeName, Object[] attributes)
<T> T	execute(org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.linq4j.tree.Expression expression, Class<T> type)
Object	execute(org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.linq4j.tree.Expression expression, Type type)
<T> org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.linq4j.Enumerator<T>	executeQuery(org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.linq4j.Queryable<T> queryable)
boolean	getAutoCommit()
String	getCatalog()
Properties	getClientInfo()
String	getClientInfo(String name)
int	getHoldability()
DatabaseMetaData	getMetaData()
int	getNetworkTimeout()
Properties	getProperties()
org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.schema.SchemaPlus	getRootSchema()
String	getSchema()
int	getTransactionIsolation()
org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.adapter.java.JavaTypeFactory	getTypeFactory()
Map<String,Class<?>>	getTypeMap()
SQLWarning	getWarnings()
boolean	isClosed()
boolean	isReadOnly()
boolean	isValid(int timeout)
boolean	isWrapperFor(Class<?> iface)
String	nativeSQL(String sql)
CallableStatement	prepareCall(String sql)
CallableStatement	prepareCall(String sql, int resultSetType, int resultSetConcurrency)
CallableStatement	prepareCall(String sql, int resultSetType, int resultSetConcurrency, int resultSetHoldability)
PreparedStatement	prepareStatement(String sql)
PreparedStatement	prepareStatement(String sql, int autoGeneratedKeys)
PreparedStatement	prepareStatement(String sql, int[] columnIndexes)
PreparedStatement	prepareStatement(String sql, int resultSetType, int resultSetConcurrency)
PreparedStatement	prepareStatement(String sql, int resultSetType, int resultSetConcurrency, int resultSetHoldability)
PreparedStatement	prepareStatement(String sql, String[] columnNames)
void	releaseSavepoint(Savepoint savepoint)
void	rollback()
void	rollback(Savepoint savepoint)
void	setAutoCommit(boolean autoCommit)
void	setCatalog(String catalog)
void	setClientInfo(String name, String value)
void	setClientInfo(Properties properties)
void	setHoldability(int holdability)
void	setNetworkTimeout(Executor executor, int milliseconds)
void	setReadOnly(boolean readOnly)
Savepoint	setSavepoint()
Savepoint	setSavepoint(String name)
void	setSchema(String schema)
void	setTransactionIsolation(int level)
void	setTypeMap(Map<String,Class<?>> map)
<T> T	unwrap(Class<T> iface)

Methods inherited from class java.lang.Object

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

Methods inherited from interface java.sql.Connection

beginRequest, endRequest, setShardingKey, setShardingKey, setShardingKeyIfValid, setShardingKeyIfValid

Constructor Details

CalciteConnectionWrapper

protected CalciteConnectionWrapper(org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.jdbc.CalciteConnection connection)

Method Details

connection

protected org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.jdbc.CalciteConnection connection()

getRootSchema

public org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.schema.SchemaPlus getRootSchema()

Specified by:

getRootSchema in interface org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.jdbc.CalciteConnection

getTypeFactory

public org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.adapter.java.JavaTypeFactory getTypeFactory()

Specified by:

getTypeFactory in interface org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.jdbc.CalciteConnection

getProperties

public Properties getProperties()

Specified by:

getProperties in interface org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.jdbc.CalciteConnection

createStatement

public Statement createStatement()
                          throws SQLException

Specified by:

createStatement in interface Connection

Throws:

SQLException

prepareStatement

public PreparedStatement prepareStatement(String sql)
                                   throws SQLException

Specified by:

prepareStatement in interface Connection

Throws:

SQLException

prepareCall

public CallableStatement prepareCall(String sql)
                              throws SQLException

Specified by:

prepareCall in interface Connection

Throws:

SQLException

nativeSQL

public String nativeSQL(String sql)
                 throws SQLException

Specified by:

nativeSQL in interface Connection

Throws:

SQLException

setAutoCommit

public void setAutoCommit(boolean autoCommit)
                   throws SQLException

Specified by:

setAutoCommit in interface Connection

Throws:

SQLException

getAutoCommit

public boolean getAutoCommit()
                      throws SQLException

Specified by:

getAutoCommit in interface Connection

Throws:

SQLException

commit

public void commit()
            throws SQLException

Specified by:

commit in interface Connection

Throws:

SQLException

rollback

public void rollback()
              throws SQLException

Specified by:

rollback in interface Connection

Throws:

SQLException

close

public void close()
           throws SQLException

Specified by:

close in interface AutoCloseable

Specified by:

close in interface Connection

Throws:

SQLException

isClosed

public boolean isClosed()
                 throws SQLException

Specified by:

isClosed in interface Connection

Throws:

SQLException

getMetaData

public DatabaseMetaData getMetaData()
                             throws SQLException

Specified by:

getMetaData in interface Connection

Throws:

SQLException

setReadOnly

public void setReadOnly(boolean readOnly)
                 throws SQLException

Specified by:

setReadOnly in interface Connection

Throws:

SQLException

isReadOnly

public boolean isReadOnly()
                   throws SQLException

Specified by:

isReadOnly in interface Connection

Throws:

SQLException

setCatalog

public void setCatalog(String catalog)
                throws SQLException

Specified by:

setCatalog in interface Connection

Throws:

SQLException

getCatalog

public String getCatalog()
                  throws SQLException

Specified by:

getCatalog in interface Connection

Throws:

SQLException

setTransactionIsolation

public void setTransactionIsolation(int level)
                             throws SQLException

Specified by:

setTransactionIsolation in interface Connection

Throws:

SQLException

getTransactionIsolation

public int getTransactionIsolation()
                            throws SQLException

Specified by:

getTransactionIsolation in interface Connection

Throws:

SQLException

getWarnings

public SQLWarning getWarnings()
                       throws SQLException

Specified by:

getWarnings in interface Connection

Throws:

SQLException

clearWarnings

public void clearWarnings()
                   throws SQLException

Specified by:

clearWarnings in interface Connection

Throws:

SQLException

createStatement

public Statement createStatement(int resultSetType,
 int resultSetConcurrency)
                          throws SQLException

Specified by:

createStatement in interface Connection

Throws:

SQLException

prepareStatement

public PreparedStatement prepareStatement(String sql,
 int resultSetType,
 int resultSetConcurrency)
                                   throws SQLException

Specified by:

prepareStatement in interface Connection

Throws:

SQLException

prepareCall

public CallableStatement prepareCall(String sql,
 int resultSetType,
 int resultSetConcurrency)
                              throws SQLException

Specified by:

prepareCall in interface Connection

Throws:

SQLException

getTypeMap

public Map<String,Class<?>> getTypeMap()
                                throws SQLException

Specified by:

getTypeMap in interface Connection

Throws:

SQLException

setTypeMap

public void setTypeMap(Map<String,Class<?>> map)
                throws SQLException

Specified by:

setTypeMap in interface Connection

Throws:

SQLException

setHoldability

public void setHoldability(int holdability)
                    throws SQLException

Specified by:

setHoldability in interface Connection

Throws:

SQLException

getHoldability

public int getHoldability()
                   throws SQLException

Specified by:

getHoldability in interface Connection

Throws:

SQLException

setSavepoint

public Savepoint setSavepoint()
                       throws SQLException

Specified by:

setSavepoint in interface Connection

Throws:

SQLException

setSavepoint

public Savepoint setSavepoint(String name)
                       throws SQLException

Specified by:

setSavepoint in interface Connection

Throws:

SQLException

rollback

public void rollback(Savepoint savepoint)
              throws SQLException

Specified by:

rollback in interface Connection

Throws:

SQLException

releaseSavepoint

public void releaseSavepoint(Savepoint savepoint)
                      throws SQLException

Specified by:

releaseSavepoint in interface Connection

Throws:

SQLException

createStatement

public Statement createStatement(int resultSetType,
 int resultSetConcurrency,
 int resultSetHoldability)
                          throws SQLException

Specified by:

createStatement in interface Connection

Throws:

SQLException

prepareStatement

public PreparedStatement prepareStatement(String sql,
 int resultSetType,
 int resultSetConcurrency,
 int resultSetHoldability)
                                   throws SQLException

Specified by:

prepareStatement in interface Connection

Throws:

SQLException

prepareCall

public CallableStatement prepareCall(String sql,
 int resultSetType,
 int resultSetConcurrency,
 int resultSetHoldability)
                              throws SQLException

Specified by:

prepareCall in interface Connection

Throws:

SQLException

prepareStatement

public PreparedStatement prepareStatement(String sql,
 int autoGeneratedKeys)
                                   throws SQLException

Specified by:

prepareStatement in interface Connection

Throws:

SQLException

prepareStatement

public PreparedStatement prepareStatement(String sql,
 int[] columnIndexes)
                                   throws SQLException

Specified by:

prepareStatement in interface Connection

Throws:

SQLException

prepareStatement

public PreparedStatement prepareStatement(String sql,
 String[] columnNames)
                                   throws SQLException

Specified by:

prepareStatement in interface Connection

Throws:

SQLException

createClob

public Clob createClob()
                throws SQLException

Specified by:

createClob in interface Connection

Throws:

SQLException

createBlob

public Blob createBlob()
                throws SQLException

Specified by:

createBlob in interface Connection

Throws:

SQLException

createNClob

public NClob createNClob()
                  throws SQLException

Specified by:

createNClob in interface Connection

Throws:

SQLException

createSQLXML

public SQLXML createSQLXML()
                    throws SQLException

Specified by:

createSQLXML in interface Connection

Throws:

SQLException

isValid

public boolean isValid(int timeout)
                throws SQLException

Specified by:

isValid in interface Connection

Throws:

SQLException

setClientInfo

public void setClientInfo(String name,
 String value)
                   throws SQLClientInfoException

Specified by:

setClientInfo in interface Connection

Throws:

SQLClientInfoException

setClientInfo

public void setClientInfo(Properties properties)
                   throws SQLClientInfoException

Specified by:

setClientInfo in interface Connection

Throws:

SQLClientInfoException

getClientInfo

public String getClientInfo(String name)
                     throws SQLException

Specified by:

getClientInfo in interface Connection

Throws:

SQLException

getClientInfo

public Properties getClientInfo()
                         throws SQLException

Specified by:

getClientInfo in interface Connection

Throws:

SQLException

createArrayOf

public Array createArrayOf(String typeName,
 Object[] elements)
                    throws SQLException

Specified by:

createArrayOf in interface Connection

Throws:

SQLException

createStruct

public Struct createStruct(String typeName,
 Object[] attributes)
                    throws SQLException

Specified by:

createStruct in interface Connection

Throws:

SQLException

setSchema

public void setSchema(String schema)
               throws SQLException

Specified by:

setSchema in interface org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.jdbc.CalciteConnection

Specified by:

setSchema in interface Connection

Throws:

SQLException

getSchema

public String getSchema()
                 throws SQLException

Specified by:

getSchema in interface org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.jdbc.CalciteConnection

Specified by:

getSchema in interface Connection

Throws:

SQLException

abort

public void abort(Executor executor)
           throws SQLException

Specified by:

abort in interface Connection

Throws:

SQLException

setNetworkTimeout

public void setNetworkTimeout(Executor executor,
 int milliseconds)
                       throws SQLException

Specified by:

setNetworkTimeout in interface Connection

Throws:

SQLException

getNetworkTimeout

public int getNetworkTimeout()
                      throws SQLException

Specified by:

getNetworkTimeout in interface Connection

Throws:

SQLException

config

public org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.config.CalciteConnectionConfig config()

Specified by:

config in interface org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.jdbc.CalciteConnection

createPrepareContext

public org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.jdbc.CalcitePrepare.Context createPrepareContext()

Specified by:

createPrepareContext in interface org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.jdbc.CalciteConnection

unwrap

public <T> T unwrap(Class<T> iface)
             throws SQLException

Specified by:

unwrap in interface Wrapper

Throws:

SQLException

isWrapperFor

public boolean isWrapperFor(Class<?> iface)
                     throws SQLException

Specified by:

isWrapperFor in interface Wrapper

Throws:

SQLException

createQuery

public <T>
org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.linq4j.Queryable<T> createQuery(org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.linq4j.tree.Expression expression,
 Class<T> rowType)

Specified by:

createQuery in interface org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.linq4j.QueryProvider

createQuery

public <T>
org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.linq4j.Queryable<T> createQuery(org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.linq4j.tree.Expression expression,
 Type rowType)

Specified by:

createQuery in interface org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.linq4j.QueryProvider

execute

public <T> T execute(org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.linq4j.tree.Expression expression,
 Class<T> type)

Specified by:

execute in interface org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.linq4j.QueryProvider

execute

public Object execute(org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.linq4j.tree.Expression expression,
 Type type)

Specified by:

execute in interface org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.linq4j.QueryProvider

executeQuery

public <T>
org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.linq4j.Enumerator<T> executeQuery(org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.linq4j.Queryable<T> queryable)

Specified by:

executeQuery in interface org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.linq4j.QueryProvider