Package io.trino.plugin.postgresql

Class PostgreSqlClient

java.lang.Object

io.trino.plugin.jdbc.BaseJdbcClient

io.trino.plugin.postgresql.PostgreSqlClient

All Implemented Interfaces:

JdbcClient

public class PostgreSqlClient
extends BaseJdbcClient

Nested Class Summary

Nested classes/interfaces inherited from class io.trino.plugin.jdbc.BaseJdbcClient

BaseJdbcClient.TopNFunction

Field Summary

Fields inherited from class io.trino.plugin.jdbc.BaseJdbcClient

connectionFactory, identifierQuote, jdbcTypesMappedToVarchar, queryBuilder, queryModifier

Constructor Summary

Constructors

Constructor	Description
PostgreSqlClient(BaseJdbcConfig config, PostgreSqlConfig postgreSqlConfig, JdbcStatisticsConfig statisticsConfig, ConnectionFactory connectionFactory, QueryBuilder queryBuilder, TypeManager typeManager, IdentifierMapping identifierMapping, RemoteQueryModifier queryModifier)

Method Summary

Modifier and Type	Method	Description
Optional<ParameterizedExpression>	convertPredicate(ConnectorSession session, ConnectorExpression expression, Map<String,ColumnHandle> assignments)
void	createTable(ConnectorSession session, ConnectorTableMetadata tableMetadata)
protected List<String>	createTableSqls(RemoteTableName remoteTableName, List<String> columns, ConnectorTableMetadata tableMetadata)
OptionalLong	delete(ConnectorSession session, JdbcTableHandle handle)
List<JdbcColumnHandle>	getColumns(ConnectorSession session, JdbcTableHandle tableHandle)
PreparedStatement	getPreparedStatement(Connection connection, String sql, Optional<Integer> columnCount)
Optional<Type>	getSupportedType(ConnectorSession session, Type type)
TableStatistics	getTableStatistics(ConnectorSession session, JdbcTableHandle handle)
protected Optional<List<String>>	getTableTypes()
Optional<JdbcExpression>	implementAggregation(ConnectorSession session, AggregateFunction aggregate, Map<String,ColumnHandle> assignments)
Optional<PreparedQuery>	implementJoin(ConnectorSession session, JoinType joinType, PreparedQuery leftSource, PreparedQuery rightSource, List<JdbcJoinCondition> joinConditions, Map<JdbcColumnHandle,String> rightAssignments, Map<JdbcColumnHandle,String> leftAssignments, JoinStatistics statistics)
protected static boolean	isCollatable(JdbcColumnHandle column)
boolean	isLimitGuaranteed(ConnectorSession session)
protected boolean	isSupportedJoinCondition(ConnectorSession session, JdbcJoinCondition joinCondition)
boolean	isTopNGuaranteed(ConnectorSession session)
protected Optional<BiFunction<String,Long,String>>	limitFunction()
protected void	renameTable(ConnectorSession session, Connection connection, String catalogName, String remoteSchemaName, String remoteTableName, String newRemoteSchemaName, String newRemoteTableName)
void	setColumnComment(ConnectorSession session, JdbcTableHandle handle, JdbcColumnHandle column, Optional<String> comment)
void	setTableComment(ConnectorSession session, JdbcTableHandle handle, Optional<String> comment)
boolean	supportsAggregationPushdown(ConnectorSession session, JdbcTableHandle table, List<AggregateFunction> aggregates, Map<String,ColumnHandle> assignments, List<List<ColumnHandle>> groupingSets)
boolean	supportsTopN(ConnectorSession session, JdbcTableHandle handle, List<JdbcSortItem> sortOrder)
Optional<ColumnMapping>	toColumnMapping(ConnectorSession session, Connection connection, JdbcTypeHandle typeHandle)
protected Optional<BaseJdbcClient.TopNFunction>	topNFunction()
WriteMapping	toWriteMapping(ConnectorSession session, Type type)
protected void	verifyColumnName(DatabaseMetaData databaseMetadata, String columnName)
protected void	verifySchemaName(DatabaseMetaData databaseMetadata, String schemaName)
protected void	verifyTableName(DatabaseMetaData databaseMetadata, String tableName)

Methods inherited from class io.trino.plugin.jdbc.BaseJdbcClient

addColumn, addColumn, applyQueryTransformations, beginCreateTable, beginInsertTable, buildInsertSql, buildProcedure, buildSql, commitCreateTable, copyTableSchema, createSchema, createSchema, createTable, createTable, createTableSql, dropColumn, dropSchema, dropSchema, dropTable, dropTable, escapeObjectNameForMetadataQuery, escapeObjectNameForMetadataQuery, execute, execute, filterSchema, finishInsertTable, getAdditionalPredicate, getCaseSensitivityForColumns, getColumnDefinitionSql, getColumns, getColumns, getConnection, getConnection, getConnection, getForcedMappingToVarchar, getIdentifierMapping, getInteger, getMaxWriteParallelism, getProcedureHandle, getRemoteIdentifiers, getSchemaNames, getSplits, getSplits, getTableHandle, getTableHandle, getTableNames, getTableProperties, getTables, getTableSchemaName, getTableStatistics, listSchemas, mapToUnboundedVarchar, postProcessInsertTableNameClause, prepareQuery, prepareQuery, preventTextualTypeAggregationPushdown, quoted, quoted, quoted, renameColumn, renameColumn, renameSchema, renameSchema, renameTable, renameTable, rollbackCreateTable, setColumnType, supportsLimit, supportsRetries, toColumnMappings, truncateTable, varcharLiteral

Methods inherited from class java.lang.Object

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

Methods inherited from interface io.trino.plugin.jdbc.JdbcClient

abortReadConnection, getSystemTable, getTableComment, getTableScanRedirection, schemaExists, setTableProperties

Constructor Details

PostgreSqlClient

@Inject
public PostgreSqlClient(BaseJdbcConfig config,
 PostgreSqlConfig postgreSqlConfig,
 JdbcStatisticsConfig statisticsConfig,
 ConnectionFactory connectionFactory,
 QueryBuilder queryBuilder,
 TypeManager typeManager,
 IdentifierMapping identifierMapping,
 RemoteQueryModifier queryModifier)

Method Details

createTable

public void createTable(ConnectorSession session,
 ConnectorTableMetadata tableMetadata)

Specified by:

createTable in interface JdbcClient

Overrides:

createTable in class BaseJdbcClient

createTableSqls

protected List<String> createTableSqls(RemoteTableName remoteTableName,
 List<String> columns,
 ConnectorTableMetadata tableMetadata)

Overrides:

createTableSqls in class BaseJdbcClient

setTableComment

public void setTableComment(ConnectorSession session,
 JdbcTableHandle handle,
 Optional<String> comment)

renameTable

protected void renameTable(ConnectorSession session,
 Connection connection,
 String catalogName,
 String remoteSchemaName,
 String remoteTableName,
 String newRemoteSchemaName,
 String newRemoteTableName)
                    throws SQLException

Overrides:

renameTable in class BaseJdbcClient

Throws:

SQLException

getPreparedStatement

public PreparedStatement getPreparedStatement(Connection connection,
 String sql,
 Optional<Integer> columnCount)
                                       throws SQLException

Specified by:

getPreparedStatement in interface JdbcClient

Overrides:

getPreparedStatement in class BaseJdbcClient

Throws:

SQLException

getTableTypes

protected Optional<List<String>> getTableTypes()

Overrides:

getTableTypes in class BaseJdbcClient

getColumns

public List<JdbcColumnHandle> getColumns(ConnectorSession session,
 JdbcTableHandle tableHandle)

Specified by:

getColumns in interface JdbcClient

Overrides:

getColumns in class BaseJdbcClient

toColumnMapping

public Optional<ColumnMapping> toColumnMapping(ConnectorSession session,
 Connection connection,
 JdbcTypeHandle typeHandle)

getSupportedType

public Optional<Type> getSupportedType(ConnectorSession session,
 Type type)

toWriteMapping

public WriteMapping toWriteMapping(ConnectorSession session,
 Type type)

implementAggregation

public Optional<JdbcExpression> implementAggregation(ConnectorSession session,
 AggregateFunction aggregate,
 Map<String,ColumnHandle> assignments)

supportsAggregationPushdown

public boolean supportsAggregationPushdown(ConnectorSession session,
 JdbcTableHandle table,
 List<AggregateFunction> aggregates,
 Map<String,ColumnHandle> assignments,
 List<List<ColumnHandle>> groupingSets)

convertPredicate

public Optional<ParameterizedExpression> convertPredicate(ConnectorSession session,
 ConnectorExpression expression,
 Map<String,ColumnHandle> assignments)

supportsTopN

public boolean supportsTopN(ConnectorSession session,
 JdbcTableHandle handle,
 List<JdbcSortItem> sortOrder)

Specified by:

supportsTopN in interface JdbcClient

Overrides:

supportsTopN in class BaseJdbcClient

topNFunction

protected Optional<BaseJdbcClient.TopNFunction> topNFunction()

Overrides:

topNFunction in class BaseJdbcClient

isCollatable

protected static boolean isCollatable(JdbcColumnHandle column)

isTopNGuaranteed

public boolean isTopNGuaranteed(ConnectorSession session)

Specified by:

isTopNGuaranteed in interface JdbcClient

Overrides:

isTopNGuaranteed in class BaseJdbcClient

limitFunction

protected Optional<BiFunction<String,Long,String>> limitFunction()

Overrides:

limitFunction in class BaseJdbcClient

isLimitGuaranteed

public boolean isLimitGuaranteed(ConnectorSession session)

Specified by:

isLimitGuaranteed in interface JdbcClient

Overrides:

isLimitGuaranteed in class BaseJdbcClient

delete

public OptionalLong delete(ConnectorSession session,
 JdbcTableHandle handle)

Specified by:

delete in interface JdbcClient

Overrides:

delete in class BaseJdbcClient

getTableStatistics

public TableStatistics getTableStatistics(ConnectorSession session,
 JdbcTableHandle handle)

implementJoin

public Optional<PreparedQuery> implementJoin(ConnectorSession session,
 JoinType joinType,
 PreparedQuery leftSource,
 PreparedQuery rightSource,
 List<JdbcJoinCondition> joinConditions,
 Map<JdbcColumnHandle,String> rightAssignments,
 Map<JdbcColumnHandle,String> leftAssignments,
 JoinStatistics statistics)

Specified by:

implementJoin in interface JdbcClient

Overrides:

implementJoin in class BaseJdbcClient

isSupportedJoinCondition

protected boolean isSupportedJoinCondition(ConnectorSession session,
 JdbcJoinCondition joinCondition)

Overrides:

isSupportedJoinCondition in class BaseJdbcClient

verifySchemaName

protected void verifySchemaName(DatabaseMetaData databaseMetadata,
 String schemaName)
                         throws SQLException

Overrides:

verifySchemaName in class BaseJdbcClient

Throws:

SQLException

verifyTableName

protected void verifyTableName(DatabaseMetaData databaseMetadata,
 String tableName)
                        throws SQLException

Overrides:

verifyTableName in class BaseJdbcClient

Throws:

SQLException

verifyColumnName

protected void verifyColumnName(DatabaseMetaData databaseMetadata,
 String columnName)
                         throws SQLException

Overrides:

verifyColumnName in class BaseJdbcClient

Throws:

SQLException

setColumnComment

public void setColumnComment(ConnectorSession session,
 JdbcTableHandle handle,
 JdbcColumnHandle column,
 Optional<String> comment)