Package com.amazonaws.athena.connectors.snowflake

Class SnowflakeRecordHandler

java.lang.Object

com.amazonaws.athena.connector.lambda.handlers.RecordHandler

com.amazonaws.athena.connectors.jdbc.manager.JdbcRecordHandler

com.amazonaws.athena.connectors.snowflake.SnowflakeRecordHandler

All Implemented Interfaces:

com.amazonaws.services.lambda.runtime.RequestStreamHandler

public class SnowflakeRecordHandler
extends JdbcRecordHandler

Field Summary

Fields inherited from class com.amazonaws.athena.connectors.jdbc.manager.JdbcRecordHandler

queryPassthrough

Fields inherited from class com.amazonaws.athena.connector.lambda.handlers.RecordHandler

configOptions

Constructor Summary

Constructors

Constructor	Description
SnowflakeRecordHandler(DatabaseConnectionConfig databaseConnectionConfig, GenericJdbcConnectionFactory jdbcConnectionFactory, Map<String,String> configOptions)
SnowflakeRecordHandler(DatabaseConnectionConfig databaseConnectionConfig, Map<String,String> configOptions)
SnowflakeRecordHandler(Map<String,String> configOptions)	Instantiates handler to be used by Lambda function directly.

Method Summary

Modifier and Type	Method	Description
PreparedStatement	buildSplitSql(Connection jdbcConnection, String catalogName, TableName tableName, org.apache.arrow.vector.types.pojo.Schema schema, Constraints constraints, Split split)	Builds split SQL string and returns prepared statement.

Methods inherited from class com.amazonaws.athena.connectors.jdbc.manager.JdbcRecordHandler

buildQueryPassthroughSql, getCredentialProvider, getJdbcConnectionFactory, makeExtractor, makeFactory, readWithConstraint

Methods inherited from class com.amazonaws.athena.connector.lambda.handlers.RecordHandler

doHandleRequest, doReadRecords, getSecret, getSpillConfig, handleRequest, onPing, resolveSecrets

Methods inherited from class java.lang.Object

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

Constructor Detail

SnowflakeRecordHandler

public SnowflakeRecordHandler(Map<String,String> configOptions)

Instantiates handler to be used by Lambda function directly. Recommend using SnowflakeMuxCompositeHandler instead.

SnowflakeRecordHandler

public SnowflakeRecordHandler(DatabaseConnectionConfig databaseConnectionConfig,
                              Map<String,String> configOptions)

SnowflakeRecordHandler

public SnowflakeRecordHandler(DatabaseConnectionConfig databaseConnectionConfig,
                              GenericJdbcConnectionFactory jdbcConnectionFactory,
                              Map<String,String> configOptions)

Method Detail

buildSplitSql

public PreparedStatement buildSplitSql(Connection jdbcConnection,
                                       String catalogName,
                                       TableName tableName,
                                       org.apache.arrow.vector.types.pojo.Schema schema,
                                       Constraints constraints,
                                       Split split)
                                throws SQLException

Description copied from class: JdbcRecordHandler

Builds split SQL string and returns prepared statement.

Specified by:

buildSplitSql in class JdbcRecordHandler

Parameters:

jdbcConnection - jdbc connection. See Connection

catalogName - Athena provided catalog name.

tableName - database table name.

schema - table schema.

constraints - constraints to push down to the database.

split - table split.

Returns:

prepared statement with sql. See PreparedStatement

Throws:

SQLException - JDBC database exception.