com.scylladb.cdc.debezium.connector.ScyllaCollectionSchema Maven / Gradle / Ivy

Go to download
package com.scylladb.cdc.debezium.connector;

import io.debezium.data.Envelope;
import io.debezium.schema.DataCollectionId;
import io.debezium.schema.DataCollectionSchema;
import org.apache.kafka.connect.data.Schema;

import java.util.HashMap;
import java.util.Map;

public class ScyllaCollectionSchema implements DataCollectionSchema {
    private final CollectionId id;
    private final Schema keySchema;
    private final Schema valueSchema;
    private final Schema beforeSchema;
    private final Schema afterSchema;
    private final Map cellSchemas;
    private final Envelope envelopeSchema;

    public ScyllaCollectionSchema(CollectionId id, Schema keySchema, Schema valueSchema, Schema beforeSchema, Schema afterSchema, Map cellSchemas, Envelope envelopeSchema) {
        this.id = id;
        this.keySchema = keySchema;
        this.valueSchema = valueSchema;
        this.beforeSchema = beforeSchema;
        this.afterSchema = afterSchema;
        this.cellSchemas = cellSchemas;
        this.envelopeSchema = envelopeSchema;
    }

    @Override
    public CollectionId id() {
        return id;
    }

    @Override
    public Schema keySchema() {
        return keySchema;
    }

    public Schema beforeSchema() {
        return beforeSchema;
    }

    public Schema afterSchema() {
        return afterSchema;
    }

    public Schema valueSchema() { return valueSchema; }

    public Schema cellSchema(String columnName) {
        return cellSchemas.get(columnName);
    }

    @Override
    public Envelope getEnvelopeSchema() {
        return envelopeSchema;
    }
}