io.pravega.schemaregistry.serializer.avro.impl.AvroSerializer Maven / Gradle / Ivy

Go to download
/**
 * Copyright (c) Dell Inc., or its subsidiaries. All Rights Reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 */
package io.pravega.schemaregistry.serializer.avro.impl;

import io.pravega.schemaregistry.serializer.avro.schemas.AvroSchema;
import io.pravega.schemaregistry.client.SchemaRegistryClient;
import io.pravega.schemaregistry.serializer.shared.codec.Encoder;
import io.pravega.schemaregistry.contract.data.SchemaInfo;
import io.pravega.schemaregistry.serializer.shared.impl.AbstractSerializer;
import org.apache.avro.Schema;
import org.apache.avro.generic.GenericDatumWriter;
import org.apache.avro.generic.IndexedRecord;
import org.apache.avro.io.BinaryEncoder;
import org.apache.avro.io.EncoderFactory;
import org.apache.avro.reflect.ReflectDatumWriter;
import org.apache.avro.specific.SpecificDatumWriter;
import org.apache.avro.specific.SpecificRecord;

import java.io.IOException;
import java.io.OutputStream;

public class AvroSerializer extends AbstractSerializer {

    private final SpecificDatumWriter specificDatumWriter;
    private final GenericDatumWriter genericDatumWriter;
    private final ReflectDatumWriter reflectDatumWriter;

    public AvroSerializer(String groupId, SchemaRegistryClient client, AvroSchema schema,
                          Encoder encoder, boolean registerSchema) {
        super(groupId, client, schema, encoder, registerSchema, true);
        Schema avroSchema = schema.getSchema();
        this.specificDatumWriter = new SpecificDatumWriter<>(avroSchema);
        this.genericDatumWriter = new GenericDatumWriter<>(avroSchema);
        this.reflectDatumWriter = new ReflectDatumWriter<>(avroSchema);
    }

    @Override
    protected void serialize(T var, SchemaInfo schemaInfo, OutputStream outputStream) throws IOException {
        BinaryEncoder encoder = EncoderFactory.get().binaryEncoder(outputStream, null);

        if (SpecificRecord.class.isAssignableFrom(var.getClass())) {
            specificDatumWriter.write(var, encoder);
        } else if (IndexedRecord.class.isAssignableFrom(var.getClass())) {
            genericDatumWriter.write(var, encoder);
        } else {
            reflectDatumWriter.write(var, encoder);
        }

        encoder.flush();
        outputStream.flush();
    }
}