Search in sources :

Example 1 with TypeRegistry

use of io.debezium.connector.postgresql.TypeRegistry in project debezium by debezium.

the class PostgresConnection method initTypeRegistry.

private static TypeRegistry initTypeRegistry(Connection db, Map<String, Integer> nameToJdbc) {
    final TypeInfo typeInfo = ((BaseConnection) db).getTypeInfo();
    TypeRegistry.Builder typeRegistryBuilder = TypeRegistry.create(typeInfo);
    try {
        try (final Statement statement = db.createStatement()) {
            // Read non-array types
            try (final ResultSet rs = statement.executeQuery(SQL_NON_ARRAY_TYPES)) {
                while (rs.next()) {
                    final int oid = rs.getInt("oid");
                    typeRegistryBuilder.addType(new PostgresType(rs.getString("name"), oid, nameToJdbc.get(rs.getString("name")), typeInfo));
                }
            }
            // Read array types
            try (final ResultSet rs = statement.executeQuery(SQL_ARRAY_TYPES)) {
                while (rs.next()) {
                    // int2vector and oidvector will not be treated as arrays
                    final int oid = rs.getInt("oid");
                    typeRegistryBuilder.addType(new PostgresType(rs.getString("name"), oid, nameToJdbc.get(rs.getString("name")), typeInfo, typeRegistryBuilder.get(rs.getInt("element"))));
                }
            }
        }
    } catch (SQLException e) {
        throw new ConnectException("Could not intialize type registry", e);
    }
    return typeRegistryBuilder.build();
}
Also used : PostgresType(io.debezium.connector.postgresql.PostgresType) SQLException(java.sql.SQLException) Statement(java.sql.Statement) ResultSet(java.sql.ResultSet) TypeInfo(org.postgresql.core.TypeInfo) TypeRegistry(io.debezium.connector.postgresql.TypeRegistry) BaseConnection(org.postgresql.core.BaseConnection) ConnectException(org.apache.kafka.connect.errors.ConnectException)

Aggregations

PostgresType (io.debezium.connector.postgresql.PostgresType)1 TypeRegistry (io.debezium.connector.postgresql.TypeRegistry)1 ResultSet (java.sql.ResultSet)1 SQLException (java.sql.SQLException)1 Statement (java.sql.Statement)1 ConnectException (org.apache.kafka.connect.errors.ConnectException)1 BaseConnection (org.postgresql.core.BaseConnection)1 TypeInfo (org.postgresql.core.TypeInfo)1