io.debezium.connector.postgresql.connection

Class PostgresConnection

java.lang.Object

io.debezium.jdbc.JdbcConnection

io.debezium.connector.postgresql.connection.PostgresConnection

All Implemented Interfaces:

AutoCloseable

public class PostgresConnection
extends JdbcConnection

JdbcConnection connection extension used for connecting to Postgres instances.

Author:

Horia Chiorean

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
private static class	PostgresConnection.SqlTypeMapper Allows to obtain the SQL type corresponding to PG types.

Nested classes/interfaces inherited from class io.debezium.jdbc.JdbcConnection

JdbcConnection.BlockingMultiResultSetConsumer, JdbcConnection.BlockingResultSetConsumer, JdbcConnection.CallPreparer, JdbcConnection.ConnectionFactory, JdbcConnection.MultiResultSetConsumer, JdbcConnection.Operations, JdbcConnection.ParameterResultSetConsumer, JdbcConnection.ResultSetConsumer, JdbcConnection.ResultSetMapper<T>, JdbcConnection.StatementFactory, JdbcConnection.StatementPreparer

Field Summary

Fields

Modifier and Type	Field and Description
private Charset	databaseCharset
protected static JdbcConnection.ConnectionFactory	FACTORY
private static org.slf4j.Logger	LOGGER
private static int	MAX_ATTEMPTS_FOR_OBTAINING_REPLICATION_SLOT Obtaining a replication slot may fail if there's a pending transaction.
private static Duration	PAUSE_BETWEEN_REPLICATION_SLOT_RETRIEVAL_ATTEMPTS
private static String	SQL_ARRAY_TYPES
private static String	SQL_NON_ARRAY_TYPES
private TypeRegistry	typeRegistry
private static String	URL_PATTERN

Constructor Summary

Constructors

Constructor and Description
PostgresConnection(Configuration config) Creates a Postgres connection using the supplied configuration.

Method Summary

Modifier and Type	Method and Description
void	close()
String	connectionString() Returns a JDBC connection string for the current configuration.
Long	currentTransactionId() Returns the PG id of the current active transaction
long	currentXLogLocation() Returns the current position in the server tx log.
protected static void	defaultSettings(Configuration.Builder builder)
private Charset	determineDatabaseCharset()
boolean	dropPublication(String publicationName) Drops the debezium publication that was created.
boolean	dropReplicationSlot(String slotName) Drops a replication slot that was created on the DB
private ServerInfo.ReplicationSlot	fetchReplicationSlotInfo(String slotName, String pluginName) Fetches the state of a replication stage given a slot name and plugin name
Charset	getDatabaseCharset()
SlotState	getReplicationSlotState(String slotName, String pluginName) Returns the current state of the replication slot
TypeRegistry	getTypeRegistry()
private static TypeRegistry	initTypeRegistry(Connection db)
private Long	parseConfirmedFlushLsn(String slotName, String pluginName, String database, ResultSet rs) Obtains the LSN to resume streaming from.
private Long	parseRestartLsn(String slotName, String pluginName, String database, ResultSet rs)
protected ServerInfo.ReplicationSlot	queryForSlot(String slotName, String database, String pluginName, JdbcConnection.ResultSetMapper<ServerInfo.ReplicationSlot> map)
ServerInfo.ReplicaIdentity	readReplicaIdentityInfo(TableId tableId) Prints out information about the REPLICA IDENTITY status of a table.
(package private) ServerInfo.ReplicationSlot	readReplicationSlotInfo(String slotName, String pluginName) Fetches a replication slot, repeating the query until either the slot is created or until the max number of attempts has been reached To fetch the slot without the retries, use the fetchReplicationSlotInfo(java.lang.String, java.lang.String) call
protected int	resolveNativeType(String typeName)
ServerInfo	serverInfo() Returns information about the PG server to which this instance is connected.
private Long	tryParseLsn(String slotName, String pluginName, String database, ResultSet rs, String column)
private static void	validateServerVersion(Statement statement)

Methods inherited from class io.debezium.jdbc.JdbcConnection

call, columnsFor, columnsFor, config, connect, connection, connection, connectionString, database, execute, execute, executeWithoutCommitting, isConnected, parseSqlStatementString, patternBasedFactory, patternBasedFactory, prepareQuery, prepareQuery, prepareQuery, prepareQuery, prepareQueryAndMap, prepareQueryWithBlockingConsumer, prepareUpdate, print, print, query, query, queryAndMap, queryAndMap, queryWithBlockingConsumer, readAllCatalogNames, readAllSchemaNames, readAllTableNames, readPrimaryKeyNames, readSchema, readTableColumn, readTableNames, setAutoCommit, tableTypes, username

Methods inherited from class java.lang.Object

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

Field Detail

LOGGER

private static org.slf4j.Logger LOGGER

URL_PATTERN

private static final String URL_PATTERN

FACTORY

protected static final JdbcConnection.ConnectionFactory FACTORY

SQL_NON_ARRAY_TYPES

private static final String SQL_NON_ARRAY_TYPES

See Also:

Constant Field Values

SQL_ARRAY_TYPES

private static final String SQL_ARRAY_TYPES

See Also:

Constant Field Values

MAX_ATTEMPTS_FOR_OBTAINING_REPLICATION_SLOT

private static final int MAX_ATTEMPTS_FOR_OBTAINING_REPLICATION_SLOT

Obtaining a replication slot may fail if there's a pending transaction. We're retrying to get a slot for 30 min.

See Also:

Constant Field Values

PAUSE_BETWEEN_REPLICATION_SLOT_RETRIEVAL_ATTEMPTS

private static final Duration PAUSE_BETWEEN_REPLICATION_SLOT_RETRIEVAL_ATTEMPTS

typeRegistry

private final TypeRegistry typeRegistry

databaseCharset

private final Charset databaseCharset

Constructor Detail

PostgresConnection

public PostgresConnection(Configuration config)

Creates a Postgres connection using the supplied configuration.

Parameters:

config - Configuration instance, may not be null.

Method Detail

connectionString

public String connectionString()

Returns a JDBC connection string for the current configuration.

Returns:

a String where the variables in urlPattern are replaced with values from the configuration

readReplicaIdentityInfo

public ServerInfo.ReplicaIdentity readReplicaIdentityInfo(TableId tableId)
                                                   throws SQLException

Prints out information about the REPLICA IDENTITY status of a table. This in turn determines how much information is available for UPDATE and DELETE operations for logical replication.

Parameters:

tableId - the identifier of the table

Returns:

the replica identity information; never null

Throws:

SQLException - if there is a problem obtaining the replica identity information for the given table

getReplicationSlotState

public SlotState getReplicationSlotState(String slotName,
                                         String pluginName)
                                  throws SQLException

Returns the current state of the replication slot

Parameters:

slotName - the name of the slot

pluginName - the name of the plugin used for the desired slot

Returns:

the SlotState or null, if no slot state is found

Throws:

SQLException

fetchReplicationSlotInfo

private ServerInfo.ReplicationSlot fetchReplicationSlotInfo(String slotName,
                                                            String pluginName)
                                                     throws SQLException

Fetches the state of a replication stage given a slot name and plugin name

Parameters:

slotName - the name of the slot

pluginName - the name of the plugin used for the desired slot

Returns:

the ServerInfo.ReplicationSlot object or a ServerInfo.ReplicationSlot.INVALID if the slot is not valid

Throws:

SQLException - is thrown by the underlying JDBC

readReplicationSlotInfo

@VisibleForTesting
ServerInfo.ReplicationSlot readReplicationSlotInfo(String slotName,
                                                                      String pluginName)
                                                               throws SQLException,
                                                                      InterruptedException

Fetches a replication slot, repeating the query until either the slot is created or until the max number of attempts has been reached To fetch the slot without the retries, use the fetchReplicationSlotInfo(java.lang.String, java.lang.String) call

Parameters:

slotName - the slot name

pluginName - the name of the plugin

Returns:

the ServerInfo.ReplicationSlot object or a ServerInfo.ReplicationSlot.INVALID if the slot is not valid

Throws:

SQLException - is thrown by the underyling jdbc driver

InterruptedException - is thrown if we don't return an answer within the set number of retries

queryForSlot

protected ServerInfo.ReplicationSlot queryForSlot(String slotName,
                                                  String database,
                                                  String pluginName,
                                                  JdbcConnection.ResultSetMapper<ServerInfo.ReplicationSlot> map)
                                           throws SQLException

Throws:

SQLException

parseConfirmedFlushLsn

private Long parseConfirmedFlushLsn(String slotName,
                                    String pluginName,
                                    String database,
                                    ResultSet rs)

Obtains the LSN to resume streaming from. On PG 9.5 there is no confirmed_flushed_lsn yet, so restart_lsn will be read instead. This may result in more records to be re-read after a restart.

parseRestartLsn

private Long parseRestartLsn(String slotName,
                             String pluginName,
                             String database,
                             ResultSet rs)

tryParseLsn

private Long tryParseLsn(String slotName,
                         String pluginName,
                         String database,
                         ResultSet rs,
                         String column)
                  throws org.apache.kafka.connect.errors.ConnectException,
                         SQLException

Throws:

org.apache.kafka.connect.errors.ConnectException

SQLException

dropReplicationSlot

public boolean dropReplicationSlot(String slotName)

Drops a replication slot that was created on the DB

Parameters:

slotName - the name of the replication slot, may not be null

Returns:

true if the slot was dropped, false otherwise

dropPublication

public boolean dropPublication(String publicationName)

Drops the debezium publication that was created.

Parameters:

publicationName - the publication name, may not be null

Returns:

true if the publication was dropped, false otherwise

close

public void close()

Specified by:

close in interface AutoCloseable

Overrides:

close in class JdbcConnection

currentTransactionId

public Long currentTransactionId()
                          throws SQLException

Returns the PG id of the current active transaction

Returns:

a PG transaction identifier, or null if no tx is active

Throws:

SQLException - if anything fails.

currentXLogLocation

public long currentXLogLocation()
                         throws SQLException

Returns the current position in the server tx log.

Returns:

a long value, never negative

Throws:

SQLException - if anything unexpected fails.

serverInfo

public ServerInfo serverInfo()
                      throws SQLException

Returns information about the PG server to which this instance is connected.

Returns:

a ServerInfo instance, never null

Throws:

SQLException - if anything fails

getDatabaseCharset

public Charset getDatabaseCharset()

determineDatabaseCharset

private Charset determineDatabaseCharset()

defaultSettings

protected static void defaultSettings(Configuration.Builder builder)

validateServerVersion

private static void validateServerVersion(Statement statement)
                                   throws SQLException

Throws:

SQLException

resolveNativeType

protected int resolveNativeType(String typeName)

Overrides:

resolveNativeType in class JdbcConnection

initTypeRegistry

private static TypeRegistry initTypeRegistry(Connection db)

getTypeRegistry

public TypeRegistry getTypeRegistry()