Search in sources :

Example 1 with AbstractReplicationMessageColumn

use of io.debezium.connector.postgresql.connection.AbstractReplicationMessageColumn in project debezium by debezium.

the class Wal2JsonReplicationMessage method transform.

private List<ReplicationMessage.Column> transform(final Document data, final String nameField, final String typeField, final String valueField, final String optionalsField) {
    final Array columnNames = data.getArray(nameField);
    final Array columnTypes = data.getArray(typeField);
    final Array columnValues = data.getArray(valueField);
    final Array columnOptionals = data.getArray(optionalsField);
    if (columnNames.size() != columnTypes.size() || columnNames.size() != columnValues.size()) {
        throw new ConnectException("Column related arrays do not have the same size");
    }
    final List<ReplicationMessage.Column> columns = new ArrayList<>(columnNames.size());
    for (int i = 0; i < columnNames.size(); i++) {
        final String columnName = columnNames.get(i).asString();
        final String columnTypeName = columnTypes.get(i).asString();
        final boolean columnOptional = columnOptionals != null ? columnOptionals.get(i).asBoolean() : false;
        final Value rawValue = columnValues.get(i);
        final PostgresType columnType = typeRegistry.get(parseType(columnName, columnTypeName));
        columns.add(new AbstractReplicationMessageColumn(columnName, columnType, columnTypeName, columnOptional, true) {

            @Override
            public Object getValue(PgConnectionSupplier connection, boolean includeUnknownDatatypes) {
                return Wal2JsonReplicationMessage.this.getValue(columnName, columnType, columnTypeName, rawValue, connection, includeUnknownDatatypes);
            }
        });
    }
    return columns;
}
Also used : AbstractReplicationMessageColumn(io.debezium.connector.postgresql.connection.AbstractReplicationMessageColumn) ArrayList(java.util.ArrayList) PGpoint(org.postgresql.geometric.PGpoint) Array(io.debezium.document.Array) PgArray(org.postgresql.jdbc.PgArray) PostgresType(io.debezium.connector.postgresql.PostgresType) AbstractReplicationMessageColumn(io.debezium.connector.postgresql.connection.AbstractReplicationMessageColumn) Value(io.debezium.document.Value) PgConnectionSupplier(io.debezium.connector.postgresql.RecordsStreamProducer.PgConnectionSupplier) ConnectException(org.apache.kafka.connect.errors.ConnectException)

Aggregations

PostgresType (io.debezium.connector.postgresql.PostgresType)1 PgConnectionSupplier (io.debezium.connector.postgresql.RecordsStreamProducer.PgConnectionSupplier)1 AbstractReplicationMessageColumn (io.debezium.connector.postgresql.connection.AbstractReplicationMessageColumn)1 Array (io.debezium.document.Array)1 Value (io.debezium.document.Value)1 ArrayList (java.util.ArrayList)1 ConnectException (org.apache.kafka.connect.errors.ConnectException)1 PGpoint (org.postgresql.geometric.PGpoint)1 PgArray (org.postgresql.jdbc.PgArray)1