io.atleon.schemaregistry.confluent.AvroRegistryDeserializer Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of atleon-schema-registry-confluent Show documentation

There is a newer version: 0.28.3

Show newest version

package io.atleon.schemaregistry.confluent;

import io.atleon.avro.AtleonReflectData;
import io.atleon.avro.AvroDeserializer;
import io.atleon.avro.GenericDatas;
import io.confluent.kafka.schemaregistry.ParsedSchema;
import io.confluent.kafka.schemaregistry.SchemaProvider;
import io.confluent.kafka.schemaregistry.avro.AvroSchemaProvider;
import org.apache.avro.Schema;
import org.apache.avro.generic.GenericData;
import org.apache.avro.specific.SpecificData;

import java.util.Map;

/**
 * A {@link RegistryDeserializer} that uses Avro and delegates to {@link AvroDeserializer}
 *
 * @param  The type of data deserialized by this deserializer
 */
public final class AvroRegistryDeserializer extends RegistryDeserializer {

    private AvroDeserializer deserializer;

    @Override
    public void configure(Map properties) {
        configure(new AvroRegistryDeserializerConfig(properties));
    }

    public void configure(AvroRegistryDeserializerConfig config) {
        super.configure(config);
        this.deserializer = createDeserializer(config);
    }

    @Override
    protected SchemaProvider createSchemaProvider() {
        return new AvroSchemaProvider();
    }

    @Override
    protected AvroDeserializer deserializer() {
        return deserializer;
    }

    @Override
    protected Schema toSchema(ParsedSchema parsedSchema) {
        return ParsedSchemas.extractTypedRawSchema(parsedSchema, Schema.class);
    }

    private AvroDeserializer createDeserializer(AvroRegistryDeserializerConfig config) {
        AvroDeserializer deserializer = AvroDeserializer.create(createGenericData(config));
        deserializer = config.readerSchemaLoading().map(deserializer::withReaderSchemaLoadingEnabled).orElse(deserializer);
        return deserializer.withReaderReferenceSchemaGenerationEnabled(config.readerReferenceSchemaGeneration());
    }

    private GenericData createGenericData(AvroRegistryDeserializerConfig config) {
        GenericData genericData = instantiateGenericData(config);
        if (config.useLogicalTypeConverters()) {
            GenericDatas.addLogicalTypeConversion(genericData);
        }
        return genericData;
    }

    private GenericData instantiateGenericData(AvroRegistryDeserializerConfig config) {
        if (useSchemaReflection) {
            return config.reflectionAllowNull() ? new AtleonReflectData.AllowNull() : new AtleonReflectData();
        } else if (config.specificAvroReader()) {
            return new SpecificData();
        } else {
            return new GenericData();
        }
    }
}